Release block on LazyXxxCoroutine start

Author: Francesco Vasco

common/kotlinx-coroutines-core-common/src/Builders.common.kt

@@ -104,9 +104,13 @@ private open class DeferredCoroutine<T>(
 
 private class LazyDeferredCoroutine<T>(
     parentContext: CoroutineContext,
-    private val block: suspend CoroutineScope.() -> T
+    block: suspend CoroutineScope.() -> T
 ) : DeferredCoroutine<T>(parentContext, active = false) {
+    private var block: (suspend CoroutineScope.() -> T)? = block
+
     override fun onStart() {
+        val block = checkNotNull(this.block) { "Already started" }
+        this.block = null
         block.startCoroutineCancellable(this, this)
     }
 }
@@ -161,9 +165,13 @@ private open class StandaloneCoroutine(
 
 private class LazyStandaloneCoroutine(
     parentContext: CoroutineContext,
-    private val block: suspend CoroutineScope.() -> Unit
+    block: suspend CoroutineScope.() -> Unit
 ) : StandaloneCoroutine(parentContext, active = false) {
+    private var block: (suspend CoroutineScope.() -> Unit)? = block
+
     override fun onStart() {
+        val block = checkNotNull(this.block) { "Already started" }
+        this.block = null
         block.startCoroutineCancellable(this, this)
     }
 }

common/kotlinx-coroutines-core-common/src/channels/Broadcast.kt

@@ -106,8 +106,10 @@ private open class BroadcastCoroutine<E>(
 private class LazyBroadcastCoroutine<E>(
     parentContext: CoroutineContext,
     channel: BroadcastChannel<E>,
-    private val block: suspend ProducerScope<E>.() -> Unit
+    block: suspend ProducerScope<E>.() -> Unit
 ) : BroadcastCoroutine<E>(parentContext, channel, active = false) {
+    private var block: (suspend ProducerScope<E>.() -> Unit)? = block
+
     override fun openSubscription(): ReceiveChannel<E> {
         // open subscription _first_
         val subscription = _channel.openSubscription()
@@ -117,6 +119,8 @@ private class LazyBroadcastCoroutine<E>(
     }
 
     override fun onStart() {
+        val block = checkNotNull(this.block) { "Already started" }
+        this.block = null
         block.startCoroutineCancellable(this, this)
     }
 }