透过源码剖析RxSwift的subscribeOn与observeOn

引言

在上一篇文章中，我们梳理了RxSwift/RxJava的定义、订阅、事件产生和消费的过程，并对源代码进行了分析。本文将着重讨论RxSwift的subscribeOn和observeOn这两个操作符的源代码实现。如果您还没有阅读过前一篇文章，建议您先移步阅读。（注意：本文中涉及的Rx源码基于RxSwift 5.1.0版本。）

subscribeOn与observeOn概述

subscribeOn和observeOn是RxSwift中用于控制事件订阅和观察的两个重要操作符。subscribeOn指定了事件源（Observable）在哪个调度器上进行订阅，而observeOn指定了事件流（Observable）在哪个调度器上进行观察。这两个操作符可以帮助我们控制事件的执行线程，从而实现异步、并发编程。

subscribeOn

subscribeOn操作符用于指定事件源（Observable）的订阅过程在哪一个调度器上执行。我们可以通过它来控制事件的产生线程，例如，我们可以将subscribeOn指定为主线程，这样事件的产生就会在主线程中进行，从而避免阻塞主线程。

observeOn

observeOn操作符用于指定事件流（Observable）的观察过程在哪一个调度器上执行。我们可以通过它来控制事件的消费线程，例如，我们可以将observeOn指定为主线程，这样事件的消费就会在主线程中进行，从而确保UI更新在主线程中执行。

subscribeOn与observeOn源码分析

subscribeOn

subscribeOn操作符的源码位于RxSwift的RxCocoa框架中，具体路径为RxCocoa/Sources/Cocoa/Observable+Operators.swift。其核心代码如下：

public func subscribeOn(_ scheduler: ImmediateSchedulerType) -> Observable<Element> {
    return SubscribeOn(source: self, scheduler: scheduler)
}

从代码中可以看出，subscribeOn操作符接受一个ImmediateSchedulerType类型的参数，并返回一个新的Observable。这个新的Observable将使用指定的调度器来订阅事件源。

observeOn

observeOn操作符的源码位于RxSwift框架中，具体路径为RxSwift/Sources/Observable/Operators/ObserveOn.swift。其核心代码如下：

public func observeOn(_ scheduler: ImmediateSchedulerType) -> Observable<Element> {
    return ObserveOn(source: self, scheduler: scheduler)
}

observeOn操作符也接受一个ImmediateSchedulerType类型的参数，并返回一个新的Observable。这个新的Observable将使用指定的调度器来观察事件流。

实际应用场景

subscribeOn和observeOn操作符在实际开发中非常有用。以下是一些常见的应用场景：

避免阻塞主线程

在iOS开发中，主线程是负责处理UI更新的线程。如果我们在主线程中执行耗时的操作，就会导致UI卡顿。为了避免这种情况，我们可以使用subscribeOn操作符将耗时的操作放在后台线程中执行，然后使用observeOn操作符将结果返回到主线程中进行UI更新。

实现并发编程

RxSwift的subscribeOn和observeOn操作符可以帮助我们轻松实现并发编程。我们可以使用subscribeOn操作符将不同的任务分配到不同的线程中执行，然后使用observeOn操作符将结果汇总到主线程中进行处理。

结束语

subscribeOn和observeOn是RxSwift中非常重要的两个操作符，它们可以帮助我们控制事件的执行线程，从而实现异步、并发编程。通过对这两个操作符源码的分析，我们可以更深入地理解RxSwift的实现原理，并将其应用到实际开发中。