我們做開發(fā)的都知道,內(nèi)存管理極為重要,創(chuàng)建的內(nèi)存如果不能及時釋放,積累到一定程度就會讓程序崩潰。RxSwift作為優(yōu)秀的開源框架對于內(nèi)存管理自然也是極為嚴(yán)格。接下來,我們探究一下普通使用的時候,什么情況下會導(dǎo)致RxSwift的內(nèi)存泄露。
Rx對于內(nèi)存管理提供了一套非常方便的檢測機制,只需要在pod里的tag的end之后里引入如下代碼,并pod update
post_install do |installer|
installer.pods_project.targets.each do |target|
if target.name == 'RxSwift'
target.build_configurations.each do |config|
if config.name == 'Debug'
config.build_settings['OTHER_SWIFT_FLAGS'] ||= ['-D', 'TRACE_RESOURCES']
end
end
end
end
end
接下來我們設(shè)計如下場景
我們在vc1里統(tǒng)計引用計數(shù),在vc2里執(zhí)行訂閱操作。當(dāng)我們訂閱執(zhí)行了之后,pop到vc1,觀察引用計數(shù)的變化。核心代碼如下:
我們在vc1里加入如下代碼:
override func viewWillAppear(_ animated: Bool) {
super.viewWillAppear(animated)
print("*****vc1即將出現(xiàn),此時引用計數(shù)為: \(RxSwift.Resources.total)")
}
override func viewDidDisappear(_ animated: Bool) {
super.viewDidDisappear(animated)
print("*****vc1已經(jīng)消失,此時引用計數(shù)為: \(RxSwift.Resources.total)")
}
override func touchesBegan(_ touches: Set<UITouch>, with event: UIEvent?) {
print("*****vc1的push即將執(zhí)行,此時引用計數(shù)為: \(RxSwift.Resources.total)")
let vc = vc2()
self.navigationController?.pushViewController(vc, animated: true)
}
操作一
首先在vc2的viewDidLoad里執(zhí)行如下操作
Observable<Any>.create {(anyObserver) -> Disposable in
anyObserver.onNext("Hello word")
return Disposables.create()
}
.subscribe(onNext: { (item) in
print("訂閱到:\(item)")
})
.disposed(by: self.disposeBag)
}
并打印析構(gòu)函數(shù)
deinit {
print("vc2執(zhí)行了析構(gòu)函數(shù)")
}
我們能看得出來,這是一個非常普通的訂閱操作,訂閱里沒有任何self的引用。
我們執(zhí)行下,看下效果
*****vc1即將出現(xiàn),此時引用計數(shù)為: 0
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 0
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 8
*****vc1即將出現(xiàn),此時引用計數(shù)為: 8
vc2執(zhí)行了析構(gòu)函數(shù)
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 0
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 8
*****vc1即將出現(xiàn),此時引用計數(shù)為: 8
vc2執(zhí)行了析構(gòu)函數(shù)
經(jīng)過反復(fù)push與pop我們能看的出來,此時引用計數(shù)非常正常,并未造成內(nèi)存泄露。
操作二
日常使用中,我們不大可能如此簡單的使用訂閱功能,我們經(jīng)常會在創(chuàng)建序列的時候使用self。現(xiàn)在,假如我們僅僅是最簡單的打印一下self(復(fù)雜使用效果一樣),代碼如下:
Observable<Any>.create {(anyObserver) -> Disposable in
print(self)
anyObserver.onNext("Hello word")
return Disposables.create()
}
.subscribe(onNext: { (item) in
print("訂閱到:\(item)")
})
.disposed(by: self.disposeBag)
}
打印如下
<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7f8adf518340>
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 8
*****vc1即將出現(xiàn),此時引用計數(shù)為: 8
vc2執(zhí)行了析構(gòu)函數(shù)
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 0
<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7f8ae205b130>
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 8
*****vc1即將出現(xiàn),此時引用計數(shù)為: 8
vc2執(zhí)行了析構(gòu)函數(shù)
此時我們發(fā)現(xiàn),依然沒有問題。
操作三
如果我們只在訂閱里使用self呢
代碼如下
Observable<Any>.create {(anyObserver) -> Disposable in
anyObserver.onNext("Hello word")
return Disposables.create()
}
.subscribe(onNext: { (item) in
print(self)
print("訂閱到:\(item)")
})
.disposed(by: self.disposeBag)
}
打印結(jié)果:
*****vc1即將出現(xiàn),此時引用計數(shù)為: 0
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 0
<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7fed7bf17960>
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 8
*****vc1即將出現(xiàn),此時引用計數(shù)為: 8
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 8
<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7fed7be32310>
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 16
*****vc1即將出現(xiàn),此時引用計數(shù)為: 16
問題出現(xiàn)了。此時vc2并沒有執(zhí)行析構(gòu),Rx的引用計數(shù)也沒用釋放。
解決辦法很簡單對self弱引用就可以了如下:
Observable<Any>.create {(anyObserver) -> Disposable in
anyObserver.onNext("Hello word")
return Disposables.create()
}
.subscribe(onNext: {[weak self](item) in
print(self)
print("訂閱到:\(item)")
})
.disposed(by: self.disposeBag)
}
原理:
整個過程的持有關(guān)系如下:

圖中有兩個引用圈:
圈1:這個圈源碼里已經(jīng)寫死的,我們無法更改。有這個圈也是必須的,因為sink本身是個臨時變量,如果不能跟另外一個臨時變量相互持有,兩個臨時變量都會被立即釋放。圈2:如果sink不會釋放,sink強持有了self,self又持有了bag,bag又持有了sink。導(dǎo)致圈2也被互相引用。想打破這個局其實很簡單,我們上述的解決方案就是在
圈2的持有1上做了手腳,弱引用一下self,此時self就跳出了圈2,當(dāng)self被釋放,bag也就被釋放,sink也就被釋放。圈1也就破了。其實同時我們還能在
持有3上搞一些事情。代碼如下:
Observable<Any>.create {(anyObserver) -> Disposable in
anyObserver.onNext("Hello word")
return Disposables.create()
}
.subscribe(onNext: {(item) in
print(self)
print("訂閱到:\(item)")
})
.disposed(by:DisposeBag())
}
我們把self持有的bag換成了一個臨時變量。打印結(jié)果如下:
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 0
<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7fb540c26950>
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 2
*****vc1即將出現(xiàn),此時引用計數(shù)為: 2
vc2執(zhí)行了析構(gòu)函數(shù)
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 0
<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7fb543f0f2e0>
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 2
*****vc1即將出現(xiàn),此時引用計數(shù)為: 2
vc2執(zhí)行了析構(gòu)函數(shù)
能夠看到問題同樣也解決了,但是一般我們不能這樣用。因為當(dāng)執(zhí)行到disposed(by:DisposeBag())時,整個序列會立即釋放。我們整個觀察序列就將毫無意義
操作四
如果我們創(chuàng)建與訂閱同時引用了self,而我們只在訂閱里弱引用的話,代碼如下:
Observable<Any>.create {(anyObserver) -> Disposable in
print(self)
anyObserver.onNext("Hello word")
return Disposables.create()
}
.subscribe(onNext: { [weak self](item) in
print(self)
print("訂閱到:\(item)")
})
.disposed(by: self.disposeBag)
}
我們再來打印:
*****vc1即將出現(xiàn),此時引用計數(shù)為: 0
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 0
<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7fba7bc20690>
Optional(<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7fba7bc20690>)
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 8
*****vc1即將出現(xiàn),此時引用計數(shù)為: 8
vc2執(zhí)行了析構(gòu)函數(shù)
*****vc1的push及執(zhí)行,此時引用計數(shù)為: 0
<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7fba7be332d0>
Optional(<_01_RxSwift?ü??≠òá?°áêü.ViewController: 0x7fba7be332d0>)
訂閱到:Hello word
*****vc1已經(jīng)消失,此時引用計數(shù)為: 8
*****vc1即將出現(xiàn),此時引用計數(shù)為: 8
vc2執(zhí)行了析構(gòu)函數(shù)
毫無問題,依然成功釋放。
結(jié)論:
我們使用序列的時候,只需要對subscribe里self進行弱引用,就能解決普通使用中的循環(huán)引用問題