Swift 协程秘技：Task 的取消机制

IOS

2023-10-07 18:16:06

任务取消机制详解：Swift 协程深入剖析

在异步编程的世界里，协程扮演着至关重要的角色，而任务（Task）是协程的基本执行单位。但凡是任务，都有可能被取消。那么，当任务被取消时，我们该怎么做？如何得知任务被取消了呢？本文将深入探讨 Swift 中任务取消的机制，为您揭开这一谜团。

取消操作背后的原理

取消任务的过程涉及两个关键步骤：

设置取消标识： 当一个任务被取消时，系统会设置一个特殊的取消标识。
轮询取消标识： 任务内部会不断轮询取消标识，如果检测到取消标识被设置，则任务将停止执行。

如何取消任务

取消任务可以通过 cancel() 方法实现。该方法会立即设置取消标识，通知任务停止执行。

task.cancel()

检测任务是否已取消

可以通过 isCancelled 属性检测任务是否已取消。该属性是一个布尔值，如果任务已被取消，则为 true，否则为 false。

if task.isCancelled {
    // 任务已取消
}

取消操作的注意事项

需要注意以下几点：

取消操作是不可逆的，一旦取消，任务无法恢复执行。
取消操作不会立即停止任务，而是会等待任务当前执行的代码块执行完毕。
任务内部可以使用 checkCancellation() 方法主动检查取消标识，从而及时响应取消操作。

示例代码

下面是一个示例代码，展示了如何取消任务并检测任务是否已取消：

import Foundation

func longRunningTask() async -> Void {
    for i in 0...1000000000 {
        if Task.isCancelled {
            print("Task cancelled at: \(i)")
            return
        }
    }
}

let task = Task {
    await longRunningTask()
}

// 取消任务
DispatchQueue.main.asyncAfter(deadline: .now() + 1) {
    task.cancel()
}