Introduce @SharedImmutable and mark all global constants with SharedImmutable/ThreadLocal to allow using coroutines in K/N workers

Author: qwwdfsad

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

@@ -5,6 +5,7 @@
 package kotlinx.coroutines
 
 import kotlinx.atomicfu.*
+import kotlinx.coroutines.internal.*
 import kotlin.coroutines.*
 import kotlin.coroutines.intrinsics.*
 import kotlin.jvm.*
@@ -276,6 +277,7 @@ internal abstract class AbstractContinuation<in T>(
 internal interface NotCompleted
 
 private class Active : NotCompleted
+@SharedImmutable
 private val ACTIVE: Active = Active()
 
 internal abstract class CancelHandler : CancelHandlerBase(), NotCompleted

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

@@ -9,6 +9,7 @@ import kotlin.coroutines.*
 import kotlin.jvm.*
 
 @Suppress("PrivatePropertyName")
+@SharedImmutable
 private val UNDEFINED = Symbol("UNDEFINED")
 
 @NativeThreadLocal

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

@@ -1151,9 +1151,11 @@ private const val RETRY = -1
 private const val FALSE = 0
 private const val TRUE = 1
 
+@SharedImmutable
 private val SEALED = Symbol("SEALED")
-
+@SharedImmutable
 private val EMPTY_NEW = Empty(false)
+@SharedImmutable
 private val EMPTY_ACTIVE = Empty(true)
 
 private class Empty(override val isActive: Boolean) : Incomplete {

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

@@ -1001,32 +1001,43 @@ internal abstract class AbstractChannel<E> : AbstractSendChannel<E>(), Channel<E
     )
 }
 
-/** @suppress **This is unstable API and it is subject to change.** */
-@JvmField internal val OFFER_SUCCESS: Any = Symbol("OFFER_SUCCESS")
+@JvmField
+@SharedImmutable
+internal val OFFER_SUCCESS: Any = Symbol("OFFER_SUCCESS")
 
-/** @suppress **This is unstable API and it is subject to change.** */
-@JvmField internal val OFFER_FAILED: Any = Symbol("OFFER_FAILED")
+@JvmField
+@SharedImmutable
+internal val OFFER_FAILED: Any = Symbol("OFFER_FAILED")
 
-/** @suppress **This is unstable API and it is subject to change.** */
-@JvmField internal val POLL_FAILED: Any = Symbol("POLL_FAILED")
+@JvmField
+@SharedImmutable
+internal val POLL_FAILED: Any = Symbol("POLL_FAILED")
 
-/** @suppress **This is unstable API and it is subject to change.** */
-@JvmField internal val ENQUEUE_FAILED: Any = Symbol("ENQUEUE_FAILED")
+@JvmField
+@SharedImmutable
+internal val ENQUEUE_FAILED: Any = Symbol("ENQUEUE_FAILED")
 
-/** @suppress **This is unstable API and it is subject to change.** */
-@JvmField internal val SELECT_STARTED: Any = Symbol("SELECT_STARTED")
+@JvmField
+@SharedImmutable
+internal val SELECT_STARTED: Any = Symbol("SELECT_STARTED")
 
-/** @suppress **This is unstable API and it is subject to change.** */
-@JvmField internal val NULL_VALUE: Any = Symbol("NULL_VALUE")
+@JvmField
+@SharedImmutable
+internal val NULL_VALUE: Any = Symbol("NULL_VALUE")
 
-/** @suppress **This is unstable API and it is subject to change.** */
-@JvmField internal val CLOSE_RESUMED: Any = Symbol("CLOSE_RESUMED")
+@JvmField
+@SharedImmutable
+internal val CLOSE_RESUMED: Any = Symbol("CLOSE_RESUMED")
 
-/** @suppress **This is unstable API and it is subject to change.** */
-@JvmField internal val SEND_RESUMED = Symbol("SEND_RESUMED")
+@JvmField
+@SharedImmutable
+internal val SEND_RESUMED = Symbol("SEND_RESUMED")
+
+@JvmField
+@SharedImmutable
+internal val HANDLER_INVOKED = Symbol("ON_CLOSE_HANDLER_INVOKED")
 
 internal typealias Handler = (Throwable?) -> Unit
-@JvmField internal val HANDLER_INVOKED = Any()
 
 /**
  * Represents sending waiter in the queue.

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

@@ -46,7 +46,9 @@ public class ConflatedBroadcastChannel<E>() : BroadcastChannel<E> {
     private val onCloseHandler = atomic<Any?>(null)
 
     private companion object {
+        @SharedImmutable
         private val CLOSED = Closed(null)
+        @SharedImmutable
         private val UNDEFINED = Symbol("UNDEFINED")
         private val INITIAL_STATE = State<Any?>(UNDEFINED, null)
     }

common/kotlinx-coroutines-core-common/src/internal/Atomic.kt

@@ -20,6 +20,7 @@ public abstract class OpDescriptor {
     abstract fun perform(affected: Any?): Any?
 }
 
+@SharedImmutable
 private val NO_DECISION: Any = Symbol("NO_DECISION")
 
 /**

common/kotlinx-coroutines-core-common/src/internal/Concurrent.common.kt

@@ -21,4 +21,8 @@ internal expect class ReentrantLock() {
 
 internal expect inline fun <T> ReentrantLock.withLock(action: () -> T): T
 
-internal expect fun <E> identitySet(expectedSize: Int): MutableSet<E>
+internal expect fun <E> identitySet(expectedSize: Int): MutableSet<E>
+
+@ExperimentalMultiplatform
+@OptionalExpectation
+internal expect annotation class SharedImmutable()

common/kotlinx-coroutines-core-common/src/selects/Select.kt

@@ -186,8 +186,11 @@ public suspend inline fun <R> select(crossinline builder: SelectBuilder<R>.() ->
     }
 
 
+@SharedImmutable
 internal val ALREADY_SELECTED: Any = Symbol("ALREADY_SELECTED")
+@SharedImmutable
 private val UNDECIDED: Any = Symbol("UNDECIDED")
+@SharedImmutable
 private val RESUMED: Any = Symbol("RESUMED")
 
 @PublishedApi

common/kotlinx-coroutines-core-common/src/sync/Mutex.kt

@@ -113,14 +113,22 @@ public suspend inline fun <T> Mutex.withLock(owner: Any? = null, action: () -> T
     }
 }
 
+@SharedImmutable
 private val LOCK_FAIL = Symbol("LOCK_FAIL")
+@SharedImmutable
 private val ENQUEUE_FAIL = Symbol("ENQUEUE_FAIL")
+@SharedImmutable
 private val UNLOCK_FAIL = Symbol("UNLOCK_FAIL")
+@SharedImmutable
 private val SELECT_SUCCESS = Symbol("SELECT_SUCCESS")
+@SharedImmutable
 private val LOCKED = Symbol("LOCKED")
+@SharedImmutable
 private val UNLOCKED = Symbol("UNLOCKED")
 
+@SharedImmutable
 private val EmptyLocked = Empty(LOCKED)
+@SharedImmutable
 private val EmptyUnlocked = Empty(UNLOCKED)
 
 private class Empty(

native/kotlinx-coroutines-core-native/src/CoroutineContext.kt

@@ -5,7 +5,9 @@
 package kotlinx.coroutines
 
 import kotlin.coroutines.*
+import kotlinx.coroutines.internal.*
 
+@ThreadLocal
 internal val currentEventLoop = ArrayList<BlockingEventLoop>()
 
 private fun takeEventLoop(): BlockingEventLoop =