28、Swift 3 自动引用计数(ARC)

Swift 使用自动引用计数(ARC)来自动化的跟踪和管理应用程序的内存

Swift 语言中,通常情况下开发者不需要去手动释放内存,因为 ARC 会在类的实例不再被使用时,自动释放其占用的内存

但偶尔有些时候还是需要在代码中实现内存管理

ARC 功能

  • 每次使用 init() 方法创建一个类的新的实例的时候,ARC 会分配一大块内存用来储存实例的信息
  • 内存中会包含实例的类型信息,以及这个实例所有相关属性的值
  • 当实例不再被使用时,ARC 释放实例所占用的内存,并让释放的内存能挪作他用
  • 为了确保使用中的实例不会被销毁,ARC 会跟踪和计算每一个实例正在被多少属性,常量和变量所引用
  • 实例赋值给属性、常量或变量,它们都会创建此实例的强引用,只要强引用还在,实例是不允许被销毁的

Swift ARC 范例

import Cocoa

class Person
{
    let name: String

    init(name: String)
    {
        self.name = name
        print("\(name) 开始初始化")
    }

    deinit
    {
        print("\(name) 被析构")
    }
}

// 值会被自动初始化为 nil,目前还不会引用到 Person 类的实例
var reference1: Person?
var reference2: Person?
var reference3: Person?

// 创建 Person 类的新实例
reference1 = Person(name: "DDKK.COM 弟弟快看,程序员编程资料站")
//赋值给其他两个变量,该实例又会多出两个强引用

reference2 = reference1
reference3 = reference1

//断开第一个强引用
reference1 = nil
//断开第二个强引用
reference2 = nil
//断开第三个强引用,并调用析构函数
reference3 = nil

编译运行以上 Swift 范例,输出结果为

$ swift main.swift
DDKK.COM 弟弟快看,程序员编程资料站 开始初始化
DDKK.COM 弟弟快看,程序员编程资料站 被析构

类实例之间的循环强引用

在上面的范例中,ARC 会跟踪新创建的 Person 实例的引用数量,并且会在 Person 实例不再被需要时销毁它

但是,我们可能会写出这样的代码,一个类永远不会有 0 个强引用

这种情况发生在两个类实例互相保持对方的强引用,并让对方不被销毁

这就是所谓的循环强引用

范例

下面的代码演示了一个不经意产生的循环强引用

范例定义了两个类:Person 和 Apartment,用来建模公寓和它其中的居民

import Cocoa

class Person
{
    let name: String
    var apartment: Apartment?
    init(name: String)
    {
        self.name = name
    }

    deinit 
    {
        print("\(name) 被析构")
    }
}

class Apartment
{
    let number: Int
    var tenant: Person?
    init(number: Int)
    {
        self.number = number
    }

    deinit
    {
        print("Apartment #\(number) 被析构")
    }
}

// 两个变量都被初始化为 nil

var p1: Person?
var number73: Apartment?

// 赋值
p1 = Person (name: "DDKK.COM 弟弟快看,程序员编程资料站")
number73 = Apartment(number: 73)

// 意感叹号是用来展开和访问可选变量 DDKK.COM 弟弟快看,程序员编程资料站 和 number73 中的实例
// 循环强引用被创建
p1!.apartment = number73
number73!.tenant = p1

// 断开 DDKK.COM 弟弟快看,程序员编程资料站 和 number73 变量所持有的强引用时,引用计数并不会降为 0,实例也不会被 ARC 销毁
// 注意,当你把这两个变量设为nil时,没有任何一个析构函数被调用。
// 强引用循环阻止了 Person 和  Apartment类实例的销毁,并在你的应用程序中造成了内存泄漏
p1 = nil
number73 = nil

运行以上范例,我们会发现析构函数没有被调用,因为 Person 和 Apartment 相互之间引用了对方的实例,造成了循环强引用

要怎么解决这个问题呢?

解决实例之间的循环强引用

Swift 提供了两种办法用来解决使用类的属性时所遇到的循环强引用问题

1、 弱引用;
2、 无主引用;

弱引用和无主引用允许循环引用中的一个实例引用另外一个实例而不保持强引用
这样实例能够互相引用而不产生循环强引用

使用原则

1、 对于生命周期中会变为nil的实例使用弱引用;
2、 对于初始化赋值后再也不会被赋值为nil的实例,使用无主引用;

范例

下面的代码演示了弱引用的使用

import Cocoa

class Module
{
    let name: String
    var sub: SubModule?

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

    deinit
    {
        print("\(name) 主模块")
    }
}

class SubModule
{
    let number: Int
    weak var topic: Module?

    init(number: Int)
    {
        self.number = number
    }

    deinit 
    {
        print("子模块 topic 数为 \(number)")
    }
}

var toc: Module?
var list: SubModule?

toc = Module(name: "ARC")
list = SubModule(number: 4)

toc!.sub = list
list!.topic = toc

toc = nil
list = nil

编译运行以上 Swift 范例,输出结果为

$ swift main.swift
ARC 主模块
子模块 topic 数为 4

范例 2

下面的代码演示了无主引用的使用

import Cocoa

class Student
{
    let name: String
    var section: Marks?

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

    deinit
    {
        print("\(name)")
    }
}
class Marks
{
    let marks: Int
    unowned let stname: Student

    init(marks: Int, stname: Student)
    {
        self.marks = marks
        self.stname = stname
    }

    deinit
    {
        print("学生的分数为 \(marks)")
    }
}

var module: Student?
module = Student(name: "ARC")
module!.section = Marks(marks: 98, stname: module!)
module = nil

编译运行以上 Swift 范例,输出结果为

$ swift main.swift
ARC
学生的分数为 98

闭包引起的循环强引用

循环强引用还会发生在将一个闭包赋值给类实例的某个属性,并且这个闭包体中又使用了实例。

这个闭包体中可能访问了实例的某个属性,例如 self.someProperty,或者闭包中调用了实例的某个方法,例如 self.someMethod

这两种情况都导致了闭包 "捕获" self,从而产生了循环强引用

范例

下面的范例演示了当一个闭包引用了 self 后是如何产生一个循环强引用的

范例中定义了一个叫 HTMLElement 的类,用一种简单的模型表示 HTML 中的一个单独的元素

import Cocoa

class HTMLElement
{

    let name: String
    let text: String?

    lazy var asHTML: () -> String = {
        if let text = self.text
        {
            return "<\(self.name)>\(text)</\(self.name)>"

        } else 
        {
            return "<\(self.name) />"
        }
    }

    init(name: String, text: String? = nil)
    {
        self.name = name
        self.text = text
    }

    deinit
    {
        print("\(name) is being deinitialized")
    }

}

// 创建实例并打印信息
var paragraph: HTMLElement? = HTMLElement(name: "p", text: "hello, world")
print(paragraph!.asHTML())

HTMLElement 类产生了类实例和 asHTML 默认值的闭包之间的循环强引用

范例的asHTML 属性持有闭包的强引用

但是,闭包在其闭包体内使用了self(引用了 self.name 和 self.text ),因此闭包捕获了self,这意味着闭包又反过来持有了 HTMLElement 实例的强引用

这样两个对象就产生了循环强引用

解决闭包引起的循环强引用:

在定义闭包时同时定义捕获列表作为闭包的一部分,通过这种方式可以解决闭包和类实例之间的循环强引用

弱引用和无主引用

当闭包和捕获的实例总是互相引用时并且总是同时销毁时,将闭包内的捕获定义为无主引用

当捕获引用有时可能会是 nil 时,将闭包内的捕获定义为弱引用

如果捕获的引用绝对不会置为 nil,应该用无主引用,而不是弱引用

范例

前面的HTMLElement 范例中,无主引用是正确的解决循环强引用的方法

下面这种 HTMLElement 类定义方法可以避免循环强引用

import Cocoa

class HTMLElement
{

    let name: String
    let text: String?

    lazy var asHTML: () -> String = {
        [unowned self] in
        if let text = self.text {
            return "<\(self.name)>\(text)</\(self.name)>"
        } else {
            return "<\(self.name) />"
        }
    }

    init(name: String, text: String? = nil)
    {
        self.name = name
        self.text = text
    }

    deinit
    {
        print("\(name) 被析构")
    }

}

//创建并打印HTMLElement实例
var paragraph: HTMLElement? = HTMLElement(name: "p", text: "hello, world")
print(paragraph!.asHTML())

// HTMLElement实例将会被销毁,并能看到它的析构函数打印出的消息
paragraph = nil

编译运行以上 Swift 范例,输出结果为

$ swift main.swift
<p>hello, world</p>
p 被析构