kotlin协程实现原理梳理

我们接触协程,往往会有如下疑问,本文一一解答

  1. 异步是怎么实现的,即执行权是怎么转移的?
  2. 挂起函数执行完毕后是怎么恢复现场,继续执行后续代码的?
  3. 协程里面各部分代码都在哪个线程上执行?

一、协程的简单使用示例

注意看注释,各部分代码在哪个线程上执行

// 使用调度器启动一个协程
launch(Dispatchers.IO) {
    // 这个代码块会在 dispatcher 的线程池中的一个线程上执行,假定是A
    print("A")

    // 调用一个挂起函数,如果内部实现没有切换执行线程,将仍旧在A线程上执行
    suspendFunction()

    // 当 suspendFunction 完成后,这个代码块会尽量在原来的A线程上恢复执行,但是有可能会是别的IO线程
    print("B")
}

二、 协程的几个关键对象

1. CoroutineScope接口

定义了协程的作用域,是生命周期管理的关键类,包含CoroutineContext

public interface CoroutineScope {
    public val coroutineContext: CoroutineContext
}

2. CoroutineContext

协程执行的上下文,上面提供了很多工具方法,比如包含一个调度器

3. 协程Coroutine

理解为一个任务,是job接口的一种实现

private open class StandaloneCoroutine(
    parentContext: CoroutineContext,
    active: Boolean
) : AbstractCoroutine<Unit>(parentContext, initParentJob = true, active = active) {
    override fun handleJobException(exception: Throwable): Boolean {
        handleCoroutineException(context, exception)
        return true
    }
}

4. 调度器CoroutineDispatcher

决定了协程任务在哪个线程上运行, 简化代码如下:

public abstract class CoroutineDispatcher {
     //分发任务到特定线程
     public abstract fun dispatch(context: CoroutineContext, block: Runnable)
}
  • Dispatchers.Main:这个调度器用于在主线程中执行协程。这通常用于更新 UI 或者执行其他需要在主线程中执行的任务。如果你尝试在没有主线程的环境中使用它,比如后端应用,它会抛出异常。
  • Dispatchers.IO:这个调度器用于执行 I/O 密集型任务,比如网络请求或者读写文件。它内部使用了一个用于 I/O 任务的线程池。
  • Dispatchers.Default:这个调度器用于执行 CPU 密集型任务,比如复杂的计算或者排序操作。它内部使用了一个用于计算的线程池。
  • Dispatchers.Unconfined:这个调度器有一个特殊的行为,协程会在调用它的线程立即执行,直到第一个挂起点。当协程被唤醒时,它会在唤醒它的线程继续执行

5. 协程创建器:launch, async等

可以看到创建器被定义成CoroutineScope的扩展函数

public fun CoroutineScope.launch(
    context: CoroutineContext = EmptyCoroutineContext,
    start: CoroutineStart = CoroutineStart.DEFAULT,
    block: suspend CoroutineScope.() -> Unit
): Job {
    val newContext = newCoroutineContext(context)
    val coroutine = if (start.isLazy)
    LazyStandaloneCoroutine(newContext, block) else
    StandaloneCoroutine(newContext, active = true)
    coroutine.start(start, coroutine, block)
    return coroutine    
}

6. 挂起函数

不用多说,异步任务可以定义成挂起函数

三、协程实现原理展示

kotlin的很多特性都是用过编译器动态修改代码来实现,协程的实现原理也是一样,他通过把协程转换为一种状态机来转让执行权和恢复原来执行代码。

我们用一个简化形式的代码来理解这一点,注意看注释。

我们假定写了如下代码:

launch(Dispatchers.IO) { 
   print("A") 
   //这是一个挂起函数
   doSomething() 
   print("B") 
 } 

上述代码会被kotlin转化为:

//状态机类,很多文档也翻译成连续体
interface Continuation<T> {
    val context: CoroutineDispatcher fun resumeWith(result: Result<T>)
 }


//创建状态机类
val coroutine = object : Continuation<Unit> { 
    var label = 0 
    val coroutineDispatcher = XXX
    override fun resumeWith(result: Result<Unit>) { 
        //使用调度器来把任务分发到特定线程!!!
        coroutineDispatcher.dispatch(Runnable { 
          when (label) {
             0 -> { 
                print("A") 
                label = 1 
                doSomething(this) //注意:挂起函数被传入了额外参数,就是Continuation实例!!!
                } 
             1 -> { 
                print("B")  
                    } 
            } 
        } 
    } 
} 


fun doSomething(Continuation c){
    //原来异步逻辑....省略
    
    //执行完毕后调用连续体,恢复原来的执行流程
    c.resumeWith(XXX)
}

//启动协程 
coroutine.resumeWith(Result.success(Unit))

上述流程概括起来为3步:

  • 在协程在编译的时候会被转化为一个状态机,实现Continuation接口,
  • 挂起函数后面的代码内容会被塞入状态机的下一个状态分支
  • Continuation实现类会被当做额外参数,传递给原来的挂起函数,挂起函数执行完毕后会继续调用Continuation.resumeWith()方法恢复执行
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,080评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,422评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,630评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,554评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,662评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,856评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,014评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,752评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,212评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,541评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,687评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,347评论 4 331
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,973评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,777评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,006评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,406评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,576评论 2 349

推荐阅读更多精彩内容