TaskLocal：Swift协程的ThreadLocal变量

2023-11-16 17:51:26

前言

在Swift协程中，变量默认是共享的，这意味着如果在一个协程中修改了一个变量的值，那么在其他协程中也能看到这个变化。这在某些情况下可能是我们想要的，但有时我们也需要定义一些只在当前协程内部共享的变量，类似于线程世界的ThreadLocal变量。

TaskLocal

Swift为协程提供了TaskLocal特性，它允许我们在一个协程内定义一个变量，该变量的值只在这个协程内部共享。这意味着其他协程无法访问或修改这个变量的值，从而实现了协程之间的变量隔离。

使用TaskLocal很简单，只需创建一个TaskLocal变量，然后在需要的时候获取或设置它的值即可。TaskLocal变量的类型可以是任何类型，但它通常用于存储一些与当前协程相关的数据，例如用户ID、会话ID或其他与当前协程相关的状态信息。

用法示例

以下是一个使用TaskLocal的示例：

import Foundation

class User {
    let id: Int
    init(id: Int) {
        self.id = id
    }
}

// 创建一个TaskLocal变量来存储当前用户
let currentUser = TaskLocal<User?>()

// 在一个协程中获取当前用户
Task {
    // 将当前用户设置为用户1
    currentUser.value = User(id: 1)

    // 打印当前用户
    print("Current user in Task 1: \(currentUser.value!.id)")

    // 在另一个协程中获取当前用户
    Task {
        // 将当前用户设置为用户2
        currentUser.value = User(id: 2)

        // 打印当前用户
        print("Current user in Task 2: \(currentUser.value!.id)")
    }

    // 在第一个协程中再次获取当前用户
    print("Current user in Task 1: \(currentUser.value!.id)")
}