前言
在 Swift 中,是使用引用计数和 copy-on-wirte 的机制来管理内存,通常情况下,开发者只管使用即可,不用太关心内存相关。但对于一些性能敏感的代码来说,开发者可能会希望在语言层面上能够控制减少 retain、release 的调用。比如下面的代码:
swift
func test() {
var x: [Int] = [1, 3, 4]
doStuffUniquely(with: x)
x = []
}
func doStuffUniquely(with value: [Int]) {
var newValue = value
newValue.append(42)
print(newValue)
}
上述代码中,变量 x 存储的是一个整型的数组。然后将其当做参数传递给了 doStuffUniquely()
,接着在 doStuffUniquely()
函数体内部对 value 进行了数据处理。最后,x 被清空。
再将 x 传递给 doStuffUniquely()
的时候应该存在一种机制将 x 的数据所有权也转交过去。因为在 test()
中的后续代码中也并没有对 x 的数据进行使用。
现有的代码会存在一定的性能浪费,因为在 doStuffUniquely()
中的 var newValue = value
执行时会进行数组内容的 copy-on-wirte,而且当 doStuffUniquely()
函数执行完之后, newValue
因为作用域的原因而被释放,从而再将新开辟存储数组元素的空间清空。
基于以上的原因,Swift 5.9 推出了 consume
关键字,使用该关键字可以避免上述代码的性能损耗。
consume 的使用
对于上述代码,修改 doStuffUniquely()
函数调用这一行即可:
scss
doStuffUniquely(with: consume x)
在传递的参数前面添加一个 consume 关键字修饰,这样就等于告诉编译器,x 变量存储的数据所有权,请转交给 doStuffUniquely()
函数的参数。需要注意的是,如果这样写之后,我们在 test()
函数中的后续代码就不可以再对 x 的数据进行修改了,因为我们已经明确的将 x 的数据所有权已经交了出去。
如果,后续再对 x 进行修改的话编译器会报错:
与 let 搭配使用
对于常量来说,也可以通过使用 consume 来进行生命周期的终结。当常量使用 consume 之后,常量就不可用了,因为它的值所有权已被交出,且它不可重新被赋值。
需要说明的是, consume
是对绑定的操作,而不是对值进行操作,所以如果将 x 赋值给另一个常量 other
,当 x consume
之后,x 已不能使用,而 other
是不受影响的:
scss
func useX(_ x: Int) -> () {}
func f() {
let x = 10
useX(x)
let other = x
_ = consume x
useX(other)
}
在 _ = consume x
执行完之后, x 的生命周期已结束。而 other 是可以正常使用的,所以 useX(other)
并没有什么问题。
与 var 搭配使用
与 var 搭配使用则与 let 使用略有不同,当 consume 修饰的变量重新赋值之后,是可以正常使用的:
ini
struct User {
var name: String
}
func test() {
var user = User(name: "jack")
_ = consume user
// user.name = "" // 编译报错
user = User(name: "rose")
print(user.name)
}
上面的代码我们可以看到,如果在 user consume
之后,直接对 user 进行 name 的赋值是不行的,原因就是 user 的数据所有权已经没了。
而对 user 进行重新赋值则是没问题的,后续也可以正常使用。