Kotlin?coroutineContext源碼層深入分析

更新時間：2022年11月15日 11:01:24 作者：Super-B

表示一個元素或者是元素集合的接口。它有一個Key（索引）的Element實例集合，每一個Element的實例也是一個CoroutineContext，即集合中每個元素也是集合

1.CoroutineContext

表示一個元素或者是元素集合的接口。它有一個Key（索引）的Element實例集合，每一個Element的實例也是一個CoroutineContext，即集合中每個元素也是集合。

如下圖所示，CoroutineContext的常見官方實現(xiàn)有以下幾種（少見的或者自定義的實現(xiàn)就不列舉，以后再聊）：

Job：協(xié)程實例，控制協(xié)程生命周期（new、acruve、completing、conpleted、cancelling、cancelled）。
CoroutineDIspatcher：協(xié)程調(diào)度器，給指定線程分發(fā)協(xié)程任務(wù)（IO、Default、Main、Unconfined）。
CoroutineName：協(xié)程名稱，用于定義協(xié)程的名稱，調(diào)試打印信息使用。
CoroutineExceptionHandler：協(xié)程異常處理器，用于處理未捕獲的異常。

2.Element的作用

Element類也是繼承自CoroutineContext接口的，該類的作用是給子類保留一個Key成員變量，用于在集合查詢的時候可以快速查找到目標coroutineContext，Key成員變量是一個泛型變量，每個繼承自Element的子類都會去覆蓋實現(xiàn)Key成員變量（一般是使用子類自己去覆蓋Key），就比如拿最簡單的CoroutineName類來舉例子：

public data class CoroutineName( 
    /**
     * User-defined coroutine name.
     */
    val name: String
) : AbstractCoroutineContextElement(CoroutineName) {
    /**
     * Key for [CoroutineName] instance in the coroutine context.
     */
    public companion object Key : CoroutineContext.Key<CoroutineName>
    /**
     * Returns a string representation of the object.
     */
    override fun toString(): String = "CoroutineName($name)"
}
@SinceKotlin("1.3")
public abstract class AbstractCoroutineContextElement(public override val key: Key<*>) : Element
/**
 * Key for the elements of [CoroutineContext]. [E] is a type of element with this key.
 */
public interface Key<E : Element>
/**
 * An element of the [CoroutineContext]. An element of the coroutine context is a singleton context by itself.
 */
public interface Element : CoroutineContext {
    /**
     * A key of this coroutine context element.
     */
    public val key: Key<*>
    public override operator fun <E : Element> get(key: Key<E>): E? =
        @Suppress("UNCHECKED_CAST")
        if (this.key == key) this as E else null
    public override fun <R> fold(initial: R, operation: (R, Element) -> R): R =
        operation(initial, this)
    public override fun minusKey(key: Key<*>): CoroutineContext =
        if (this.key == key) EmptyCoroutineContext else this
}

上面的CoroutineName構(gòu)造函數(shù)定義為

public data class CoroutineName( 
    /**
     * User-defined coroutine name.
     */
    val name: String
) : AbstractCoroutineContextElement(CoroutineName)

父類構(gòu)造函數(shù)中傳遞的參數(shù)是CoroutineName，但是我們發(fā)現(xiàn)CoroutineName也不是Key接口public interface Key<E : Element>的實現(xiàn)，為啥可以這樣直接傳遞呢？但是我們仔細看發(fā)現(xiàn)CoroutineName類定義了伴生對象： public companion object Key : CoroutineContext.Key<CoroutineName>，在kotlin中伴生對象是可以直接省略 類.companion.調(diào)用方式的，CoroutineName類也就代表著伴生對象，所以可以直接作為CoroutineName父類構(gòu)造函數(shù)的參數(shù)，神奇的kotlin語法搞得我一愣一愣的。

類似的還有Job，CoroutineDIspatcher，CoroutineExceptionHandler的成員變量Key的覆蓋實現(xiàn)：

//Job
public interface Job : CoroutineContext.Element {
    /**
     * Key for [Job] instance in the coroutine context.
     */
    public companion object Key : CoroutineContext.Key<Job> { //省略 }
     //省略 
}
//CoroutineExceptionHandler
public interface CoroutineExceptionHandler : CoroutineContext.Element {
    /**
     * Key for [CoroutineExceptionHandler] instance in the coroutine context.
     */
    public companion object Key : CoroutineContext.Key<CoroutineExceptionHandler>
 	//省略 
}
// CoroutineDIspatcher
@SinceKotlin("1.3")
public interface ContinuationInterceptor : CoroutineContext.Element {
    /**
     * The key that defines *the* context interceptor.
     */
    companion object Key : CoroutineContext.Key<ContinuationInterceptor>
}

3.CoroutineContext相關(guān)的操作符原理解析

CoroutineContext的操作符？？有點莫名其妙的感覺，僅僅憑借我的直覺的話很難理解，但是平常使用協(xié)程的過程中，我們經(jīng)常會使用這些相關(guān)的操作符，比如 +，[]等等符號，下面代碼示例：

val comb = Job() + CoroutineName("")
val cName = comb[CoroutineName]

上面的+代表兩個coroutineContext合并到集合中，這里的集合實際上是一個鏈表，后面會講到。

上面的[]代表著從集合中索引出CoroutineName類型的CoroutineContext，這里也可以看出來僅僅通過key就查找出元素和map很相似，那么可以知道value是唯一的。key都是coroutineContext子類作為泛型類型的，具有唯一性，那也可以間接推斷出上面+操作其實也會覆蓋擁有相同key的value的值。

還有其他操作函數(shù)：fold展開操作， minusKey刪除集合中存在的元素。

還有一個問題就是，這個集合到底是什么類型的集合，已經(jīng)如何管理的，我們來一一解答：

3.1.什么類型的集合

CoroutineConetxt集合是鏈表結(jié)構(gòu)的集合，是一個從本節(jié)點開始，向左遍歷parent節(jié)點的一個鏈表，節(jié)點的都是CoroutineContext的子類，分為Element，CombinedContext，EmptyCoroutineContext三種。

有以下代碼作為舉例：

val scope = CoroutineScope(CoroutineName("") + Job() + CoroutineExceptionHandler{<!--{C}%3C!%2D%2D%20%2D%2D%3E--> _, _ -> } + Dispatchers.Default)

假如CoroutineScope自己的coroutineContext變量集合中是包含CoroutineName，Job，CoroutineExceptionHanlder，CoroutineDIspatcher四種上下文的，那么他們組成的集合結(jié)構(gòu)可能就會是下圖所示的鏈表結(jié)構(gòu)，

使用scope查找對應(yīng)的Job的話直接調(diào)用scope[Job]方法，替代Job的話調(diào)用 scope + Job()，看源碼就是使用scope的上下文集合替換Job

public operator fun CoroutineScope.plus(context: CoroutineContext): CoroutineScope =
    ContextScope(coroutineContext + context)

為啥是鏈表結(jié)構(gòu)的集合呢，接下來直接看源碼就知道了。

3.2.如何管理

我們集合的鏈表結(jié)構(gòu)，每個節(jié)點都是CombinedContext類型，里面包含了element，left兩個成員變量，left指向鏈表的左邊，element表示當前節(jié)點的上下文元素（一般是job，name，handler，dispatcher四種），鏈表的最左端節(jié)點一定是Element元素

Element

主要實現(xiàn)在combinedContext，Element元素的方法實現(xiàn)比較簡單，不單獨列舉。

combinedContext

構(gòu)造函數(shù)

@SinceKotlin("1.3")
internal class CombinedContext(
    private val left: CoroutineContext,
    private val element: Element
) : CoroutineContext, Serializable {

get函數(shù):

	//Element
    public override operator fun <E : Element> get(key: Key<E>): E? =
          @Suppress("UNCHECKED_CAST")
          if (this.key == key) this as E else null
	//CombinedContext
    override fun <E : Element> get(key: Key<E>): E? {
        var cur = this
        while (true) {
            cur.element[key]?.let { return it }
            val next = cur.left
            if (next is CombinedContext) {
                cur = next
            } else {
                return next[key]
            }
        }
    }

在代碼中一般不會使用get方法，而是使用context[key]來代替，類似于map集合的查詢。上下文是Element類型，key是對應(yīng)類型那么返回當前Element，不是當前類型，返回null；上下文是CombinedContext類型，指針cur指向當前節(jié)點，while循環(huán)開始，當前的element元素的key查找到了，那么就返回當前combinedContext，如果沒找到，那么將指針指向left節(jié)點，如果left節(jié)點是combinedContext類型，那么重復上述操作，如果是Element類型直接判斷是否可以查找到key值。那么從這里看出鏈表的最左端元素一定是Element節(jié)點。

contain函數(shù)

    private fun contains(element: Element): Boolean =
        get(element.key) == element
    private fun containsAll(context: CombinedContext): Boolean {
        var cur = context
        while (true) {
            if (!contains(cur.element)) return false
            val next = cur.left
            if (next is CombinedContext) {
                cur = next
            } else {
                return contains(next as Element)
            }
        }
    }

類似于get操作，contains函數(shù)直接調(diào)用get方法來判斷元素是不是和傳入?yún)?shù)相等。

containAll函數(shù)就是遍歷參數(shù)的鏈表節(jié)點是不是都包含在當前鏈表中。

fold函數(shù)

   	//coroutineContext
    public fun <R> fold(initial: R, operation: (R, Element) -> R): R

從表面意思就是展開操作，第一個入?yún)?CoroutineContext，第二個入?yún)?lambda表達式用表達式的兩個參數(shù)CoroutineContext, Element 返回一個新的 CoroutineContext：

operation :（R , Element) -> R

	Job.fold(CoroutineName("測試"),{ coroutineContext , element ->
        TODO("return new CoroutineContext")
    }) //example
    //Element
	public override fun <R> fold(initial: R, operation: (R, Element) -> R): R =
    	operation(initial, this)

作為receiver的上下文是Element，調(diào)用fold的話，是讓ELement和入?yún)⒌腃oroutineContext作為lambda表達式的兩個參數(shù)調(diào)用該lambda表達式返回結(jié)果。

    MainScope().coroutineContext.fold(Job(),{ coroutineContext , element ->
        TODO("return new CoroutineContext")
    }) //example
	//CombinedContext
	public override fun <R> fold(initial: R, operation: (R, Element) -> R): R =
        operation(left.fold(initial, operation), element)

作為receiver的上下文是CombinedContext，調(diào)用fold的話，是讓left深度遞歸調(diào)用fold函數(shù)，一直到鏈表的最左端節(jié)點，我們知道鏈表的最左端節(jié)點一定是Element，那么根據(jù)上面的代碼，Element的fold函數(shù)內(nèi)調(diào)用operation返回一個CoroutineContext后，遞歸回溯到上一層，繼續(xù)調(diào)用operation返回一個CoroutineContext，繼續(xù)回溯，一直回溯到開始調(diào)用MainScope().coroutineContext.fold(Job的地方。如下圖所示：

minusKey函數(shù)

該函數(shù)的意思是從上下文集合中刪除key對應(yīng)的上下文。

  	//Element
    public override fun minusKey(key: Key<*>): CoroutineContext =
            if (this.key == key) EmptyCoroutineContext else this

接收者receiver是Element類型的話，如果入?yún)ey和receiver是相等的話，那么返回EmptyCoroutineContext空上下文，否則返回receiver本身。（可見找到得到key的話會返回空上下文，找不到的話返回本身）

	//CombinedContext
	public override fun minusKey(key: Key<*>): CoroutineContext {
	    element[key]?.let { return left }
	    val newLeft = left.minusKey(key)
	    return when {
	        newLeft === left -> this
	        newLeft === EmptyCoroutineContext -> element
	        else -> CombinedContext(newLeft, element)
	    }
	}

接收者receiver是CombinedContext類型的話，

element[key]不為空說明當前節(jié)點就是要找的節(jié)點，直接返回該節(jié)點的left節(jié)點（代表著把當前節(jié)點跳過，也就是移除該節(jié)點）。
element[key]為空那么說明當前節(jié)點不是要找的節(jié)點，需要向鏈表的左端left去尋找目標，深度遞歸遍歷left.minusKey(key)，返回的newLeft有三種情況：
newLeft === left，在左邊找不到目標Key（根據(jù)Element.minusKey函數(shù)發(fā)現(xiàn)，返回的是this的話就是key沒有匹配到），從該節(jié)點到左端節(jié)點都可以返回。
newLeft === EmptyCoroutineContext ，在左邊找到了目標key（根據(jù)Element.minusKey函數(shù)發(fā)現(xiàn)，返回的是EmptyCoroutineContext 的話key匹配到了Element），找到了目標那么需要將目標跳過，那么從本節(jié)點開始返回，左邊節(jié)點需要跳過移除，該節(jié)點就成了鏈表的最左端節(jié)點Element。
不是上述的情況，那么就是newLeft是觸發(fā)了1.或者4.情況，返回的是left的element元素，或者是本節(jié)點的left節(jié)點跳過了，返回的是left.left節(jié)點，這樣將newLeft和本節(jié)點的element構(gòu)造出新的CombinedContext節(jié)點。

上述操作，都只是在鏈表上跳過節(jié)點，然后將跳過的節(jié)點左節(jié)點left和右節(jié)點創(chuàng)建新的CombinedContext，產(chǎn)生一個新的鏈表出來。

操作例子：

刪除最左端節(jié)點

刪除中間節(jié)點：

結(jié)論：minusKey的操作只是將原始鏈表集合中排除某一個節(jié)點，然后復制一個鏈表返回，所以并不會影響原始集合

plus函數(shù)

該函數(shù)重寫+操作符，函數(shù)定義operator fun plus(context: CoroutineContext): CoroutineContext ,作用是對上下文集合進行添加（相同會覆蓋）指定上下文操作。這個函數(shù)只有CoroutineContext實現(xiàn)了，代碼如下：

    /**
     * Returns a context containing elements from this context and elements from  other [context].
     * The elements from this context with the same key as in the other one are dropped.
     */
    public operator fun plus(context: CoroutineContext): CoroutineContext =
        if (context === EmptyCoroutineContext) this else // fast path -- avoid lambda creation
            context.fold(this) { acc, element ->
                val removed = acc.minusKey(element.key)
                if (removed === EmptyCoroutineContext) element else {
                    // make sure interceptor is always last in the context (and thus is fast to get when present)
                    val interceptor = removed[ContinuationInterceptor]
                    if (interceptor == null) CombinedContext(removed, element) else {
                        val left = removed.minusKey(ContinuationInterceptor)
                        if (left === EmptyCoroutineContext) CombinedContext(element, interceptor) else
                            CombinedContext(CombinedContext(left, element), interceptor)
                    }
                }
            }

入?yún)⑷绻荅mptyContext，那么直接返回；不是空的話，對入?yún)⑦M行fold操作，上面講了fold操作是將context鏈表展開，從鏈表最左端開始向context回溯調(diào)用fold函數(shù)的入?yún)ambda表達式。那么我們就知道了![A (CombineContext) + B(CombineContext)](https://img-blog.csdnimg.cn/40fb2861842a4c8f8c918752e3f89fd0.png) 是如何操作的了，首先B作為plus的入?yún)?，那么B先展開到B鏈表結(jié)構(gòu)的最左端，然后執(zhí)行l(wèi)ambda操作{ acc, element -> ... }, 這個lambda里面

第一步

context.fold(this) { acc, element ->
	acc.minusKey(Element.Key)
	// ...
}
//CombineContext的實現(xiàn)
public override fun <R> fold(initial: R, operation: (R, Element) -> R): R =
    operation(left.fold(initial, operation), element)

根據(jù)CombineContext的實現(xiàn)，知道lambda的acc參數(shù)是A (CombineContext)，element參數(shù)是B(CombineContext)的fold遞歸的當前位置的element的元素，acc.minusKey(Element.Key)所做的事情就是移除A (CombineContext)鏈表中的B(CombineContext)的element元素。

第二步

if (removed === EmptyCoroutineContext) element else {
	// make sure interceptor is always last in the context (and thus is fast to get when present)
	val interceptor = removed[ContinuationInterceptor]
	// ...
}

第一步移除掉element之后，判斷剩余的removed鏈表是不是empty的，如果為空，返回B(CombineContext)的fold遞歸位置的element元素；不為空，接著從removed鏈表中獲取ContinuationInterceptor上下文（也就是dispatcher）。

第三步

if (interceptor == null) CombinedContext(removed, element) else {
    val left = removed.minusKey(ContinuationInterceptor)
    if (left === EmptyCoroutineContext) CombinedContext(element, interceptor) else
        CombinedContext(CombinedContext(left, element), interceptor)
}

獲取的interceptor為空，那將element和removed鏈表構(gòu)造出一個新的CombinedContext節(jié)點返回；如果不為空，從removed鏈表中移除interceptor返回一個不包含interceptor的鏈表left；移除后left鏈表為空，那么將element和interceptor構(gòu)造出一個新的CombinedContext節(jié)點返回；left鏈表不為空，那么將left, element構(gòu)造出一個新的CombinedContext節(jié)點，將新的CombinedContext節(jié)點和interceptor早構(gòu)造出一個新的節(jié)點返回。

每一層遞歸fold操作結(jié)束后，返回一個新的context給上一層繼續(xù)遞歸，直到結(jié)束為止。

操作例子圖如下：

有如下兩個集合A (CombineContext) + B(CombineContext)：

第一次遞歸回溯：

第二次遞歸回溯：

回溯深度取決于入?yún)的鏈表長度，B有多長回溯就會發(fā)生幾次，這里沒有加入interceptor上下文元素，減少畫圖復雜度。

plus操作結(jié)論：

1.發(fā)現(xiàn)每次返回節(jié)點的時候，都會將interceptor移除后，放到節(jié)點的最右邊的位置，可以知道interceptor一定在鏈表的頭部；

2.lambda表達式中，一定會先移除掉相同key的上下文元素，然后用后加入的element和left鏈表新建一個CombinedContext節(jié)點插入到頭部

3.plus操作會覆蓋掉有相同key的上下文元素

4.驗證以及總結(jié)

經(jīng)過對上面的源碼的分析，可以推斷出一些上下文元素的操作符操作后，集合的元素排列狀態(tài)。比如下面操作：

    private fun test() {
        val coroutineContext = Job() + CoroutineName("name1") + Dispatchers.IO + CoroutineExceptionHandler{ c,e -> }
        Log.i(TAG, "coroutineContext $coroutineContext")
        val newContext = coroutineContext + SupervisorJob()
        Log.i(TAG, "newContext $newContext")
        val newContext2 = newContext + (Job() + CoroutineName("name2"))
        Log.i(TAG, "newContext2 $newContext2")
        Log.i(TAG, "newContext2[CoroutineName] ${newContext2[CoroutineName]}")
    }

打印的日志如下：

I/MainActivity: coroutineContext [
                                   JobImpl{Active}@b32c44,
                                   CoroutineName(name1),
                                   com.meeting.kotlinapplication.MainActivity$test$$inlined$CoroutineExceptionHandler$1@45ec12d,
                                   Dispatchers.IO
                               ]

I/MainActivity: newContext [
                               CoroutineName(name1),
                               com.meeting.kotlinapplication.MainActivity$test$$inlined$CoroutineExceptionHandler$1@45ec12d,
                               SupervisorJobImpl{Active}@1022662,
                               Dispatchers.IO
                           ]

I/MainActivity: newContext2 [
                               com.meeting.kotlinapplication.MainActivity$test$$inlined$CoroutineExceptionHandler$1@45ec12d,
                               JobImpl{Active}@76863f3,
                               CoroutineName(name2),
                               Dispatchers.IO
                           ]

I/MainActivity: newContext2[CoroutineName] CoroutineName(name2)

I/MainActivity: coroutineContext [
                                   JobImpl{Active}@b32c44,
                                   CoroutineName(name1),
                                   com.meeting.kotlinapplication.MainActivity$test$$inlined$CoroutineExceptionHandler$1@45ec12d,
                                   Dispatchers.IO
                               ]