Kotlin协程的JVM实现源码分析(下)

协程 根据 是否保存切换 调用栈 ,分为:

  1. 有栈协程(stackful coroutine)
  2. 无栈协程(stackless coroutine)

在代码上的区别是:是否可在普通函数里调用,并暂停其执行。

Kotlin协程,必须在挂起函数中调用和恢复,属于 无栈协程

常见的语言,协程实现:

  • 有栈协程:Go、Lua
  • 无栈协程:Kotlin、C++ 20、Clojure、JavaScript

二、无栈协程 和 Continuation

2.1 CPS(Continuation-passing-style)

在上篇源码分析中,不难发现 执行的结果,都是通过 Continuation 来返回。

2.1.1 Continuation

Continuation 就是 一个通用的回调接口,返回 Result<T> 值 或 异常。

Continuation is a generic callback interface. ------ Roman Elizarov

kotlin 复制代码
public interface Continuation<in T> {

    public val context: CoroutineContext

    public fun resumeWith(result: Result<T>)
}
2.1.2 CPS

挂起函数 调用 其他挂起函数时,会将自己的 Continuation对象 作为 completion 参数 传递,

这种传递Continuation的方式,称为 连续传递风格(Continuation-passing-style) ,简称为 CPS

挂起函数 编译后,会创建基于 ContinuationImpl 对象,把 调用者Continuation 传给 completion 构造参数:

kotlin 复制代码
internal abstract class BaseContinuationImpl(
    public val completion: Continuation<Any?>?
)
2.1.3 Continuation结果返回

上篇知道 协程执行在 BaseContinuationImpl.resumeWith 方法,

同样 结果返回逻辑 也在这里,看下代码:

和 传递逻辑顺序 相反,结果按 逐步向上 返回。

分析:当获取结果后,通过 while 循环,completion 将结果向上传递,一般是协程 StandaloneCoroutine 作为最终的 completion 完成结果回调。

2.2 状态机

无栈协程,是通过 状态机状态 保存恢复 来实现协程挂起恢复。
和 每个 回调 都要创建 回调对象 相比,状态机 通过 状态 记录 执行位置,

当 挂起函数完成后,只需 恢复状态 接着执行后面的代码。

其实就是通过 switch(label) 做判断,判断位置执行。

状态机 vs 回调,有以下几个优点:

  1. 复用 方法对象和状态,避免每次分配对象
  2. 简化 循环 和 使用 高阶函数

以下面 请求解析数据 为例,launch {} 对应的 lambda挂起函数 ,分析 Kotlin 状态机状态:

kotlin 复制代码
GlobalScope.launch {
  // 挂起点1
  val data = getData()
  // 挂起点2
  val result = parseData(data)
  println("data: $data, result: $result")
}

Kotlin编译后逻辑,以 伪代码 表示:

kotlin 复制代码
class $main$1 extends SuspendLambda {
  // 挂起点的位置
  int label;
  // 状态 对象 保存 和 恢复
  Object L$0;
  // 更多状态: L$1 L$2 ...

  Object invokeSuspend(Object result) {
    Object obj;
    switch (this.label) {
      case 0:
        this.label = 1;
        obj = getData(this);
        // 表示挂起,存储 状态 label = 1,
        // 恢复时再次调用 invokeSuspend,恢复执行下面
        if (obj == COROUTINE_SUSPENDED) {
          return COROUTINE_SUSPENDED;
        }
        // 没有break,如果没有挂起,直接 执行下面的过程

      case 1:
		// 挂起恢复后
		String data = (String) result;
		// 如果没有挂起,直接执行则是:
		// String data = (String) obj;
        this.label = 2;
        // 保存 状态
        this.L$0 = data;
        obj = parseData(data, this);
        if (obj == COROUTINE_SUSPENDED) {
          return COROUTINE_SUSPENDED;
        }

      case 2:
		// 挂起恢复后
		Integer num = (Integer) result;
		// 如果没有挂起,直接执行则是:
		// Integer num = (Integer) obj;
		// 恢复状态
        String data = (String) this.L$0;
        System.out.println("data: " + data + ",num: " + num);
        return Unit.INSTANCE;

      default:
        throw new IllegalStateException("call to 'resume' before 'invoke' with coroutine");
    }
    
  }
}

2.3 CPS Transform

上面说到调用挂起函数 continuation 会作为函数参数传递,但是 声明挂起函数时,

并没有 continuation参数。而是 Kotlin 会在参数列表 自动加上 Continuation 参数,这个操作叫做 CPS Transform

举例,下面挂起函数:

kotlin 复制代码
suspend fun <T> CompletableFuture<T>.await(): T

而在 CPS Transform 后,实际的代码是:

kotlin 复制代码
fun <T> CompletableFuture<T>.await(continuation: Continuation<T>): Any?

小结

  • Kotlin协程,通过 状态机 实现,复用闭包。
  • 挂起函数, 编译成 Continuation 回调对象,CPS。
  • suspend 以同步的编程方式,执行异步方法

文档

相关推荐
长亭外的少年1 小时前
Kotlin 编译失败问题及解决方案:从守护进程到 Gradle 配置
android·开发语言·kotlin
JIAY_WX1 小时前
kotlin
开发语言·kotlin
阿龟在奔跑2 小时前
引用类型的局部变量线程安全问题分析——以多线程对方法局部变量List类型对象实例的add、remove操作为例
java·jvm·安全·list
王佑辉2 小时前
【jvm】方法区常用参数有哪些
jvm
王佑辉2 小时前
【jvm】HotSpot中方法区的演进
jvm
Domain-zhuo2 小时前
什么是JavaScript原型链?
开发语言·前端·javascript·jvm·ecmascript·原型模式
麦田里的守望者江8 小时前
KMP 中的 expect 和 actual 声明
android·ios·kotlin
Dnelic-9 小时前
解决 Android 单元测试 No tests found for given includes:
android·junit·单元测试·问题记录·自学笔记
菠菠萝宝14 小时前
【YOLOv8】安卓端部署-1-项目介绍
android·java·c++·yolo·目标检测·目标跟踪·kotlin
恋猫de小郭15 小时前
Kotlin Multiplatform 未来将采用基于 JetBrains Fleet 定制的独立 IDE
开发语言·ide·kotlin