defer 是干什么的
一句話概括就是 defer block 里的代碼會(huì)在函數(shù) return 之前執(zhí)行惯雳,無論函數(shù)是從哪個(gè)分支 return 的,還是有 throw的石景,還是自然而然走到最后一行的拙吉。
這個(gè)關(guān)鍵字就跟 Java 里的 try-catch-finally 的finally一樣,不管 try catch 走哪個(gè)分支揪荣,它都會(huì)在函數(shù) return 之前執(zhí)行。而且它比 Java 的finally還更強(qiáng)大的一點(diǎn)是仗颈,它可以獨(dú)立于 try catch 存在,
所以它可以成為整理函數(shù)流程的幫手挨决。
在函數(shù) return 之前無論如何都要做的處理,
可以放進(jìn)這個(gè) block 里脖祈,讓代碼看起來更干凈舒服损拢。
swift 文檔上的例子:
var fridgeIsOpen = false
let fridgeContent = ["milk", "eggs", "leftovers"]
func fridgeContains(_ food: String) -> Bool {
fridgeIsOpen = true
defer {
fridgeIsOpen = false
}
let result = fridgeContent.contains(food)
return result
}
fridgeContains("banana")
print(fridgeIsOpen)
例子里執(zhí)行的順序是:
先fridgeIsOpen = true福压,
然后是函數(shù)體正常的流程,
最后在 return 之前執(zhí)行 fridgeIsOpen = false或舞。
簡單的幾個(gè)使用場景
try catch 結(jié)構(gòu)
最典型的場景,我想也是 defer 這個(gè)關(guān)鍵字誕生的主要原因吧:
func foo() {
defer {
print("finally")
}
do {
throw NSError()
print("impossible")
} catch {
print("handle error")
}
}
不管 do block 是否 throw error映凳,有沒有 catch 到,還是 throw 出去了诈豌,都會(huì)保證在整個(gè)函數(shù) return 前執(zhí)行 defer。
在這個(gè)例子里就是:
先 print 出 "handle error"
再 print 出 "finally"矫渔。
do block 里也可以寫 defer:
do {
defer {
print("finally")
}
throw NSError()
print("impossible")
} catch {
print("handle error")
}
那么它執(zhí)行的順序就會(huì)是在 catch block 之前彤蔽,也就是先 print 出 "finally" 再 print 出 "handle error"庙洼。
清理工作、回收資源
跟 swift 文檔舉的例子類似油够,defer一個(gè)很適合的使用場景就是用來做清理工作。比如文件操作:
關(guān)閉文件
func foo() {
let fileDescriptor = open(url.path, O_EVTONLY)
defer {
close(fileDescriptor)
}
// use fileDescriptor...
}
可以避免 哪個(gè)分支忘了寫石咬,或者中間 throw 個(gè) error,導(dǎo)致 fileDescriptor 沒法正常關(guān)閉鬼悠。
dealloc 手動(dòng)分配的空間
func foo() {
let valuePointer = UnsafeMutablePointer.allocate(capacity: 1)
defer {
valuePointer.deallocate(capacity: 1)
}
// use pointer...
}
加/解鎖:下面是 swift 里類似 Objective-C 的 synchronized block 的一種寫法亏娜,可以使用任何一個(gè) NSObject 作 lock
成對調(diào)用 可以用 defer 把它們放在一起。
func foo() {
objc_sync_enter(lock)
defer {
objc_sync_exit(lock)
}
// do something...
}
調(diào) completion block
有時(shí)候一個(gè)函數(shù)分支比較多可能某個(gè)小分支 return 之前就忘了調(diào) completion block镇匀,結(jié)果藏下 bug。
func foo(completion: () -> Void) {
defer {
self.isLoading = false
completion()
}
guard error == nil else { return }
// handle success
}
有時(shí)候 completion 要根據(jù)情況傳不同的參數(shù)汗侵,這時(shí) defer 就不好用了。如果 completion block 被存下來了晰韵,我們還是可以用它來確保執(zhí)行后能釋放:
func foo() {
defer {
self.completion = nil
}
if (succeed) {
self.completion(.success(result))
} else {
self.completion(.error(error))
}
}
調(diào) super 方法
有時(shí)候 override 一個(gè)方法,主要目的是在 super 方法之前做一些準(zhǔn)備工作雪猪,比如 UICollectionViewLayout 的 prepare(forCollectionViewUpdates:)栏尚,那么我們就可以把調(diào)用 super 的部分放在 defer 里:
func override foo() {
defer {
super.foo()
}
// some preparation before super.foo()...
}
一些細(xì)節(jié)
任意 scope 都可以有 defer
雖然大部分的使用場景是在函數(shù)里只恨,不過理論上任何一個(gè) { } 之間都是可以寫 defer 的。比如一個(gè)普通的循環(huán):
var sumOfOdd = 0
for i in 0...10 {
defer {
print("Look! It's \(i)")
}
if i % 2 == 0 {
continue
}
sumOfOdd += i
}
continue 或者 break 都不會(huì)妨礙 defer 的執(zhí)行官觅。甚至 closure 里也可以寫 defer:
{
defer { print("bye!") }
print("hello!")
}
只不過 這樣沒什么意義……
必須執(zhí)行到 defer 才會(huì)觸發(fā)
假設(shè)有這樣一個(gè)問題:一個(gè) scope 里的 defer 能保證一定會(huì)執(zhí)行嗎?
答案 NO……比如下面這個(gè)例子:
func foo() throws {
do {
throw NSError()
print("impossible")
}
defer {
print("finally")
}
}
try?foo()
不會(huì)執(zhí)行 defer休涤,不會(huì) print 任何東西。這個(gè)故事告訴我們功氨,至少要執(zhí)行到 defer 這一行序苏,它才保證后面會(huì)觸發(fā)捷凄。同樣道理提前 return 也是一樣不行的:
func foo() {
guard false else { return }
defer {
print("finally")
}
}
多個(gè) defer
一個(gè) scope 可以有多個(gè) defer,順序是像棧一樣FIFO執(zhí)行的:每遇到一個(gè) defer 就像壓進(jìn)一個(gè)棧里纵势,到 scope 結(jié)束的時(shí)候踱阿,后進(jìn)棧的先執(zhí)行钦铁。如下面的代碼才漆,會(huì)按 1、2醇滥、3超营、4、5阅虫、6 的順序 print 出來。
func foo() {
print("1")
defer {
print("6")
}
print("2")
defer {
print("5")
}
print("3")
defer {
print("4")
}
}