在使用 Block
時,除了使用 __weak
修飾符避免迴圈引用外,還有一點經常容易忘記。蘋果把它稱為:“Strong-Weak Dance”。
問題來源
這是一種 強引用 –> 弱引用 –> 強引用 的變換過程。在弄明白為什麼要如此大費周章之前,我們首先來看看一般的寫法會有什麼問題。
1 2 3 4 |
__weak MyViewController *wself = self; self.completionHandler = ^(NSInteger result) { [wself.property removeObserver: wself forKeyPath:@"pathName"]; }; |
這種寫法可以避免迴圈引用,但是我們要考慮這樣的問題:
假設 block
被放在子執行緒中執行,而且執行過程中 self
在主執行緒被釋放了。由於 wself
是一個弱引用,因此會自動變為 nil
。而在 KVO 中,這會導致崩潰。
Strong-Weak Dance
解決以上問題的方法很簡單,新增一行程式碼即可:
1 2 3 4 5 |
__weak MyViewController *wself = self; self.completionHandler = ^(NSInteger result) { __strong __typeof(wself) sself = wself; // 強引用一次 [sself.property removeObserver: sself forKeyPath:@"pathName"]; }; |
這樣一來,self
所指向物件的引用計數變成 2,即使主執行緒中的 self
因為超出作用於而釋放,物件的引用計數依然為 1,避免了物件的銷燬。
思考
在和小夥伴的討論過程中,他提出了幾個問題。雖然都不難,但是有利於把各種知識融會貫通起來。
- Q:下面這行程式碼,將一個弱引用的指標賦值給強引用的指標,可以起到強引用效果麼?
1__strong __typeof(wself) sself = wself;
A:會的。引用計數描述的是物件而不是指標。這句話的意思是:sself 強引用 wself 指向的那個物件
因此物件的引用計數會增加一個。
- Q:
block
內部定義了sself
,會不會因此強引用了sself
?A:不會。
block
只有截獲外部變數時,才會引用它。如果是內部新建一個,則沒有任何問題。 - Q:如果在
block
內部沒有強引用,而是通過if
判斷,是不是也可以,比如這樣寫:123456__weak MyViewController *wself = self;wself.completionHandler = ^(NSInteger result) {if (wself) { // 只有當 wself 不為 nil 時,才執行以下程式碼[wself.property removeObserver: wself forKeyPath:@"pathName"];}};A:不可以!考慮到多執行緒執行,也許在判斷的時候,
self
還沒釋放,但是執行self
裡面的程式碼時,就剛好釋放了。 - Q:那按照這個說法,
block
內部強引用也沒用啊。也許block
執行以前,self
就釋放了。A:有用!如果在
block
執行以前,self
就釋放了,那麼block
的引用計數降為 0,所以自己就會被釋放。這樣它根本就不會被執行。另外,如果執行一個為nil
的閉包會導致崩潰。 - Q:如果在執行
block
的過程中,block
被釋放了怎麼辦?A:簡單來說,
block
還會繼續執行,但是它捕獲的指標會具有不確定的值,詳細內容請參考這篇文章
@strongify 和 @weakify
這是 ReactiveCocoa 中定義的一個巨集。一般可以這樣使用:
1 2 3 4 5 |
@weakify(self); self.completionHandler = ^(NSInteger result) { @strongify(self); [self.property removeObserver: sself forKeyPath:@"pathName"]; }; |
本文並非分析它們的實現原理,所以就簡單解釋兩點:
- 這裡的“@”沒有任何用處,僅表示強調,這個巨集實際上包含了一個空的
AutoreleasePool
,這也就是為什麼一定要加上“@”。 - 它的原理還是和之前一樣,生成了一段形如
__weak MyViewController *wself = self;
這種格式的程式碼:
12#define rac_strongify_(INDEX, VAR) \__strong __typeof__(VAR) VAR = metamacro_concat(VAR, _weak_);
Swift 中的情況
感謝 @Cyrus_dev 的提醒,在 Swift 中也有 Strong-Weak Dance 的概念。最簡單的方法就是直接沿用 OC 的思路:
1 2 3 4 5 |
self.completionHandler = { [weak self] in if let strongSelf = self { /// .... } }; |
這種寫法的缺點在於,我們不能寫 if let self = self
,因此需要重新定義一個變數 strongSelf
,命名方式顯得不夠優雅。
除此以外還可以使用 Swift 標準庫提供的函式 withExtendedLifetime
:
1 2 3 4 5 |
self.completionHandler = { [weak self] in withExtendedLifetime(self) { /// ... } }; |
這種寫法的缺點在於,self
依然是可選型別的,還需要把它解封后才能使用。
最後,還有一種解決方案是自定義 withExtendedLifetime
函式:
1 2 3 4 5 6 7 |
extension Optional { func withExtendedLifetime(body: T -> Void) { if let strongSelf = self { body(strongSelf) } } } |
至於這種寫法是否更加優雅,就見仁見智了:
1 2 3 4 5 |
self.completionHandler = { [weak self] in self.withExtendedLifetime { /// 這裡用 $0 表示 self } }; |