如何确保挂起函数调用在协程取消时仍能执行完成？

问题背景 #

我们有一个包含挂起函数的数据库接口：

interface Database {
  suspend fun insert(text: String): Long

  suspend fun markAsDone(id: Long)
}

业务逻辑里需要执行attempt操作，成功后必须把状态同步到数据库，最初的实现是这样：

suspend fun doTheThing() {
    if (attempt(this.id)) db.markAsDone(this.id)
}

但这里有个问题：doTheThing是挂起函数，要是在attempt执行成功之后、markAsDone还没跑完的时候，协程被取消了，数据库里的状态就没同步上，等程序重启后可能会重复执行这个操作。

你自己想到用自定义Job启动协程的方案，但觉得不够优雅：

suspend fun doTheThing() {
    if (attempt(this.id)) {
        launch(Job()) {
            db.markAsDone(this.id)
        }
    }
}

更优雅的解决办法 #

1. 用NonCancellable上下文保证操作完成 #

直接把markAsDone放到NonCancellable上下文中执行，这样就算外部协程被取消，这个数据库操作也能跑完：

suspend fun doTheThing() {
    if (attempt(this.id)) {
        withContext(NonCancellable) {
            db.markAsDone(this@doTheThing.id)
        }
    }
}

NonCancellable是Kotlin协程提供的特殊上下文，它会忽略协程的取消请求，确保里面的挂起函数能执行完毕。不需要额外开新协程，代码简洁直接，完美解决取消导致的同步失败问题。

2. 给数据库操作加幂等性保障 #

如果担心极端情况（比如进程突然崩溃），可以让markAsDone本身具备幂等性——也就是重复调用不会搞乱数据。比如在数据库层面加判断：执行标记操作前先查一下这个ID是不是已经标记完成了，再决定要不要更新。这样就算因为意外重复执行，也不会有问题。

3. 用固定协程作用域优化你的方案 #

如果确实需要把数据库操作放到后台协程里执行，别临时创建Job()，用一个生命周期更长的固定作用域（比如应用级的），这样管理起来更规范：

// 先定义一个应用级的协程作用域
val AppScope = CoroutineScope(SupervisorJob() + Dispatchers.IO)

suspend fun doTheThing() {
    if (attempt(this.id)) {
        AppScope.launch {
            db.markAsDone(this@doTheThing.id)
        }
    }
}

固定作用域能避免临时协程带来的管理混乱，同时确保后台操作能稳定完成。

总结 #

优先选第一种方案，用NonCancellable最直接高效；再搭配幂等性设计，能应对更极端的异常情况；如果需要后台执行，固定协程作用域比临时Job优雅得多。

内容的提问来源于stack exchange，提问作者M I P A