剖析 Swift 的弱引用与集合类型：完美搭配

IOS

2023-09-13 07:12:48

深入浅出：认识弱引用

弱引用本质上是一种指向对象的间接引用，当对象不再被任何强引用持有时，弱引用会自动将其指向的对象置为 nil。这与强引用不同，强引用会将对象牢牢“抓住”，即使对象不再被使用也不会释放。

为何将弱引用与集合类型联手？

集合类型，如数组、集合和词典，在 Swift 中被广泛使用，它们提供了强大的数据结构和高效的访问方式。然而，当集合类型包含对对象的强引用时，可能会导致循环引用，从而造成内存泄露和性能问题。

通过将弱引用与集合类型结合，可以有效避免循环引用。弱引用可以确保集合类型中的对象在不再被使用时自动释放，从而防止内存泄露和性能下降。

实战演示：弱引用与集合类型的完美合作

为了更好地理解弱引用与集合类型的结合使用，让我们通过几个示例代码来进行演示：

示例一：数组中的弱引用

class Person {
    let name: String
    weak var friend: Person?
    
    init(name: String) {
        self.name = name
    }
    
    deinit {
        print("\(name) is being deallocated.")
    }
}

var people = [Person(name: "Alice"), Person(name: "Bob")]

people[0].friend = people[1]
people[1].friend = people[0]

people.removeAll()

在这个例子中，我们定义了一个 Person 类，并使用弱引用 friend 来存储对另一个 Person 对象的引用。当我们使用 people.removeAll() 清除数组时，由于 friend 是弱引用，因此 Person 对象不会被强引用持有，从而触发其 deallocation 并释放内存。

示例二：集合中的弱引用

class Book {
    let title: String
    weak var author: Author?
    
    init(title: String) {
        self.title = title
    }
    
    deinit {
        print("\(title) is being deallocated.")
    }
}

class Author {
    let name: String
    var books = Set<Book>()
    
    init(name: String) {
        self.name = name
    }
    
    deinit {
        print("\(name) is being deallocated.")
    }
}

var books = Set<Book>()
var authors = Set<Author>()

let book1 = Book(title: "The Great Gatsby")
let book2 = Book(title: "To Kill a Mockingbird")

let author1 = Author(name: "F. Scott Fitzgerald")
let author2 = Author(name: "Harper Lee")

book1.author = author1
book2.author = author2

authors.insert(author1)
authors.insert(author2)
books.insert(book1)
books.insert(book2)

books.removeAll()
authors.removeAll()

在这个示例中，我们定义了一个 Book 类和一个 Author 类，并使用弱引用 author 来存储对 Author 对象的引用，以及使用集合 books 来存储对 Book 对象的引用。当我们使用 books.removeAll() 和 authors.removeAll() 清除集合时，由于 author 和 books 是弱引用，因此 Author 和 Book 对象不会被强引用持有，从而触发其 deallocation 并释放内存。