浅谈Swift的内存管理

原文博客地址: 浅谈Swift的内存管理
2020年的第一篇博客, 算是2020年开了个好头, 但愿可以继续坚持下去, 继续记录分享更多更高质量的文章
今年期待已久的Swift5.0稳定版就已经发布了, 感兴趣的小伙伴可看个人这篇博客:Swift 5.0新特性更新
这篇博客可主要分享Swift的内存管理的相关介绍和剖析, 测试环境: Xcode 11.2.1, Swift 5.1.2

内存管理

和OC同样, 在Swift中也是采用基于引用计数的ARC内存管理方案(针对堆空间的内存管理)
在Swift的ARC中有三种引用
- 强引用(strong reference)：默认状况下，代码中涉及到的引用都是强引用
- 弱引用(weak reference)：经过weak定义弱引用
- 无主引用(unowned reference)：经过unowned定义无主引用

weak

弱引用(weak reference)：经过weak定义弱引用
- 必须是可选类型的var，由于实例销毁后，ARC会自动将弱引用设置为nil
- ARC自动给弱引用设置nil时，不会触发属性观察
在介绍weak弱引用以前, 先看一下下面一段代码

class Animal {
    deinit {
        print("Animal deinit")
    }
}

func test() {
    let animal = Animal()
}

print("will deinit")
test()
print("did deinit")
复制代码

上面这段代码中在test函数调用结束以后, 该做用的内存就会被回收, animal对象天然就会被销毁, 毫无疑问上面的输出结果应该是html

will deinit
Animal deinit
did deinit
复制代码

一样下面这段代码, 一样也是在a1对象被置为nil的时候内存会被回收, 对象就会被销毁swift

var a1: Animal? = Animal()
print("will deinit")
a1 = nil
print("did deinit")
复制代码

下面是一个被weak修饰的弱引用对象,
咱们都知道, 被weak修饰的弱引用对象, 在对象销毁的时候, 会被自动置为nil
因此被weak修饰的弱引用对象必须是可选类型的var, 两个条件缺一不可

weak var a2: Animal? = Animal()

// 如下两种方式都会报错的
weak var a2: Animal = Animal()
weak let a2: Animal? = Animal()
复制代码

unowned

无主引用(unowned reference)：经过unowned定义无主引用
不会产生强引用，实例销毁后仍然存储着实例的内存地址(相似于OC中的unsafe_unretained）
试图在实例销毁后访问无主引用，会产生运行时错误（以下野指针)

Fatal error: Attempted to read an unowned reference but object 0x0 was already deallocate
复制代码

另外须要注意的是, weak、unowned只能用在类实例上面, 以下所示安全

// 该协议表示只能被类遵照, AnyObject表明全部的类实例
protocol Liveable: AnyObject {}
class Person {}

weak var p0: Person?
weak var p1: AnyObject?
// 全部能遵循Liveable协议的确定都是类
weak var p2: Liveable?

unowned var p10: Person?
unowned var p11: AnyObject?
unowned var p12: Liveable?
复制代码

循环引用

weak、unowned都能解决循环引用的问题，unowned要比weak少一些性能消耗
在生命周期中可能会变为nil的使用weak
初始化赋值后不再会变为nil的使用unowne
说道循环引用就天然想到了闭包

闭包的循环引用

闭包表达式默认会对用到的外层对象产生额外的强引用（对外层对象进行了retain操做）, 看一下下面的代码中deinit会被调用吗?bash

class Person {
    var fn: (() -> ())?
    func run() { print("run") }
    deinit { print("deinit") }
}

func test() {
    let p = Person()
    p.fn = { 
        p.run()
    }
}

test()
复制代码

上面代码中, p对象强引用着fn闭包, fn闭包也强引用着p对象, 天然就形成了循环引用问题
最后没有任何输出结果, 咱们看一下上述代码的汇编执行过程

从上面汇编代码能够看出, 整个过程经历了
一次init引用计数为----1
一次retain引用计数会加(1), 结果为----2
一次release引用计数会减(1), 结果为----1
那么最后的引用计数就是1, 因此p对象确定没有被释放
下面是使用解决循环引用的状况
- 在闭包表达式的捕获列表里, 声明weak或unowned引用，用以解决循环引用问题

// 使用weak
func test() {
    let p = Person()
    p.fn = { [weak p] in
        p?.run()
    }
}

// 使用unowned
func test() {
    let p = Person()
    p.fn = { [unowned p] in
        p.run()
    }
}
复制代码

上述两种方式均可以解决循环引用的问题, 运行后就发现Person对象调用了deinit
这里咱们再看一下汇编代码以下, 从下面汇编代码中能够很明显看到, 引用计数最后为0, 对象被释放

另外下面这段代码实际上是等价的微信

func test() {
    let p = Person()
    p.fn = { [unowned p] in
        p.run()
    }
}

// 和上面等价代码
func test() {
    let p = Person()
    p.fn = { [unowned ownedP = p, weak weakP = p] in
        ownedP.run()
        // weakP?.run()
    }
}
复制代码

特别注意点, 这里要区分捕获列表和参数列表, 下面看看fn有参数的状况下markdown

class Person {
    var fn: ((Int) -> ())?
    func run() { print("run") }
    deinit { print("deinit") }
}

func test() {
    let p = Person()
    p.fn = {
        (num) in
        print("num = \(num)")
    }
}
复制代码

那么闭包的参数列表和捕获列表同事存在的状况以下代码所示闭包

func test() {
    let p = Person()
    p.fn = {
        [weak p](num) in
        print("num = \(num)")
        p?.run()
    }
}
复制代码

self的循环引用

若是想在引用闭包的同时引用self, 这个闭包必须是lazy的
由于实例在初始化完毕以后才能引用self

class Person {
    lazy var fn: (() -> ()) = {
        self.run()
    }
    func run() { print("run") }
    deinit { print("deinit") }
}

func test() {
    let p = Person()
    p.fn()
}

test()
复制代码

上面代码中若是fn闭包去掉lazy, 编译器会直接报错
在Swift中, 为了保证初始化的安全, 设定了两段式初始化, 在全部的存储属性被初始化完成以后, 初始化器才可以使用self
并且在上述fn闭包中, 若是fn内部用到了实例成员(属性和方法), 则编译器会强制要求明确写出self
lazy既保证只有在使用的时候才会被初始化一次
可是上述代码一样存在循环引用的问题, Person对象强引用着fn闭包, fn闭包也强引用着self
一样使用weak和unowned解决循环引用的问题

// weak解决循环引用
lazy var fn: (() -> ()) = {
    [weak self] in
    self?.run()
}

// unowned解决循环引用
lazy var fn: (() -> ()) = {
    [unowned self] in
    self.run()
}
复制代码

另外再看看下面这种状况, 是都存在循环引用的问题函数

class Student {
    var age: Int = 2
    lazy var getAge: Int = {
        self.age
    }()
    deinit { print("deinit") }
}

func test() {
    let p = Student()
    print(p.getAge)
}

test()

/* 输出结果 2 deinit */
复制代码

经过输出结果看一看出调用了deinit, 说明对象最终被释放, 并未出现循环引用的问题, 下面比较一下oop

// 存在循环引用
class Person {
    lazy var fn: (() -> ()) = {
        self.run()
    }
    func run() { print("run") }
    deinit { print("deinit") }
}

// 不存在循环引用
class Student {
    var age: Int = 2
    lazy var getAge: Int = {
        self.age
    }()
    deinit { print("deinit") }
}
复制代码

上述两种写法的区别, 本质上说
Person对象中的fn闭包属于闭包赋值
Student对象那个中的getAge属于闭包调用(相似函数调用)
至关于在在Student对象调用getAge结束以后, 做用域内的变量就会被释放

// getAge也能够写成以下形式
lazy var getAge: Int = {
    return self.age
}()

// 也能够理解为
lazy var getAge: Int = self.age
复制代码

内存访问冲突

在Swift中的内存访问冲突主要在两个访问知足下列条件时发生性能

至少一个是写入操做
它们访问的是同一块内存
它们的访问时间重叠(好比在同一个函数内)
对比看看如下两个函数操做

// 不存在内存访问冲突
var number = 1
func plus(_ num: inout Int) -> Int {
    return num + 1
}
number = plus(&number)

// 存在内存访问冲突
var step = 1
func increment(_ num: inout Int) {
    num += step
}
increment(&step)
复制代码

上面第二部分代码就是同时对step变量执行读写操做, 运行时会报出以下错误

Simultaneous accesses to 0x100002028, but modification requires exclusive access.
复制代码

再看下面对于结构体和元组的使用, 这里先定义一个全局函数和一个结构体

// 改变两个传入参数的值, 读取并修改传入参数的值
func balance(_ x: inout Int, _ y: inout Int) {
    let sum = x + y
    x = sum / 2
    y = sum - x
}

// 定义Player结构体
struct Player {
    var name: String
    var health: Int
    var energy: Int
    mutating func shareHealth(with teammate: inout Player) {
        balance(&teammate.health, &health)
    }
}
复制代码

再看下面的使用示例, 二者都会有一个内存访问冲突的错误

// 这里读写的是同一个maria
var maria = Player(name: "Maria", health: 50, energy: 10)
balance(&maria.health, &maria.energy)

// 这里读写的是同一个tuple
var tuple = (health: 10, energy: 20)
balance(&tuple.health, &tuple.energy)
复制代码

可是有时候的确会有上面这种访问同一块内存的需求, 若是下面的条件知足, 就说明重叠访问结构体的属性是安全的

访问的是实例存储属性, 不是计算属性或者类属性
结构体是局部变量而非全局变量
结构体要么没有被闭包捕获要么只被非逃逸闭包捕获

// 这里能够在局部做用域内定义成局部变量, 就不会有问题了
func test() {
    var maria = Player(name: "Maria", health: 50, energy: 10)
    var tuple = (health: 10, energy: 20)
    balance(&tuple.health, &tuple.energy)
    balance(&maria.health, &maria.energy)
}
复制代码

指针

class Person {}
var person = Person()
复制代码

在Swift中class声明的类(Person)是引用类型, 初始化的person对象其本质上就是一个指针变量
而person里面存储的就是这个指针变量的地址值, 也就能够根据这个地址值去访问被分配的内存空间
指针在某种意义上被定性为不安全的, 举个例子:
- 当前指针变量的地址值对应的空间只有32个字节, 但有可能访问的是超过32个字节的空间, 这样就可能会出问题的

指针分类

在Swift中也有专门的指针类型，这些都被定性为Unsafe（不安全的），常见的有如下4种类型

UnsafePointer<Pointee>, 相似于C语言中的const Pointee *, 只能访问内存不能修改内存, 这里的Pointee是指泛型
UnsafeMutablePointer<Pointee>相似于C语言中的Pointee *, 能够访问和修改内存, 这里的Pointee是指泛型
UnsafeRawPointer相似于const void *, 不支持泛型
UnsafeMutableRawPointer相似于void, 不支持泛型

下面看一下具体的使用示例

var age = 10
func sum1(_ ptr: UnsafeMutablePointer<Int>) {
    // 经过访问pointee属性, 获取ptr指针的内存地址所存储的值
    // UnsafeMutablePointer的pointee属性是可读可写的
    ptr.pointee += 10
}
func sum2(_ ptr: UnsafePointer<Int>) {
    // UnsafePointer的pointee属性是只读的
    // ptr.pointee += 10
    print(ptr.pointee)
}
func sum3(_ num: inout Int) {
    // 
    num += 10
}

// 和inout输入输出参数同样接受变量的地址值
sum1(&age)
sum2(&age)
sum3(&age)
print(age)


func sum4(_ ptr: UnsafeMutableRawPointer) {
    // 可读可写, 取值
    print("age = ", ptr.load(as: Int.self))
    // 可读可写, 赋值
    ptr.storeBytes(of: 50, as: Int.self)
}
func sum5(_ ptr: UnsafeRawPointer) {
    // 只读, 取值
    print("age = ", ptr.load(as: Int.self))
}

sum4(&age)
sum5(&age)
复制代码

得到变量的指针

Swift中有能够直接获取变量的指针的方法

// 获取可变的变量指针, value参数接受变量地址
@inlinable public func withUnsafeMutablePointer<T, Result>(to value: inout T, _ body: (UnsafeMutablePointer<T>) throws -> Result) rethrows -> Result
// 获取不可变的变量指针, value参数接受变量
@inlinable public func withUnsafePointer<T, Result>(to value: T, _ body: (UnsafePointer<T>) throws -> Result) rethrows -> Result
// 获取不可变的变量指针, value参数接受变量地址
@inlinable public func withUnsafePointer<T, Result>(to value: inout T, _ body: (UnsafePointer<T>) throws -> Result) rethrows -> Result
复制代码

上述方法中返回值默认是变量的指针地址, 也能够是其余的数据类型, 主要取决于body闭包的返回值, 返回值类型由闭包中的Result泛型决定

var age = 10
var ptr1 = withUnsafeMutablePointer(to: &age) { $0 }   // UnsafeMutablePointer<Int>
var ptr2 = withUnsafePointer(to: &age) { $0 }          // UnsafePointer<Int>
ptr1.pointee = 22
print(ptr2.pointee) // 22
print(ptr2)         // 0x0000000100008310

var ptr3 = withUnsafeMutablePointer(to: &age) { UnsafeMutableRawPointer($0) }   // UnsafeMutableRawPointer
var ptr4 = withUnsafePointer(to: &age) { UnsafeRawPointer($0) }                 // UnsafeRawPointer
// as参数是须要存储什么类型的数据
ptr3.storeBytes(of: 33, as: Int.self)
print(ptr4.load(as: Int.self)) // 33
print(ptr4)         // 0x0000000100008310
复制代码

建立指针

以前获取到的指针都是根据已经存在的内存获取的
这里就看看从新分配一块内存指向堆空间

malloc

Swift提供了malloc直接分配内存建立指针的方式

// 根据须要分配的内存大小建立一个指针
public func malloc(_ __size: Int) -> UnsafeMutableRawPointer!
// 释放内存
public func free(_: UnsafeMutableRawPointer!)


// 下面这两个函数, 是赋值和取值的函数, 以前简单介绍过
// 参数一: 须要存储的值
// 参数二: 偏移量, 从第几个字节开始存储, 默认从第一个
// 参数三: 须要存储的值的类型
@inlinable public func storeBytes<T>(of value: T, toByteOffset offset: Int = 0, as: T.Type)

// 参数一: 偏移量, 从第几个字节开始存储, 默认从第一个
// 参数二: 须要存储的值的类型
@inlinable public func load<T>(fromByteOffset offset: Int = 0, as type: T.Type) -> T
复制代码

代码示例以下

// 建立指针
var ptr = malloc(16)
// 存储值
ptr?.storeBytes(of: 10, as: Int.self)
// 这里toByteOffset参数若是传0, 就会覆盖前8个字节的数据
ptr?.storeBytes(of: 12, toByteOffset: 8, as: Int.self)
// 取值
print(ptr?.load(as: Int.self) ?? 0)
print(ptr?.load(fromByteOffset: 8, as: Int.self) ?? 0)
// 销毁, 释放内存
free(ptr)
复制代码

allocate

使用allocate方式建立指针, 代码示例以下

// byteCount: 须要申请的字节数, alignment: 对其字节数
var ptr2 = UnsafeMutableRawPointer.allocate(byteCount: 16, alignment: 1)
// 存储
ptr2.storeBytes(of: 9, as: Int.self)

// 根据字节偏移存储
// 这里的ptr3是ptr2偏移8个字节的新的指针地址
var ptr3 = ptr2.advanced(by: 8)  // UnsafeMutableRawPointer
ptr3.storeBytes(of: 12, as: Int.self)

// 上面这种方式等价于
ptr2.storeBytes(of: 12, toByteOffset: 8, as: Int.self)

// 取值一样
print(ptr2.load(as: Int.self))
// 下面这两种取值方式也是同样的
print(ptr2.advanced(by: 8).load(as: Int.self))
print(ptr2.load(fromByteOffset: 8, as: Int.self))

// 释放内存
ptr2.deallocate()
复制代码

这里须要注意的是只有UnsafeMutableRawPointer才有allocate分配方法, UnsafeRawPointer是没有这个方法的, 下面说到的UnsafeMutablePointer<T>类型也是, UnsafePointer<T>没有allocate分配方法

// capacity: 容量, 便可以存储3个Int类型的数据, 也就是24个字节
var ptr = UnsafeMutablePointer<Int>.allocate(capacity: 3)
// 初始化内存, 用10初始化钱8个字节
ptr.initialize(to: 10)
// 用10初始化前两个容量的内存, 即16个字节
ptr.initialize(repeating: 10, count: 2)
// 使用successor获取下一个存储位, 也就是下一个Int的位置
var ptr1 = ptr.successor()  // UnsafeMutablePointer<Int>
ptr1.initialize(to: 20)
// 存储第三个Int值
ptr.successor().successor().initialize(to: 30)

// 取值的两种方式
print(ptr.pointee)        // 第一个值
print((ptr + 1).pointee)  // 第二个值
print((ptr + 2).pointee)  // 第三个值

// 下面这种方式和上面等价
print(ptr[0])
print(ptr[1])
print(ptr[2])

// 前面若是使用了initialize, 则必须调用反初始化
// 并且count要和上面allocate(capacity: 3)的capacity一致, 不然会形成内存泄露的问题
ptr.deinitialize(count: 3)
ptr.deallocate()
复制代码

指针之间的转换

前面提到过Swift中的指针类型有四种

UnsafePointer<Pointee>相似于const Pointee *
UnsafeMutablePointer<Pointee>相似于Pointee *
UnsafeRawPointer相似于const void *
UnsafeMutableRawPointer相似于void *
那么上面的类型, 可否经过其中的一种建立另一种指针呢, 下面咱们来看一下

init

UnsafeMutableRawPointer中有一个初始化方法能够根据UnsafeMutablePointer建立自身

public init<T>(_ other: UnsafeMutablePointer<T>)

var ptr = UnsafeMutablePointer<Int>.allocate(capacity: 3)
var ptr1 = UnsafeMutableRawPointer(ptr)
复制代码

assumingMemoryBound

反过来, UnsafeMutableRawPointer也提供了一个方法用于建立UnsafePointer

public func assumingMemoryBound<T>(to: T.Type) -> UnsafePointer<T>

var ptr = UnsafeMutableRawPointer.allocate(byteCount: 16, alignment: 1)
var ptr1 = ptr.assumingMemoryBound(to: Int.self)
// 初始化前8个字节
ptr1.pointee = 11
// 初始化后8个字节
// 特别注意, 这里的(ptr + 8)是指ptr向后偏移8个字节, 要和以前的区分开
(ptr + 8).assumingMemoryBound(to: Int.self).pointee = 12

ptr.deallocate()
复制代码

unsafeBitCast

unsafeBitCast是忽略数据类型的强制转换，不会由于数据类型的变化而改变原来的内存数

// 把第一个参数类型转成第二个参数类型
@inlinable public func unsafeBitCast<T, U>(_ x: T, to type: U.Type) -> U


var ptr = UnsafeMutableRawPointer.allocate(byteCount: 16, alignment: 1)
unsafeBitCast(ptr, to: UnsafeMutablePointer<Int>.self).pointee = 13
// 注意, 这里的(ptr + 8)是指ptr向后偏移8个字节, 要和以前的区分开
unsafeBitCast(ptr + 8, to: UnsafeMutablePointer<Double>.self).pointee = 14.23

ptr.deallocate()
复制代码

欢迎您扫一扫下面的微信公众号，订阅个人博客！