diff --git a/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/misc/Benchmarks.kt b/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/misc/Benchmarks.kt index a3b5dcba..7209c51e 100644 --- a/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/misc/Benchmarks.kt +++ b/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/misc/Benchmarks.kt @@ -1,12 +1,15 @@ package com.beust.kobalt.misc -fun benchmarkMillis(run: () -> Unit) : Long { +fun benchmarkMillis(run: () -> T) : Pair { val start = System.currentTimeMillis() - run() - return System.currentTimeMillis() - start + val result = run() + return Pair(System.currentTimeMillis() - start, result) } -fun benchmarkSeconds(run: () -> Unit) = benchmarkMillis(run) / 1000 +fun benchmarkSeconds(run: () -> T) : Pair { + val result = benchmarkMillis(run) + return Pair(result.first / 1000, result.second) +}