micrometer record Kotlin coroutine suspend functions

recordSuspend.kt

class Service(val microMeter: MicroMeterRegistry, val otherService: OtherService){
  suspend fun service(id: Long){
    recordAsync("theService") { otherService.getTheThing(id) }
  }
  
  private suspend fun <A> recordAsync(name: String,  block: suspend () -> A): A = metricsRegistry.timer(name)
    .recordSuspend(block)
}

suspend fun <A> Timer.recordSuspend(block: suspend () -> A): A =
    when (val timer = this) {
        is AbstractTimer -> timer.recordSuspendInternal(block)
        else -> block()
    }


@Suppress("INVISIBLE_MEMBER")
private suspend fun <A> AbstractTimer.recordSuspendInternal(block: suspend () -> A): A {
    val s = clock.monotonicTime()
    return try {
        block()
    } finally {
        val e = clock.monotonicTime()
        record(e - s, TimeUnit.NANOSECONDS)
    }
}