Merge pull request #240 from ephemient/kt/benchmark

Ensure cleanup after runBlockingBenchmark
ephemient · Jan 10, 2025 · 5e48f00 · 5e48f00
2 parents 8890dac + bafeac4
commit 5e48f00
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/...-exe/src/nonblockingBench/kotlin/com/github/ephemient/aoc2024/exe/RunBlockingBenchmark.kt b/...-exe/src/nonblockingBench/kotlin/com/github/ephemient/aoc2024/exe/RunBlockingBenchmark.kt
@@ -16,8 +16,12 @@ import kotlinx.coroutines.completeWith
 actual fun <T> runBlockingBenchmark(block: suspend CoroutineScope.() -> T): T {
     val deferred = CompletableContinuation<T>(Dispatchers.Unconfined)
     block.startCoroutine(CoroutineScope(deferred.context), deferred)
-    @OptIn(ExperimentalCoroutinesApi::class)
-    return deferred.getCompleted()
+    try {
+        @OptIn(ExperimentalCoroutinesApi::class)
+        return deferred.getCompleted()
+    } finally {
+        deferred.cancel()
+    }
 }
 
 @OptIn(InternalForInheritanceCoroutinesApi::class)