揭秘 Block 内存泄露的幕后黑手：深入解析循环引用的本质

2024-02-04 07:16:42

在 iOS 开发的浩瀚星空中，Block 闪耀着璀璨的光芒，几乎是每个开发者的必备武器。然而，隐藏在这耀眼光芒背后的，是一个潜藏的隐患：循环引用，随时准备吞噬你宝贵的内存。今天，让我们踏上揭开 Block 内存泄露秘密之旅，深入解析循环引用的本质，为你的代码保驾护航。

Block 的前世今生

Block，顾名思义，是一块代码片段，它以一种独特的方式封装了代码和数据。它可以跨越函数边界，将变量捕获到它的作用域中。这种能力使得 Block 成为处理异步任务、事件处理和闭包的理想工具。

然而，Block 的强大也有其代价。与普通对象不同，Block 不受 ARC（自动引用计数）的管理。相反，它们使用引用计数来跟踪它们被引用的次数。当引用计数变为 0 时，Block 将被释放。

循环引用的陷阱

问题就出在 Block 捕获变量的时候。当 Block 捕获一个强引用变量（即非可选类型）时，它会将该变量的引用计数增加 1。同时，变量也会持有 Block 的强引用，导致一个致命的循环引用。

在这个循环引用的旋涡中，变量和 Block 相互引用，导致彼此无法释放。引用计数永远不会降到 0，内存泄露就这样悄然诞生。

实战演练：避免循环引用

避免循环引用的关键在于打破变量和 Block 之间的相互引用。以下是一些行之有效的技巧：

使用弱引用： 使用弱引用（weak或unowned）来捕获变量。这将允许变量在不再需要时被释放，而不会影响 Block。
使用捕获列表： 在创建 Block 时使用捕获列表，显式指定要捕获的变量。这可以防止意外捕获导致循环引用。
使用闭包： 在某些情况下，可以使用闭包代替 Block 来捕获变量。闭包会自动捕获变量的弱引用，避免循环引用。

举一反三：技术指南

示例 1：使用弱引用

class MyClass {
    var name: String

    init(name: String) {
        self.name = name
    }

    func createBlock() -> (() -> Void) {
        weak var weakSelf = self
        return { [weak weakSelf] in
            print(weakSelf?.name)
        }
    }
}

在上面的示例中，Block 使用弱引用捕获了 self，避免了循环引用。

示例 2：使用捕获列表

class MyClass {
    var name: String

    init(name: String) {
        self.name = name
    }

    func createBlock() -> (() -> Void) {
        return { [weak self] in
            print(self?.name)
        }
    }
}