Add autoCloseScope contract

arrow-kt · Nov 15, 2024 · f2c51bb · f2c51bb
1 parent 7b11fb7
commit f2c51bb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/arrow-libs/core/arrow-autoclose/src/commonMain/kotlin/arrow/AutoCloseScope.kt b/arrow-libs/core/arrow-autoclose/src/commonMain/kotlin/arrow/AutoCloseScope.kt
@@ -1,7 +1,12 @@
+@file:OptIn(ExperimentalContracts::class)
+
 package arrow
 
 import arrow.atomic.Atomic
 import arrow.atomic.update
+import kotlin.contracts.ExperimentalContracts
+import kotlin.contracts.InvocationKind
+import kotlin.contracts.contract
 import kotlin.coroutines.cancellation.CancellationException
 
 /**
@@ -62,6 +67,7 @@ import kotlin.coroutines.cancellation.CancellationException
  * <!--- KNIT example-autocloseable-02.kt -->
  */
 public inline fun <A> autoCloseScope(block: AutoCloseScope.() -> A): A {
+  contract { callsInPlace(block, InvocationKind.EXACTLY_ONCE) }
   val scope = DefaultAutoCloseScope()
   var throwable: Throwable? = null
   return try {