Kotlin现有协程取消时重启新协程的实现方案咨询

需求背景 #

当现有协程因异常或其他原因被取消时，需要自动创建新协程继续执行任务。当前通过invokeOnCompletion handler检测取消状态，但由于该方法不是挂起函数，所以借助Channel实现重启逻辑，现有临时实现代码如下：

import kotlinx.coroutines.*
import kotlinx.coroutines.channels.Channel

lateinit var backGroundTask: Job

suspend fun main(): Unit = supervisorScope {
    val restartChannel = Channel<Boolean>()
    val restartBackGroundTaskListener = startListeningForRestartEvent(restartChannel)
    backGroundTask = someBackGroundTask()
    backGroundTask.invokeOnCompletion {
        launch { restartChannel.send(true) }
    }
    val restartBackGroundTask = restartBackGroundTaskIntermittently()
    delay(1000)
    restartBackGroundTask.join()
    restartBackGroundTaskListener.join()
}

fun CoroutineScope.startListeningForRestartEvent(restartEvents: Channel<Boolean>): Job =
        launch {
            for (event in restartEvents) {
                when (event) {
                    true -> {
                        println("newJob created ..")
                        backGroundTask = someBackGroundTask()
                        backGroundTask.invokeOnCompletion {
                            launch { restartEvents.send(true) }
                        }
                    }
                    false -> {}
                }
            }
        }

fun CoroutineScope.someBackGroundTask(): Job =
    launch {
        repeat(1000) {
            delay(400)
            println(it)
        }
    }

// 模拟因某些原因取消协程的场景
fun CoroutineScope.restartBackGroundTaskIntermittently(): Job =
    launch {
        repeat(1000) {
            delay(2000)
            backGroundTask.cancelAndJoin()
        }
    }

目前已有事件循环，希望利用它触发新任务创建，startListeningForRestartEvent会持续监听重启事件，收到true时创建新的someBackGroundTask。

问题 #

1. 是否有更优的实现方式？
2. 在invokeOnCompletion内执行launch { restartChannel.send(true) }是否可行？
3. NonCancellable协程能否在此场景中发挥作用？

问题1：更优的实现方式 #

有两种更简洁内聚的实现思路：

• 递归绑定重启逻辑：去掉额外的Channel和全局变量，直接在任务的invokeOnCompletion中判断完成原因，触发新任务启动并绑定相同的重启逻辑，逻辑更紧凑：

fun CoroutineScope.startRestartableTask(): Job = launch {
    fun restart() {
        someBackGroundTask().apply {
            invokeOnCompletion { cause ->
                // 仅在取消或异常时重启，可根据需求调整判断条件
                if (cause != null) {
                    restart()
                }
            }
        }
    }
    restart()
    awaitCancellation() // 等待外部取消整个重启流程
}

• Flow重试机制：将任务包装为Flow，利用retryWhen操作符实现自动重启，贴合协程响应式编程风格：

fun CoroutineScope.startRestartableTaskWithFlow(): Job = launch {
    flow {
        repeat(1000) {
            emit(it)
            delay(400)
        }
    }.retryWhen { cause, _ ->
        // 取消或异常时触发重试
        cause is CancellationException || cause is Exception
    }.collect { println(it) }
}

问题2：invokeOnCompletion内执行launch { restartChannel.send(true) }是否可行？ #

可行，但需注意几个细节：

• invokeOnCompletion会在协程完成后的任意线程执行，用launch将发送操作调度到当前协程作用域的调度器，不会阻塞完成逻辑。
• 要保证restartChannel不会被提前关闭，否则发送操作会抛出异常，建议使用Channel<Boolean>(Channel.UNLIMITED)或确保监听协程始终活跃。
• 原代码中的全局变量backGroundTask存在线程安全风险，多个协程同时修改可能导致状态不一致，建议将变量移至作用域内管理。

问题3：NonCancellable协程能否在此场景中发挥作用？ #

可以，但仅在特定场景下有用：

• NonCancellable的核心作用是让协程在已取消状态下仍能执行挂起操作。如果重启前需要执行必须完成的挂起逻辑（比如记录日志、保存状态），可以用withContext(NonCancellable)包裹：

backGroundTask.invokeOnCompletion {
    launch {
        withContext(NonCancellable) {
            // 这里的挂起操作不会被取消，比如执行资源清理
            delay(100)
            restartChannel.send(true)
        }
    }
}

• 如果只是单纯发送重启信号，普通launch已足够，因为Channel发送操作本身不会被取消（除非Channel关闭），只有当存在不可打断的前置挂起逻辑时，NonCancellable才会发挥作用。

内容的提问来源于stack exchange，提问作者Vencat