CoroutineContext 源码剖析

Android

2023-11-02 23:46:22

什么是 CoroutineContext

CoroutineContext 是协程的上下文环境，它包含了一组协程执行所需的信息，例如协程的调度器、协程的父协程、协程的生命周期状态等。CoroutineContext 可以通过 withContext 函数来访问和修改。

CoroutineContext 的实现

CoroutineContext 的实现位于 kotlinx.coroutines 包中，它是一个密封类，包含了以下几个子类：

EmptyCoroutineContext：表示一个空的协程上下文。
AbstractCoroutineContext：CoroutineContext 的抽象基类，它定义了 plus、minusKey、get 等方法。
CombinedContext：一个组合协程上下文，它可以组合多个协程上下文。
KeyCoroutineContext：一个键值对协程上下文，它包含了一个键和一个值。

CoroutineContext 的使用

CoroutineContext 可以通过 withContext 函数来访问和修改。withContext 函数的语法如下：

suspend fun <R> withContext(context: CoroutineContext, block: suspend CoroutineScope.() -> R): R

其中，context 参数指定了协程的上下文环境，block 参数指定了协程要执行的代码块。当协程执行 withContext 函数时，它将进入指定的协程上下文环境，并执行 block 参数指定的代码块。

CoroutineContext 的源码分析

接下来，我们将对 CoroutineContext 的源码进行逐行分析。

sealed class CoroutineContext {
    companion object Key<E> {
        @kotlin.internal.InlineOnly
        inline operator fun <E> invoke(name: String): Key<E> = KeyImpl(name)

        fun <E> create(name: String): Key<E> = KeyImpl(name)
    }

    abstract fun <E : Any> get(key: Key<E>): E?
    abstract fun minusKey(key: Key<*>): CoroutineContext
    abstract fun plus(context: CoroutineContext): CoroutineContext

    fun <R> fold(initial: R, operation: (R, Element) -> R): R {
        var acc = initial
        var cur = this
        while (true) {
            val element = cur.foldRecurse(acc, operation) ?: break
            acc = element.acc
            cur = element.left
        }
        return acc
    }

    internal abstract fun foldRecurse(
        initial: Any?,
        operation: (Any?, Element) -> Any?
    ): Element?
}