Combine框架下Future内嵌Task的取消问题：async/await任务无法终止

我在Combine框架中使用async/await，给Future类编写了扩展并嵌入Task代码块，目前能正常获取结果或错误，但存在一个问题：当取消Publisher时，异步函数的执行进程无法停止。

参考实现的Future扩展代码：

extension Future where Failure == Error {

    convenience init(taskPriority: TaskPriority? = nil, asyncFunc: @escaping () async throws -> Output) {
        self.init { promise in
            Task(priority: taskPriority) {
                do {
                    let result = try await asyncFunc()
                    promise(.success(result))
                } catch {
                    promise(.failure(error))
                }
            }
        }
    }
}

Core Data数据保存的辅助函数：

func saveIntoCoreData() throws {
        throw DatabaseError.canNotSave
    }

    func doSomeThing() async throws -> String {
        return try await withCheckedThrowingContinuation { continuation in
            do {
                Thread.sleep(forTimeInterval: 5)
                try saveIntoCoreData()
                continuation.resume(with: .success("Result"))
            } catch {
                continuation.resume(with: .failure(error))
            }
        }
    }

调用Future时，期望取消Publisher能触发取消错误并终止Core Data保存操作，但仅收到Publisher的取消事件，数据仍被保存。尝试使用Task.checkCancellation()但任务并未取消，该方法无效。

核心问题是Task未与Combine的取消信号绑定，且原异步函数中的阻塞操作不支持取消。需从两方面修改：

1. 让Future的Task响应Combine取消事件 #

修改Future扩展，在初始化闭包中返回一个Cancellable，当Publisher被取消时手动取消对应的Task：

extension Future where Failure == Error {
    convenience init(taskPriority: TaskPriority? = nil, asyncFunc: @escaping () async throws -> Output) {
        self.init { promise in
            let task = Task(priority: taskPriority) {
                do {
                    let result = try await asyncFunc()
                    promise(.success(result))
                } catch {
                    promise(.failure(error))
                }
            }
            
            // 绑定取消信号：Publisher取消时终止Task
            return AnyCancellable {
                task.cancel()
            }
        }
    }
}

2. 改造异步函数支持取消 #

原代码中Thread.sleep是阻塞线程的同步操作，无法被Task取消，需替换为支持取消的Task.sleep，并在关键步骤前检查取消状态：

enum DatabaseError: Error {
    case canNotSave
}

func saveIntoCoreData() throws {
    throw DatabaseError.canNotSave
}

func doSomeThing() async throws -> String {
    // 用Task.sleep替代Thread.sleep，响应Task取消
    try await Task.sleep(nanoseconds: 5 * 1_000_000_000)
    // 检查是否被取消，若已取消则抛出CancellationError终止后续操作
    try Task.checkCancellation()
    try saveIntoCoreData()
    return "Result"
}

原理说明 #

• Combine的Future初始化闭包允许返回一个Cancellable，当Publisher被取消时，该对象的cancel()方法会被调用，此时我们手动终止Task。
• Task.sleep是异步的取消感知操作，当Task被取消时会立即抛出CancellationError，避免不必要的等待。
• Task.checkCancellation()会主动检查Task的取消状态，若已取消则抛出错误，终止后续的Core Data保存操作。

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