7. 取消与超时

协程是协作式取消：调用 cancel() 只是「打个招呼」，协程会在下一个挂起点（delay / withContext / await 等）检查并真正停止。它不是线程的强制 interrupt，所以写重活时你要主动配合检查。

一、全家福（按使用频率）

二、协作式取消：原理与场景

2.1 循环里必须主动检查

下面代码写在 ViewModel 的方法中。注意 repeat 里有 ensureActive()——纯 CPU 循环没有挂起点，如果不主动检查，cancel() 要等循环结束才生效。

class PollingViewModel(private val repo: UserRepository) : ViewModel() {
    private var job: Job? = null

    // ① 启动轮询（写在 ViewModel 方法里）
    fun startPolling() {
        job = viewModelScope.launch {
            repeat(1000) { i ->
                ensureActive()              // 主动检查：被取消就在这里抛出 CancellationException
                repo.pushHeartbeat(i)        // 挂起函数，本身也会在入口检查取消
                delay(1000)                  // 挂起点，取消会在此刻生效
            }
        }
    }

    // 离开页面 / 退到后台时停掉
    fun stopPolling() {
        job?.cancel()                        // 只取消，不等待
        // job?.cancelAndJoin()             // 取消并等它真正结束（在 suspend 函数里用）
    }
}

2.2 取消会向上冒泡

结构化并发下，父作用域取消，所有子孙协程一起取消。这正是「页面关了，里面所有请求自动停」的来源。

viewModelScope.launch {            // 父
    launch { fetchA() }            // 子 A
    launch { fetchB() }            // 子 B
    // 本协程取消 → A、B 一并取消
}

三、超时控制

3.1 超时就抛异常 vs 返回 null

下面写在 ViewModel 的方法中，数据请求来自 Repository。

class UserViewModel(private val repo: UserRepository) : ViewModel() {

    // ① 超时抛异常：用 try/catch 兜成 UI 错误态
    fun loadWithTimeout() = viewModelScope.launch {
        try {
            val data = withTimeout(2000L) { repo.fetchSlowData() }   // >2s 抛 TimeoutCancellationException
            _state.value = UiState.Success(data)
        } catch (e: TimeoutCancellationException) {
            _state.value = UiState.Error("加载超时，请重试")
        }
    }

    // ② 超时返回 null：可选数据，拿不到就走兜底
    fun loadOptional() = viewModelScope.launch {
        val banner = withTimeoutOrNull(800L) { repo.fetchBanner() }  // 超时返回 null
        _banner.value = banner ?: fallbackBanner                     // 静默兜底
    }
}

3.2 取消时做不可打断的收尾

有时取消也要把一半的进度安全落库，不能半途而废。用 NonCancellable 包住收尾逻辑：

val job = scope.launch {
    try {
        doWork()
    } finally {
        // 取消也会进 finally；用 NonCancellable 保证收尾不被再打断
        withContext(NonCancellable) {
            saveProgress()          // 安全落盘
            closeResources()
        }
    }
}

四、真实开发中会踩的坑

try {
    repo.fetch()
} catch (e: CancellationException) {
    throw e                       // 必须原样抛出，保留取消信号
} catch (e: IOException) {
    _state.value = UiState.Error(e.message)
}