Swift 的 Optional.take()：一次性消费值的神器

原文：Exploring the take() method for Optionals

什么是 take()？

take() 是 Swift6 标准库提供的 Optional 实例方法：

"如果有值就返回它，同时把自己置为 nil；如果已经是 nil 就返回 nil。" 等价于"原子地消费一次"。

语义类似于 Rust 的 Option::take()、C++ 的 std::optional::reset() + 返回值。

为什么需要它？

场景	不用 take()的问题	take()带来的好处
一次性 Token、验证码、密钥	手动 if let + 置 nil容易忘	原子消费，不会二次使用
懒加载缓存	并发下可能重复创建	把"判断 + 清空"做成原子操作
资源释放	提前 nil导致重复初始化	保证只拿一次，线程安全

探索实现

标准库中该函数的定义：public mutating func take() -> Wrapped?

如果我们自己写，两行扩展即可拥有：

extension Optional {
    /// 如果 self 有值则返回它并把 self 置为 nil，否则返回 nil
    public mutating func take() -> Wrapped? {
        defer { self = nil }
        return self
    }
}

注意：必须是 mutating------值类型要改自己。

官方并不是通过extension实现，而是在定义主体里实现的。实现更优雅，使用了consume消费自身

public mutating func take() -> Self {
    let result = consume self
    self = nil
    return result
}

实战对比：代码更短、意图更清晰

传统写法

var token: String? = "a640f873-bf52-45b0-b13a-5bcef123aa2a"

if let value = token {
    print(value)               // 使用
    token = nil                // 手动失效
} else {
    print("token 已被使用过")
}

使用 take()

var token: String? = "a640f873-bf52-45b0-b13a-5bcef123aa2a"

if let value = token.take() {
    print(value)               // 打印一次
    print(token as Any)        // nil
} else {
    print("token 已被使用过")
}

一行完成"取值 + 清空"，不会漏掉置 nil。

进阶场景

线程安全的一次性缓存

class ImageCache {
    private var _image: UIImage?
    
    /// 只有第一次调用会真正下载，后续都返回 nil
    func takeImage() -> UIImage? {
        return _image.take()        // 原子拿图
    }
}

防止重复提交网络请求

var pendingTask: URLSessionDataTask? = URLSessionDataTask()

// 用户狂点按钮也只发一次
if let task = pendingTask.take() {
    task.resume()
} else {
    print("请求已发出，请勿重复点击")
}

与 Optional chaining 组合

var config: Config? = loadConfig()
let dbPath = config.take()?.databasePath   // 取出即清空

与 Optional.map 的区别

操作	是否改变原 Optional	语义
map	❌ 不变	只转换值，不消费
take	✅ 置 nil	一次性拿值

var x: Int? = 3
let a = x.map { $0 * 2 }   // a = 6,  x 仍是 3
let b = x.take()           // b = 3,  x 变成 nil

小结：什么时候用 take()？

1. 一次性凭证（Token、验证码、邀请码）
2. 懒加载且只加载一次的单例资源
3. 需要原子"拿完即焚"的并发场景
4. 想让代码更短、意图更明显

记住：take() = "拿值 + 废掉它"。