Kotlin协程源码简单聊一聊,浅浅看看表层

outside_default.png

/   今日科技快讯   /

近日,经历了一次停办,两次规模严重缩水之后,2023年世界移动通信大会(MWC)终于再现往日盛况:有来自200多个国家和地区的2000多家厂商参加,并在此次展会上发布了最新的产品与科技,仅中国就有100余家厂商参展。

每年的MWC都被称作“全球移动通信技术发展的风向标”,今年大会以“时不我待——明日科技,将至已至”为主题,具体又围绕着5G新动能、数字万物、开放网络、超越现实+ 、金融科技等五大热点主题展开。

/   作者简介   /

本篇文章转自史大拿的博客,文章主要分享了对kotlin协程源码的分析,相信会对大家有所帮助!

原文地址:

https://juejin.cn/post/7204752219915288636

/   前言   /

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

/   launch 浅析   /

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

outside_default.png

在launch的时候,会执行CoroutineScope.newCoroutineContext函数。这里会传入一个EmptyCoroutineContext。CoroutineScope.newCoroutineContext会走foldCopies , 这个函数是用来合并2个协程的。

先来看看 foldCopies的参数:

  • coroutineContext // 可以看出,此时coroutineContext为JobImpl我们稍后来看看它是在什么时候赋值的

  • context // 默认什么都没有传,是EmptyCoroutineContext

  • true // isNewCoroutine是否创建新的Coroutine

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

outside_default.png

我们在创建协程作用域的时候,会初始化一个Job,Job的默认实现为JobImpl。好了,再回到CoroutineScope.newCoroutineContext方法。

afddacbdd5e09b0629ec6a75e3fe9f3f.png

执行完foldCopies后,我们知道此时返回结果combined = jobImpl,最后当返回的时候if (combined !== Dispatchers.Default && combined[ContinuationInterceptor] == null)最终给他添加一个默认调度器Dispatchers.Default。

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

再次回到主线流程。

e9ad6d3f1234252f9765f41cebd83b5d.png

此时newContext = [JobImpl,Dispatchers.Default],如果说有子协程的情况下,newCoroutineContext这个方法就会使用最新的coroutineContext。例如这样:

outside_default.png

这都是foldCopies的功劳,foldCopies的细节就不看了,没意义。比较烧脑,不想看。再接着往下走:

outside_default.png

在聊协程启动模式的时候说过,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,在这里就能捕获到。

outside_default.png

继续往下看:

outside_default.png

接下来就是开启一个协程。

这里通过kotlin的特性重载操作符,直接去它类中找到对应的方法即可。因为此时协程启动模式是DEFAULT,那么直接看block.startCoroutineCancellable(receiver, completion)即可。

接着往下看createCoroutineUnintercepted(receiver, completion)函数,这个函数需要去kotiln官网看,太深了,我没看,我找到了提供给需要的人!源码地址:

https://github.com/JetBrains/kotlin/blob/v1.8.20-Beta/libraries/stdlib/jvm/src/kotlin/coroutines/intrinsics/IntrinsicsJvm.kt

这里看看拦截的方法:

outside_default.png

可以看出,此时就会分发continuation。先来看看不写ContinuationInterceptor是什么效果。

outside_default.png

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

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

outside_default.png

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

/   跟随源码手动创建协程   /

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

outside_default.png

开启一个协程的时候,通过函数扩展。扩展了startCoroutineCancellable。那么我们是否也可以尝试的写一写。

这里有一个注意点:我们并不能调用startCoroutineCancellable因为它是内部方法,我们无法直接调用。

outside_default.png

只能拿createCoroutine来代替,来看看我们的代码:

28c7759d6b0d0b57114ec6e941a7c8c5.png

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

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}")
        "我是返回数据"
    }
}

代码都写完了,但是这里并没有任何结果!

没有结果是正常现象,因为我们只是创建了一个协程,默认是挂起状态,只有恢复的时候才会执行代码。例如这样:

outside_default.png

如果你觉得这样很麻烦,也可以直接将createCoroutine改为startCoroutine。

  • createCoroutine // 创建协程 [默认挂起]

  • startCoroutine // 直接开启一个协程

outside_default.png

那么我们自定义CoroutineContxt是否还起作用呢?

outside_default.png

这里的时候,我们通过Job#cancel()可以发现,并没有取消协程,恢复代码还是继续执行了。

这是因为当我们使用launch{}开启一个协程的时候,系统帮我们维护了job的各种状态。这里很简单,我们自己维护job即可。

outside_default.png

系统代码指定是没有这么简单,不过模拟一下就可以了,没必要太纠结。那如果是异常如何捕获呢?

outside_default.png

需要注意的是,通过CoroutineExceptionHandler捕获异常,这里监听的是 resumeWith方法。

outside_default.png

只需要再次向上throw一下即可。

/   简单的   /

刚才我们介绍的是有receiver的startCoroutine,还有一个没有receiver的。

outside_default.png

直接来看代码:

outside_default.png

这段代码很简单,只是创建一个普通的协程并恢复,当然也可以通过CoroutineContext来控制协程,上面已经提到了,这里就不重复了。有创建协程手动恢复,就有直接创建并恢复。

outside_default.png

有receiver和没有receiver的区别也很明显,那就是在suspend方法体中是否有receiver。

/   总结   /

本篇我们阅读了CoroutineScope#launch{},并学习了手动管理协程的4个方法:

  • (suspend R.() -> T).startCoroutine // 有receiver直接恢复协程

  • (suspend R.() -> T).createCoroutine // 有receiver创建协程,需要手动恢复

  • (suspend () -> T).startCoroutine //  没有receiver直接恢复协程

  • (suspend () -> T).createCoroutine // 没有receiver创建协程,需要手动恢复

这四胞胎说实话,是真没啥用,但就是得了解一下...

完整代码地址:

https://gitee.com/lanyangyangzzz/coroutine-project/blob/main/app/src/main/java/com/szj/coroutine/project/jvm/blog6/SafeContinuationTest.kt

推荐阅读:

我的新书,《第一行代码 第3版》已出版!

Kotlin协程开发的基础入门知识

LeakCanary是怎么检测到内存泄露的,看完这篇你就懂了

欢迎关注我的公众号

学习技术或投稿

outside_default.png

outside_default.png

长按上图,识别图中二维码即可关注

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值