Kotlin 协程线程切换机制详解

一、核心实现原理

Kotlin 协程通过挂起函数 + 调度器 + 状态机实现线程切换：

Kotlin 复制代码

viewModelScope.launch(Dispatchers.Main) {          // 1. 主线程启动
    val data = withContext(Dispatchers.IO) {        // 2. 切换到IO线程
        fetchDataFromNetwork()                     // 3. 执行网络请求
    }
    updateUI(data)                                 // 4. 自动切回主线程
}

二、核心组件解析

调度器 (Dispatcher)：
- Dispatchers.Main：Android 主线程
- Dispatchers.IO：I/O 密集型线程池
- Dispatchers.Default：CPU 密集型线程池
- Dispatchers.Unconfined：无限制调度器
挂起函数 (Suspend Function)：
- 编译器将挂起点转换为状态机
- 使用 Continuation 保存/恢复执行状态
协程上下文 (CoroutineContext)：
- 包含调度器、异常处理器等元素
- 使用 CoroutineContext 接口实现组合

三、线程切换流程

四、关键技术点

CPS 变换 (Continuation Passing Style)：

Kotlin 复制代码

// 编译器转换前
suspend fun fetchData(): String

// 转换后
fun fetchData(cont: Continuation<String>): Any

状态机实现：

java 复制代码

class FetchDataStateMachine(
    cont: Continuation<String>
) : ContinuationImpl(cont) {
    // 状态标识
    int label = 0
    
    // 恢复执行点
    override fun invokeSuspend(result: Result<String>) {
        when (label) {
            0 -> { /* 初始状态 */ }
            1 -> { /* 恢复状态 */ }
        }
    }
}

线程调度核心代码：

Kotlin 复制代码

class FetchDataStateMachine(
    cont: Continuation<String>
) : ContinuationImpl(cont) {
    // 状态标识
    int label = 0
    
    // 恢复执行点
    override fun invokeSuspend(result: Result<String>) {
        when (label) {
            0 -> { /* 初始状态 */ }
            1 -> { /* 恢复状态 */ }
        }
    }
}

五、总结

Q： Kotlin 协程是如何实现线程切换的？

A：

Kotlin 协程通过三个核心机制实现线程切换：

调度器 (Dispatcher)

协程使用 Dispatchers（如 Main、IO、Default）指定代码块执行线程。调度器底层维护线程池，如 IO 调度器使用 64 线程池。
挂起/恢复机制

当遇到 withContext 等挂起点时：
- 保存当前执行状态到 Continuation 对象
- 释放当前线程资源
- 目标调度器从线程池获取新线程执行代码
- 执行完成后恢复 Continuation 并切回原线程
编译器转换

Kotlin 编译器通过 CPS（Continuation Passing Style）转换：
- 将挂起函数转换为状态机
- 每个挂起点对应状态机状态
- 使用 Continuation 对象保存局部变量和执行点

示例流程：

Kotlin 复制代码

// 主线程启动
launch(Dispatchers.Main) {
    // 状态0：主线程执行
    val data = withContext(Dispatchers.IO) { 
        // 状态1：切换到IO线程
    }
    // 状态2：自动切回主线程 
}

当执行到 withContext 时，协程挂起并保存状态（包括局部变量），IO 线程执行完成后，调度器将结果和状态派发回主线程恢复执行。

优势：

非阻塞式线程切换
同步写法实现异步操作
精准控制生命周期（通过 Job 结构化并发）

六、高频面试追问

Continuation 是什么？

是保存协程执行状态的回调接口，核心方法 resumeWith(result) 用于恢复协程执行。
协程比线程高效在哪里？
- 线程切换涉及内核态转换
- 协程切换在用户态完成
- 单个线程可运行数万个协程
Dispatchers.IO 和 Default 区别？
- IO：针对阻塞 I/O 优化（网络/文件），最大 64 线程
- Default：CPU 密集型计算，固定 CPU 核数线程
如何避免协程内存泄漏？

使用 viewModelScope/lifecycleScope 自动取消，或在 onDestroy 中手动取消 job.cancel()