本篇文章已授权微信公众号 guolin_blog （郭霖）独家发布[2023-03-02]

前言: kotlin协程源码十分庞大, 本篇只能吧我理解的源码聊一聊,不会特别深入研究,只会浅浅的看看表层. 本来计划协程系列是10篇左右,后续是flow热流冷流之类的, 冷流操作符之类的应该不会在写了, flow当作Rxjava来用就可以,后续可能还会写一篇关于热流的文章. 也可能没有:) 主要是不好写,文字写出来还是比较生硬....

如果没有看过前几篇,建议先看看前几篇, 本篇遇到前几篇的知识不会重复说!

launch 浅析

源码阅读从最简单的一个launch开始!

在launch的时候, 会执行 CoroutineScope.newCoroutineContext函数这里会传入一个EmptyCoroutineContext

CoroutineScope.newCoroutineContext 会走 foldCopies , 这个函数是用来合并2个协程的

先来看看 foldCopies的参数

coroutineContext // 可以看出,此时coroutineContext为JobImpl 我们稍后来看看它是在什么时候赋值的
context // 默认什么都没有传,是 EmptyCoroutineContext
true // isNewCoroutine 是否创建新的Coroutine

我们在创建CoroutineScope的时候,会对coroutineContext赋值

我们在创建协程作用域的时候, 会初始化一个Job, Job的默认实现为JobImpl

好了,在回到 CoroutineScope.newCoroutineContext方法

执行完foldCopies后,

我们知道此时返回结果combined = jobImpl,

最后当返回的时候 if (combined !== Dispatchers.Default && combined[ContinuationInterceptor] == null)

最终给他添加一个默认调度器 [Dispatchers.Default]

tips: ContinuationInterceptor 是协程拦截器, 下面会说,不要急

再次回到主线流程

此时newContext = [JobImpl , Dispatchers.Default]

如果说,有子协程的情况下,newCoroutineContext 这个方法就会使用最新的coroutineContext, 例如这样:

这都是 foldCopies 的功劳, foldCopies的细节就不看了,没意义 [比较烧脑,不想看]

再接着往下走:

在聊协程启动模式的时候说过,coroutineStart一共有4中模式

DEFAULT
LAZY
UNDISPATCHED
ATOMIC // 实验中

此时会判断是否是懒加载模式, 很明显不是懒加载,所以会走 StandaloneCoroutine

StandaloneCoroutine 会通过 handleJobException 捕获一些异常.

比如说在JVM中使用Main线程的时候,会提示

Module with the Main dispatcher had failed to initialize. For tests Dispatchers.setMain from kotlinx-coroutines-test module can be used

在这里就能捕获到

在继续往下看:

接下来就是开启一个协程

这里通过kotlin 的特性重载操作符,直接去它类中找到对应的方法即可

因为此时协程启动模式是DEFAULT,那么直接看:

block.startCoroutineCancellable(receiver, completion)

即可

在接着往下看

createCoroutineUnintercepted(receiver, completion) 函数源码位置

这个函数需要去kotiln官网看,太深了,我没看,我找到了提供给需要的人!

这里看看拦截的方法

可以看出,此时就会分发 continuation

先来看看不写 ContinuationInterceptor 是什么效果

这段代码写过无数次, 因为协程默认启动模式需要调度,所以协程体内的代码还没来得及执行,main函数就已经结束了

那么在来看看添加协程拦截器后有什么变化

可以看出, 协程会立即执行,我猜测是拦截的过程中会立即触发恢复,所以才会有这样的效果, 可惜的是我没找到恢复的代码

跟随源码手动创建协程

刚才看源码的时候看到了这段代码

开启一个协程的时候, 通过函数扩展 扩展了 startCoroutineCancellable

那么我们是否也可以尝试的写一写

这里有一个注意点:

我们并不能调用 startCoroutineCancellable因为它是内部方法,我们无法直接调用

只能拿 createCoroutine 来代替

来看看我们的代码:

这里我把范型都替换成了具体的值,否则的话看的更迷糊

data class TestBean(val message: String)
fun main() {
    fun test(receiver: TestBean, block: suspend TestBean.() -> String) {
            block.createCoroutine(receiver, object : Continuation<String> { // 创建协程
                override val context: CoroutineContext
                    get() = EmptyCoroutineContext

                override fun resumeWith(result: Result<String>) {// 当协程恢复的时候执行
                    println("resume:$result")
                }
            })
    }

    val bean = TestBean("测试数据")

    test(bean) {
        println("我是test内部方法${this}")
        "我是返回数据"
    }
}