ARC模式下的循環引用引起內存泄漏
自從iOS 5時代自動引用計數(Automatic Reference Counting)技術發布,Cocoa工程師們才扔下了內存管理的包袱,從此在Objective-C修行道路上的一座大山被削平。然而,即使ARC很強大,我們日常搬磚時同樣是有內存泄漏風險的,今天我就跟大家聊聊這些你可能還沒有注意到的坑。
測試原理
我們知道ARC模式下,NSObject的MRC相關方法都不可以使用了,但dealloc方法如果實現了,同樣還是會調用的,只是不允許在dealloc方法中調用[super dealloc],所以我們在dealloc方法中加入log信息就可以跟蹤到我們的實例是否釋放。
容易忽視的引用循環
我們知道引用計數內存管理的設計理念,就是根據實例的計數值來決定是否釋放實例內存空間。
例如我們的ViewController擁有一個block類型的property
- @property (nonatomic, strong) void (^ testBlock)(void);
我們在viewDidLoad中加入如下代碼
- [self setTestBlock:^{
- self.title = @"測試";
- }];
這個代碼從表面上看沒有什么問題,但編譯器會給出warning,
Catering 'self' strongly in this block is likely to lead a retain cycle
翻譯過來意思是在block中使用self指針,可能會引起一個引用循環,導致self無法釋放。
什么是引用循環(retain cycle)
假設我們有兩個實例A和B,B是A的一個strong型的property,則B的引用計數是1,當A的需要釋放的時候,A則會調用[B release]來釋放B,B的引用計數則減為0,釋放。
可如果這時候將B的一個strong型property指向A,則A與B互相為強引用,問題就來了。因為B強引用A,A的引用計數永遠不會減為0,當A原本的強引用對象被釋放以后,A和B成為了一個相互引用的孤島,永遠不會被釋放了,這就會引起內存泄漏。
在上面的例子中,就是一種非常普遍的引用循環情況,加入如上代碼的VC在dismiss或者pop以后,并不會執行dealloc方法,證明內存泄漏了。而引起泄漏的原因就是在作為self的property的block中,使用self指針導致self被block強引用,形成引用循環。
如何解決引用循環問題
在編譯器提示上面的warning的時候一定不要忽視,正確的解決辦法如下:
- __unsafe_unretained Demo1ViewController * weakSelf = self;
- [self setTestBlock:^{
- weakSelf.title = @"測試";
- }];
或者使用__weak也可以,原理也很簡單,就是聲明一個弱引用對象在block中替代self,這樣在我們測試中,下面代碼就能正常輸出log,標志著VC被正確釋放。
- - (void)dealloc
- {
- NSLog(@"%s",__func__);
- }
2016-09-07 13:17:38.879 ReactiveCocoaDemo[7473:3432323] -[Demo1ViewController dealloc]
其他會引起引用循環的狀況
NSTimer
NSTimer在VC釋放前,一定要調用[timer invalidate],不調用的后果就是NSTimer無法釋放其target,如果target正好是self(VC本身),則引用循環。
這里要補充一點,引用循環不是只能有兩個對象,三個四個更多都是可以的,甚至環數也不一定只有一個,所以要養成良好的代碼習慣,在NSTimer停用前調用invalidate方法。
WKUserContentController
這個類一般會在使用WKWebView的時候配套使用,如果你發現項目中調用了addScriptMessageHandler方法,就要注意了,檢查有沒有在VC釋放前對稱調用removeScriptMessageHandlerForName方法,如果沒有則引起引用循環。
調用方法如下:
- [self.wkWebView.configuration.userContentController removeScriptMessageHandlerForName:@"qdpay"];
注意WKUserContentController和WKWebView中還有一個WKWebViewConfiguration。
引用大循環
就像前面說的,引用循環可能是一個大循環。我遇到過一種情況,就是給UITableViewCell設置block屬性響應事件,在block中強引用了self,導致self->tableView->cell->self形成循環。
改善block寫法避免強引用self
如果要從根本改變這種易發的錯誤,要從寫法開始改變,開始避免。將上面的代碼改寫如下:
- @property (nonatomic, strong) void (^ testBlock)(__kindof UIViewController* sender);
- [self setTestBlock:^(__kindof UIViewController * vc) {
- vc.title = @"123";
- }];
- self.testBlock(self);
將self作為參數傳入block即可避免強引用,從邏輯角度來看,代碼更健壯。
結束
上面列舉的幾個引用循環引起的內存泄漏,編譯器是沒有任何提示的,并且也不影響App運行,不會crash,但作為嚴謹的程序猿,我們不能容忍這種的小泄漏,雖然不影響大局,但積少成多終將影響系統的運行速度。