TimedDSL

TimedDSL.kt

import arrow.core.continuations.AtomicRef
import arrow.core.continuations.update
import kotlin.time.Duration
import kotlin.time.ExperimentalTime
import kotlin.time.TimedValue
import kotlin.time.measureTimedValue

@OptIn(ExperimentalTime::class)
class TimedDSL {
  val total: AtomicRef<Duration> = AtomicRef(Duration.ZERO)

  inline fun <A> time(block: () -> A): A {
    val (a, duration) = measureTimedValue(block)
    total.update { it + duration }
    return a
  }
}

/**
 * time {
 *   time { delay(1.seconds) }
 *   println("Some other code").also { delay(10.seconds) }
 *   time { delay(300.milliseconds) }
 *   "final result"
 * } // TimedValue("final result", 1.3 seconds)
 */
@OptIn(ExperimentalTime::class)
inline fun <A> time(block: TimedDSL.() -> A): TimedValue<A> {
  val timed = TimedDSL()
  val a = timed.block()
  return TimedValue(a, timed.total.get())
}