Swift闭包的上下文捕获探秘

2023-10-20 16:33:34

在Swift中，闭包的上下文捕获是指闭包可以引用外部作用域中的变量。上下文捕获主要分为两种类型：隐式捕获和显式捕获。

隐式捕获

隐式捕获是指闭包可以自动捕获外部作用域中的变量，而无需显式声明。例如，以下代码演示了隐式捕获：

var value = 10

func outerFunction() {
    let closure = {
        print(value)
    }
    
    closure()
}

outerFunction()

在这个例子中，闭包closure隐式捕获了外部作用域中的变量value。当闭包被调用时，它可以访问value的值并将其打印出来。

隐式捕获的一个优点是它可以使代码更加简洁和易读。然而，它也有一个缺点，那就是它可能导致强引用循环。强引用循环是指两个或多个对象相互引用，从而导致内存泄漏。例如，以下代码演示了强引用循环：

class MyClass {
    var closure: (() -> Void)?
    
    init() {
        closure = { [weak self] in
            self?.doSomething()
        }
    }
    
    func doSomething() {
        print("Doing something")
    }
}

let object = MyClass()
object.closure?()

在这个例子中，类MyClass有一个闭包属性closure。闭包closure隐式捕获了类MyClass的实例object。当闭包closure被调用时，它可以访问object并调用它的方法doSomething()。然而，类MyClass也引用了闭包closure。这就导致了强引用循环。

为了避免强引用循环，可以使用显式捕获来捕获外部作用域中的变量。

显式捕获

显式捕获是指闭包可以显式声明它要捕获的外部作用域中的变量。例如，以下代码演示了显式捕获：

var value = 10

func outerFunction() {
    let closure: () -> Void = { [value] in
        print(value)
    }
    
    closure()
}

outerFunction()

在这个例子中，闭包closure显式捕获了外部作用域中的变量value。当闭包closure被调用时，它可以使用value的值。

显式捕获的一个优点是它可以避免强引用循环。然而，它也有一个缺点，那就是它可能使代码更加冗长和难以阅读。

[weak self]的使用

在Swift中，[weak self]是一个常见的用法，它可以用来防止强引用循环。例如，以下代码演示了[weak self]的使用：

class MyClass {
    weak var delegate: MyDelegate?
    
    func doSomething() {
        delegate?.doSomethingElse()
    }
}

protocol MyDelegate {
    func doSomethingElse()
}

class MyDelegateImpl: MyDelegate {
    func doSomethingElse() {
        print("Doing something else")
    }
}

let object = MyClass()
let delegate = MyDelegateImpl()
object.delegate = delegate
object.doSomething()

在这个例子中，类MyClass有一个弱引用属性delegate。弱引用属性不会阻止对象被释放。当类MyClass调用方法doSomething()时，它会调用委托对象delegate的方法doSomethingElse()。然而，如果对象delegate已经被释放，那么方法doSomethingElse()就不会被调用。

使用[weak self]可以避免强引用循环，从而防止内存泄漏。然而，它也有一个缺点，那就是它可能使代码更加冗长和难以阅读。