RxSwift核心邏輯底層剖析(二)

RXSwift.jpg

注:代碼塊中的...是一些不需要顯示的代碼家浇,直接用...替代了躯畴,rx的思維導圖我后面重新畫了一張刨晴,老漂亮了,畫出經(jīng)驗了茫死。

一.創(chuàng)建序列:

 let ob = Observable<Any>.create { (obserber) -> Disposable in
            obserber.onNext("onNext1")
            return Disposables.create()
        }

Observable遵守了ObservableType協(xié)議跪但,create方法是協(xié)議的一個擴展方法。

create的具體實現(xiàn)可以在工程目錄下搜索create.swift查看峦萎。

create方法返回了一個AnonymousObservable(subscribe)類屡久,AnonymousObservable初始化保存了閉包。保存在了self._subscribeHandler屬性中爱榔,self就是AnonymousObservable被环,也就是 AnonymousObservable. _subscribeHandler,這個很重要,后面會用到(函數(shù)式編程思想)详幽。

final private class AnonymousObservable<Element>: Producer<Element> {
    typealias SubscribeHandler = (AnyObserver<Element>) -> Disposable

    let _subscribeHandler: SubscribeHandler

    init(_ subscribeHandler: @escaping SubscribeHandler) {
        self._subscribeHandler = subscribeHandler
    }

AnonymousObservable,繼承于父類Producer筛欢,父類Producer繼承于Observable,重寫了subscribe方法。
下面是父類Observable

public class Observable<Element> : ObservableType {
    /// Type of elements in sequence.
    public typealias E = Element
    
    init() {
#if TRACE_RESOURCES
        _ = Resources.incrementTotal()
#endif
    }
    
    public func subscribe<O: ObserverType>(_ observer: O) -> Disposable where O.E == E {
        rxAbstractMethod()
    }
    
    public func asObservable() -> Observable<E> {
        return self
    }
    
    deinit {
#if TRACE_RESOURCES
        _ = Resources.decrementTotal()
#endif
    }

解釋:

  1. func asObservable() 悴能,返回一個self揣钦,實現(xiàn)了非序列向序列的轉(zhuǎn)化,這個是萬物皆序列的第二種解釋漠酿,第一種解釋前面的博客介紹過了冯凹。例如UISwitch().rx.value并不是一個序列,我們可以通過UISwitch().rx.value.asObservable() ,轉(zhuǎn)成一個序列使用炒嘲。
  1. func subscribe實現(xiàn)的是一個抽象方法宇姚,里面沒有具體的實現(xiàn)。

這里畫了一張序列繼承鏈的思維導圖夫凸,供大家參考浑劳,如有不對的地方望不吝賜教:


序列繼承鏈.png

二.序列執(zhí)行訂閱操作

 let _ = ob.subscribe(onNext: { (text) in
            print("訂閱到:\(text)")
        }, onError: { (error) in
            print("error: \(error)")
        }, onCompleted: {
            print("完成")
        }) {
            print("銷毀")
        }

ob.subscribe方法中就下面的這段代碼需要我們?nèi)シ治觥?/p>

  let observer = AnonymousObserver<E> { event in
                switch event {
                case .next(let value):
                    onNext?(value)
                case .error(let error):
                    if let onError = onError {
                        onError(error)
                    }
                    else {
                        Hooks.defaultErrorHandler(callStack, error)
                    }
                    disposable.dispose()
                case .completed:
                    onCompleted?()
                    disposable.dispose()
                }
            }
            return Disposables.create(
                self.asObservable().subscribe(observer),
                disposable
            )

首先通過let observer = AnonymousObserver<E>{...}創(chuàng)建了一個觀察者。AnonymousObserver<E>中的E來自于public func subscribe(onNext: ((E) -> Void)? = nil, ....)中的E夭拌,而這個E(序列類型)魔熏,要從序列的subscribe方法中尋找,序列的subscribe方法是來自于:class AnonymousObservable<Element> ---->父類Producer ----> Observable<Element>

public class Observable<Element> : ObservableType {
    /// Type of elements in sequence.
    public typealias E = Element
    ...
}

接著說let observer = AnonymousObserver<E>AnonymousObserver類鸽扁,let observer = AnonymousObserver<E>{...}這個其實是進行了一個初始化的操作蒜绽, self._eventHandler = eventHandler保存了尾隨閉包:

final class AnonymousObserver<ElementType> : ObserverBase<ElementType> {
    typealias Element = ElementType
    
    typealias EventHandler = (Event<Element>) -> Void
    
    private let _eventHandler : EventHandler
    
    init(_ eventHandler: @escaping EventHandler) {
#if TRACE_RESOURCES
        _ = Resources.incrementTotal()
#endif
        self._eventHandler = eventHandler
    }

    override func onCore(_ event: Event<Element>) {
        return self._eventHandler(event)
    }
    
...
}

這里的AnonymousObserver繼承的ObserverBase,其實是一個基類訂閱者桶现,包含核心方法on躲雅, onCore發(fā)送消息,和dispose方法執(zhí)行銷毀操作骡和。

這里我們能夠看到相赁,如果AnonymousObserver<E>后面的閉包能夠被調(diào)用,就能實現(xiàn)序列執(zhí)行訂閱操作中尾隨閉包的調(diào)動慰于,也就完成了序列的響應,也就是說后面我們要搞清楚序列和訂閱是如何關(guān)聯(lián)上的钮科。
observer的調(diào)用是在:

 return Disposables.create(
                self.asObservable().subscribe(observer),
                disposable
            )

self.asObservable().subscribe(observer)self.asObservable()代表的就是序列,也就變成 ob.subscribe(observer)(ob是剛開始創(chuàng)建的序列), obsubscribe婆赠,需要到其父類(AnonymousObservable)的父類(Producer)中尋找跺嗽。

class Producer<Element> : Observable<Element> {
    override init() {
        super.init()
    }

    override func subscribe<O : ObserverType>(_ observer: O) -> Disposable where O.E == Element {
        if !CurrentThreadScheduler.isScheduleRequired {
            // The returned disposable needs to release all references once it was disposed.
            let disposer = SinkDisposer()
            let sinkAndSubscription = self.run(observer, cancel: disposer)
            disposer.setSinkAndSubscription(sink: sinkAndSubscription.sink, subscription: sinkAndSubscription.subscription)

            return disposer
        }
        else {
            return CurrentThreadScheduler.instance.schedule(()) { _ in
                let disposer = SinkDisposer()
                let sinkAndSubscription = self.run(observer, cancel: disposer)
                disposer.setSinkAndSubscription(sink: sinkAndSubscription.sink, subscription: sinkAndSubscription.subscription)

                return disposer
            }
        }
    }
...
}

可以看到其實就是執(zhí)行了一個 let sinkAndSubscription = self.run(observer, cancel: disposer)run方法,同時返回了一個銷毀者,run的調(diào)用者不是AnonymousObservable就是Producer页藻,所以查找run方法的實現(xiàn),如下植兰。

run方法的查找

以下是run方法的實現(xiàn):

final private class AnonymousObservable<Element>: Producer<Element> {
    typealias SubscribeHandler = (AnyObserver<Element>) -> Disposable

    let _subscribeHandler: SubscribeHandler

    init(_ subscribeHandler: @escaping SubscribeHandler) {
        self._subscribeHandler = subscribeHandler
    }

    override func run<O : ObserverType>(_ observer: O, cancel: Cancelable) -> (sink: Disposable, subscription: Disposable) where O.E == Element {
        let sink = AnonymousObservableSink(observer: observer, cancel: cancel)
        let subscription = sink.run(self)
        return (sink: sink, subscription: subscription)
    }
}

我們看到run方法中創(chuàng)建了一個let sink = AnonymousObservableSink(observer: observer, cancel: cancel)份帐,保存了observer(訂閱者)和cancel(銷毀者)。然后調(diào)用了sink.run(self)楣导,這里的self就是對應的類AnonymousObservable废境,也就是sink.run(AnonymousObservable)也就是sink.run傳入了一個序列,這一點非常重要噩凹,說明在sink中獲得了序列巴元,后面sink會把序列和訂閱者關(guān)聯(lián)。這個run方法的執(zhí)行如下:

func run(_ parent: Parent) -> Disposable {
        return parent._subscribeHandler(AnyObserver(self))
    }

parent._subscribeHandler(AnyObserver(self)) 其實就是 AnonymousObservable._subscribeHandler(AnyObserver(self)),這個方法是不是很熟悉驮宴,是的逮刨,就是create初始化保存的代碼塊。這里我把run方法的整個代碼塊貼出來堵泽,因為非常重要:

final private class AnonymousObservableSink<O: ObserverType>: Sink<O>, ObserverType {
    typealias E = O.E
    typealias Parent = AnonymousObservable<E>

    // state
    private let _isStopped = AtomicInt(0)

    #if DEBUG
        fileprivate let _synchronizationTracker = SynchronizationTracker()
    #endif

    override init(observer: O, cancel: Cancelable) {
        super.init(observer: observer, cancel: cancel)
    }

    func on(_ event: Event<E>) {
        #if DEBUG
            self._synchronizationTracker.register(synchronizationErrorMessage: .default)
            defer { self._synchronizationTracker.unregister() }
        #endif
        switch event {
        case .next:
            if load(self._isStopped) == 1 {
                return
            }
            self.forwardOn(event)
        case .error, .completed:
            if fetchOr(self._isStopped, 1) == 0 {
                self.forwardOn(event)
                self.dispose()
            }
        }
    }

    func run(_ parent: Parent) -> Disposable {
        return parent._subscribeHandler(AnyObserver(self))
    }
}

AnonymousObservable._subscribeHandler(AnyObserver(self)),這個方法是怎么實現(xiàn)訂閱者和序列的關(guān)聯(lián)修己,主要是通過AnyObserver(self),也就是 AnyObserver(AnonymousObservableSink)迎罗,在AnyObserver這個結(jié)構(gòu)體中,自定義屬性self.observer = observer.on(注意啦尤辱,這里的self.observer是一個結(jié)構(gòu)體的一個function)厢岂,保存了 observeron方法的代碼塊光督,observersink傳過來的,也就是

on方法

on 方法(AnonymousObservableSink類內(nèi)的方法)的內(nèi)部就是如下(核心邏輯):

 func on(_ event: Event<E>) {
        #if DEBUG
            self._synchronizationTracker.register(synchronizationErrorMessage: .default)
            defer { self._synchronizationTracker.unregister() }
        #endif
        switch event {
        case .next:
            if load(self._isStopped) == 1 {
                return
            }
            self.forwardOn(event)
        case .error, .completed:
            if fetchOr(self._isStopped, 1) == 0 {
                self.forwardOn(event)
                self.dispose()
            }
        }
    }

上面的意思總結(jié)起來就是 結(jié)構(gòu)體AnyObserver的一個函數(shù)AnyObserver. observer保存了一個on函數(shù)塊
回歸到 AnonymousObservable._subscribeHandler(AnyObserver(self))可帽,外界的obserber.onNext("onNext--1")窗怒,其實就是是調(diào)用的AnyObserver.onNext.而onNext其實就是調(diào)用了一個on方法(AnyObserver遵從了ObserverType協(xié)議映跟,這個onNext是協(xié)議的擴展方法)

extension ObserverType {
    
    public func onNext(_ element: E) {
        self.on(.next(element))
    }
    
    public func onCompleted() {
        self.on(.completed)
    }
    
    public func onError(_ error: Swift.Error) {
        self.on(.error(error))
    }
}

這個on方法,其實就是前面提到的AnonymousObservableSink中的on方法扬虚,再上一次代碼塊:

final private class AnonymousObservableSink<O: ObserverType>: Sink<O>, ObserverType {
   ...
    func on(_ event: Event<E>) {
        #if DEBUG
            self._synchronizationTracker.register(synchronizationErrorMessage: .default)
            defer { self._synchronizationTracker.unregister() }
        #endif
        switch event {
        case .next:
            if load(self._isStopped) == 1 {
                return
            }
            self.forwardOn(event)
        case .error, .completed:
            if fetchOr(self._isStopped, 1) == 0 {
                self.forwardOn(event)
                self.dispose()
            }
        }
    }
...
}

走到這里,在執(zhí)行switch荸镊,---> 里面的self.forwardOn(event)堪置,調(diào)用的而其實是父類的forwardOn方法:

class Sink<O : ObserverType> : Disposable {
   ...
    final func forwardOn(_ event: Event<O.E>) {
        #if DEBUG
            self._synchronizationTracker.register(synchronizationErrorMessage: .default)
            defer { self._synchronizationTracker.unregister() }
        #endif
        if isFlagSet(self._disposed, 1) {
            return
        }
        self._observer.on(event)
    }
   ...
}

最后通過self._observer.on(event)調(diào)用 我們之前創(chuàng)建訂閱者時的閉包(let observer = AnonymousObserver<E> {閉包} )

let observer = AnonymousObserver<E> { event in
                
                #if DEBUG
                    synchronizationTracker.register(synchronizationErrorMessage: .default)
                    defer { synchronizationTracker.unregister() }
                #endif
                
                switch event {
                case .next(let value):
                    onNext?(value)
                case .error(let error):
                    if let onError = onError {
                        onError(error)
                    }
                    else {
                        Hooks.defaultErrorHandler(callStack, error)
                    }
                    disposable.dispose()
                case .completed:
                    onCompleted?()
                    disposable.dispose()
                }
            }

然后就可以快樂的調(diào)用 onNext舀锨,onError岭洲,onCompleted輸出print(訂閱到....)了盾剩,終于完了,好累告私。以上就是RXSwift底層實現(xiàn)的核心邏輯了驻粟。如有不對的地方,煩請不吝賜教格嗅,本人不勝感激~~~

二次補充:學習到后面后,回來縷了一下邏輯玄柏,畫了一張流程簡化的新圖:
RXSwift核心邏輯簡化圖.png
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末粪摘,一起剝皮案震驚了整個濱河市绍坝,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌椎咧,老刑警劉巖把介,帶你破解...
    沈念sama閱讀 222,104評論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異脚牍,居然都是意外死亡巢墅,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,816評論 3 399
  • 文/潘曉璐 我一進店門驯遇,熙熙樓的掌柜王于貴愁眉苦臉地迎上來蓄髓,“玉大人双吆,你說我怎么就攤上這事『美郑” “怎么了蔚万?”我有些...
    開封第一講書人閱讀 168,697評論 0 360
  • 文/不壞的土叔 我叫張陵,是天一觀的道長昵慌。 經(jīng)常有香客問我淮蜈,道長,這世上最難降的妖魔是什么淳蔼? 我笑而不...
    開封第一講書人閱讀 59,836評論 1 298
  • 正文 為了忘掉前任裁眯,我火速辦了婚禮,結(jié)果婚禮上存皂,老公的妹妹穿的比我還像新娘逢艘。我一直安慰自己,他們只是感情好猜憎,可當我...
    茶點故事閱讀 68,851評論 6 397
  • 文/花漫 我一把揭開白布搔课。 她就那樣靜靜地躺著爬泥,像睡著了一般。 火紅的嫁衣襯著肌膚如雪袍啡。 梳的紋絲不亂的頭發(fā)上境输,一...
    開封第一講書人閱讀 52,441評論 1 310
  • 那天颖系,我揣著相機與錄音辩越,去河邊找鬼。 笑死趁啸,一個胖子當著我的面吹牛督惰,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播访娶,決...
    沈念sama閱讀 40,992評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼震肮,長吁一口氣:“原來是場噩夢啊……” “哼留拾!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起痴柔,我...
    開封第一講書人閱讀 39,899評論 0 276
  • 序言:老撾萬榮一對情侶失蹤咳蔚,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后侈询,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體糯耍,經(jīng)...
    沈念sama閱讀 46,457評論 1 318
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,529評論 3 341
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了舵鳞。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,664評論 1 352
  • 序言:一個原本活蹦亂跳的男人離奇死亡抛虏,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出贺奠,到底是詐尸還是另有隱情错忱,我是刑警寧澤挂据,帶...
    沈念sama閱讀 36,346評論 5 350
  • 正文 年R本政府宣布崎逃,位于F島的核電站,受9級特大地震影響个绍,放射性物質(zhì)發(fā)生泄漏巴柿。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 42,025評論 3 334
  • 文/蒙蒙 一凯旋、第九天 我趴在偏房一處隱蔽的房頂上張望钉迷。 院中可真熱鬧,春花似錦荒椭、人聲如沸舰蟆。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,511評論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽题造。三九已至,卻和暖如春界赔,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背咐低。 一陣腳步聲響...
    開封第一講書人閱讀 33,611評論 1 272
  • 我被黑心中介騙來泰國打工见擦, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人鲤屡。 一個月前我還...
    沈念sama閱讀 49,081評論 3 377
  • 正文 我出身青樓酒来,卻偏偏與公主長得像,于是被迫代替她去往敵國和親辽社。 傳聞我的和親對象是個殘疾皇子翘鸭,可洞房花燭夜當晚...
    茶點故事閱讀 45,675評論 2 359

推薦閱讀更多精彩內(nèi)容