RxSwift核心逻辑底层剖析(二)

RXSwift.jpg

注:代码块中的...是一些不需要显示的代码,直接用...替代了,rx的思维导图我后面重新画了一张,老漂亮了,画出经验了。

一.创建序列:

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

Observable遵守了ObservableType协议,create方法是协议的一个扩展方法。

create的具体实现可以在工程目录下搜索create.swift查看。

create方法返回了一个AnonymousObservable(subscribe)类,AnonymousObservable初始化保存了闭包。保存在了self._subscribeHandler属性中,self就是AnonymousObservable,也就是 AnonymousObservable. _subscribeHandler,这个很重要,后面会用到(函数式编程思想)。

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,实现了非序列向序列的转化,这个是万物皆序列的第二种解释,第一种解释前面的博客介绍过了。例如UISwitch().rx.value并不是一个序列,我们可以通过UISwitch().rx.value.asObservable() ,转成一个序列使用。
  1. func subscribe实现的是一个抽象方法,里面没有具体的实现。

这里画了一张序列继承链的思维导图,供大家参考,如有不对的地方望不吝赐教:


序列继承链.png

二.序列执行订阅操作

 let _ = ob.subscribe(onNext: { (text) in
            print("订阅到:\(text)")
        }, onError: { (error) in
            print("error: \(error)")
        }, onCompleted: {
            print("完成")
        }) {
            print("销毁")
        }

ob.subscribe方法中就下面的这段代码需要我们去分析。

  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>{...}创建了一个观察者。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,其实是一个基类订阅者,包含核心方法ononCore发送消息,和dispose方法执行销毁操作。

这里我们能够看到,如果AnonymousObserver<E>后面的闭包能够被调用,就能实现序列执行订阅操作中尾随闭包的调动,也就完成了序列的响应,也就是说后面我们要搞清楚序列和订阅是如何关联上的。
observer的调用是在:

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

self.asObservable().subscribe(observer)self.asObservable()代表的就是序列,也就变成 ob.subscribe(observer)(ob是刚开始创建的序列), 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
            }
        }
    }
...
}

可以看到其实就是执行了一个 let sinkAndSubscription = self.run(observer, cancel: disposer)run方法,同时返回了一个销毁者,run的调用者不是AnonymousObservable就是Producer,所以查找run方法的实现,如下。

run方法的查找

以下是run方法的实现:

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方法中创建了一个let sink = AnonymousObservableSink(observer: observer, cancel: cancel),保存了observer(订阅者)和cancel(销毁者)。然后调用了sink.run(self),这里的self就是对应的类AnonymousObservable,也就是sink.run(AnonymousObservable)也就是sink.run传入了一个序列,这一点非常重要,说明在sink中获得了序列,后面sink会把序列和订阅者关联。这个run方法的执行如下:

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)),这个方法是怎么实现订阅者和序列的关联,主要是通过AnyObserver(self),也就是 AnyObserver(AnonymousObservableSink),在AnyObserver这个结构体中,自定义属性self.observer = observer.on(注意啦,这里的self.observer是一个结构体的一个function),保存了 observeron方法的代码块,observersink传过来的,也就是

on方法

on 方法(AnonymousObservableSink类内的方法)的内部就是如下(核心逻辑):

 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()
            }
        }
    }

上面的意思总结起来就是 结构体AnyObserver的一个函数AnyObserver. observer保存了一个on函数块
回归到 AnonymousObservable._subscribeHandler(AnyObserver(self)),外界的obserber.onNext("onNext--1"),其实就是是调用的AnyObserver.onNext.而onNext其实就是调用了一个on方法(AnyObserver遵从了ObserverType协议,这个onNext是协议的扩展方法)

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()
            }
        }
    }
...
}

走到这里,在执行switch,---> 里面的self.forwardOn(event),调用的而其实是父类的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)调用 我们之前创建订阅者时的闭包(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()
                }
            }

然后就可以快乐的调用 onNextonErroronCompleted输出print(订阅到....)了,终于完了,好累。以上就是RXSwift底层实现的核心逻辑了。如有不对的地方,烦请不吝赐教,本人不胜感激~~~

二次补充:学习到后面后,回来缕了一下逻辑,画了一张流程简化的新图:
RXSwift核心逻辑简化图.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,242评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,769评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,484评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,133评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,007评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,080评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,496评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,190评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,464评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,549评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,330评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,205评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,567评论 3 298
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,889评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,160评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,475评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,650评论 2 335

推荐阅读更多精彩内容