前言
很多项目使用流行的Rxjava2 + Retrofit搭建网络框架,Rxjava现在已经发展到Rxjava2,之前一直都只是再用Rxjava,但从来没有了解下Rxjava的内部实现,接下来一步步来分析Rxjava2的源码,Rxjava2分Observable和Flowable两种(无被压和有被压),我们今天先从简单的无背压的observable来分析。源码基于rxjava:2.1.1。
一、Rxjava如何创建事件源、发射事件、何时发射事件、如何将观察者和被观察者关联起来
简单的例子
先来段最简单的代码,直观的了解下整个Rxjava运行的完整流程。
1 private void doSomeWork() {
2 Observable<String> observable = Observable.create(new ObservableOnSubscribe<String>() {
3 @Override
4 public void subscribe(ObservableEmitter<String> e) throws Exception {
5 e.onNext("a");
6 e.onComplete();
7 }
8 });
9 Observer observer = new Observer<String>() {
10
11 @Override
12 public void onSubscribe(Disposable d) {
13 Log.i("lx", " onSubscribe : " + d.isDisposed());
14 }
15
16 @Override
17 public void onNext(String str) {
18 Log.i("lx", " onNext : " + str);
19 }
20
21 @Override
22 public void onError(Throwable e) {
23 Log.i("lx", " onError : " + e.getMessage());
24 }
25
26 @Override
27 public void onComplete() {
28 Log.i("lx", " onComplete");
29 }
30 };
31 observable.subscribe(observer);
32 }
上面代码之所以将observable和observer单独声明,最后再调用observable.subscribe(observer);
是为了分步来分析:
- 被观察者 Observable 如何生产事件的
- 被观察者 Observable 何时生产事件的
- 观察者Observer是何时接收到上游事件的
- Observable 与Observer是如何关联在一起的
Observable
Observable是数据的上游,即事件生产者
首先来分析事件是如何生成的,直接看代码 Observable.create()
方法。
1 @SchedulerSupport(SchedulerSupport.NONE)
2 public static <T> Observable<T> create(ObservableOnSubscribe<T> source) { // ObservableOnSubscribe 是个接口,只包含subscribe方法,是事件生产的源头。
3 ObjectHelper.requireNonNull(source, "source is null"); // 判空
4 return RxJavaPlugins.onAssembly(new ObservableCreate<T>(source));
5 }
最重要的是RxJavaPlugins.onAssembly(new ObservableCreate<T>(source));这句代码。继续跟踪进去
1 /**
2 * Calls the associated hook function.
3 * @param <T> the value type
4 * @param source the hook's input value
5 * @return the value returned by the hook
6 */
7 @SuppressWarnings({ "rawtypes", "unchecked" })
8 @NonNull
9 public static <T> Observable<T> onAssembly(@NonNull Observable<T> source) {
10 Function<? super Observable, ? extends Observable> f = onObservableAssembly;
11 if (f != null) {
12 return apply(f, source);
13 }
14 return source;
15 }
看注释,原来这个方法是个hook function。 通过调试得知静态对象onObservableAssembly默认为null, 所以此方法直接返回传入的参数source。
onObservableAssembly可以通过静态方法RxJavaPlugins. setOnObservableAssembly ()设置全局的Hook函数, 有兴趣的同学可以自己去试试。 这里暂且不谈,我们继续返回代码。
现在我们明白了:
1 Observable<String> observable=Observable.create(new ObservableOnSubscribe<String>() {
2 ...
3 ...
4 })
相当于:
1 Observable<String> observable=new ObservableCreate(new ObservableOnSubscribe<String>() {
2 ...
3 ...
4 }))
好了,至此我们明白了,事件的源就是new ObservableCreate()
对象,将ObservableOnSubscribe
作为参数传递给ObservableCreate
的构造函数。
事件是由接口ObservableOnSubscribe
的subscribe方法上产的,至于何时生产事件,稍后再分析。
Observer
Observer 是数据的下游,即事件消费者
Observer是个interface,包含 :
1 void onSubscribe(@NonNull Disposable d);
2 void onNext(@NonNull T t);
3 void onError(@NonNull Throwable e);
4 void onComplete();
上游发送的事件就是再这几个方法中被消费的。上游何时发送事件、如何发送,稍后再表。
subscribe
重点来了,接下来最重要的方法来了:observable.subscribe(observer);
从这个方法的名字就知道,subscribe是订阅,是将观察者(observer)与被观察者(observable)连接起来的方法。只有subscribe方法执行后,上游产生的事件才能被下游接收并处理。其实自然的方式应该是observer订阅(subscribe) observable, 但这样会打断rxjava的链式结构。所以采用相反的方式。
接下来看源码,只列出关键代码
1 public final void subscribe(Observer<? super T> observer) {
2 ObjectHelper.requireNonNull(observer, "observer is null");
3 ......
4 observer = RxJavaPlugins.onSubscribe(this, observer); // hook ,默认直接返回observer
5 ......
6 subscribeActual(observer); // 这个才是真正实现订阅的方法。
7 ......
8 }
9
10 // subscribeActual 是抽象方法,所以需要到实现类中去看具体实现,也就是说实现是在上文中提到的ObservableCreate中
11 protected abstract void subscribeActual(Observer<? super T> observer);
接下来我们来看ObservableCreate.java:
1 public ObservableCreate(ObservableOnSubscribe<T> source) {
2 this.source = source; // 事件源,生产事件的接口,由我们自己实现
3 }
4
5 @Override
6 protected void subscribeActual(Observer<? super T> observer) {
7 CreateEmitter<T> parent = new CreateEmitter<T>(observer); // 发射器
8 observer.onSubscribe(parent); //直接回调了观察者的onSubscribe
9
10 try {
11 // 调用了事件源subscribe方法生产事件,同时将发射器传给事件源。
12 // 现在我们明白了,数据源生产事件的subscribe方法只有在observable.subscribe(observer)被执行
13 后才执行的。 换言之,事件流是在订阅后才产生的。
14 //而observable被创建出来时并不生产事件,同时也不发射事件。
15 source.subscribe(parent);
16 } catch (Throwable ex) {
17 Exceptions.throwIfFatal(ex);
18 parent.onError(ex);
19 }
20 }
现在我们明白了,数据源生产事件的subscribe方法只有在observable.subscribe(observer)被执行后才执行的。 换言之,事件流是在订阅后才产生的。而observable被创建出来时并不生产事件,同时也不发射事件。
接下来我们再来看看事件是如何被发射出去,同时observer是如何接收到发射的事件的CreateEmitter<T> parent = new CreateEmitter<T>(observer);
CreateEmitter 实现了ObservableEmitter接口,同时ObservableEmitter接口又继承了Emitter接口。
CreateEmitter 还实现了Disposable接口,这个disposable接口是用来判断是否中断事件发射的。
从名称上就能看出,这个是发射器,故名思议是用来发射事件的,正是它将上游产生的事件发射到下游的。
Emitter是事件源与下游的桥梁。
CreateEmitter 主要包括方法:
1 void onNext(@NonNull T value);
2 void onError(@NonNull Throwable error);
3 void onComplete();
4 public void dispose() ;
5 public boolean isDisposed();
是不是跟observer的方法很像?
我们来看看CreateEmitter中这几个方法的具体实现:
只列出关键代码
1 public void onNext(T t) {
2 if (!isDisposed()) { // 判断事件是否需要被丢弃
3 observer.onNext(t); // 调用Emitter的onNext,它会直接调用observer的onNext
4 }
5 }
6 public void onError(Throwable t) {
7 if (!isDisposed()) {
8 try {
9 observer.onError(t); // 调用Emitter的onError,它会直接调用observer的onError
10 } finally {
11 dispose(); // 当onError被触发时,执行dispose(), 后续onNext,onError, onComplete就不会继
12 续发射事件了
13 }
14 }
15 }
16
17 @Override
18 public void onComplete() {
19 if (!isDisposed()) {
20 try {
21 observer.onComplete(); // 调用Emitter的onComplete,它会直接调用observer的onComplete
22 } finally {
23 dispose(); // 当onComplete被触发时,也会执行dispose(), 后续onNext,onError, onComplete
24 同样不会继续发射事件了
25 }
26 }
27 }
CreateEmitter 的onError和onComplete方法任何一个执行完都会执行dispose()中断事件发射,所以observer中的onError和onComplete也只能有一个被执行。
现在终于明白了,事件是如何被发射给下游的。当订阅成功后,数据源ObservableOnSubscribe开始生产事件,调用Emitter的onNext,onComplete向下游发射事件,
Emitter包含了observer的引用,又调用了observer onNext,onComplete,这样下游observer就接收到了上游发射的数据。
总结
Rxjava的流程大概是:
- Observable.create 创建事件源,但并不生产也不发射事件。
- 实现observer接口,但此时没有也无法接受到任何发射来的事件。
- 订阅 observable.subscribe(observer), 此时会调用具体Observable的实现类中的subscribeActual方法,
此时会才会真正触发事件源生产事件,事件源生产出来的事件通过Emitter的onNext,onError,onComplete发射给observer对应的方法由下游observer消费掉。从而完成整个事件流的处理。
observer中的onSubscribe在订阅时即被调用,并传回了Disposable, observer中可以利用Disposable来随时中断事件流的发射。
今天所列举的例子是最简单的一个事件处理流程,没有使用线程调度,Rxjava最强大的就是异步时对线程的调度和随时切换观察者线程,未完待续。
上面分析了Rxjava是如何创建事件源,如何发射事件,何时发射事件,也清楚了上游和下游是如何关联起来的。
下面着重来分析下Rxjava强大的线程调度是如何实现的。
二、RxJava的线程调度机制
简单的例子
1 private void doSomeWork() {
2 Observable.create(new ObservableOnSubscribe<String>() {
3 @Override
4 public void subscribe(ObservableEmitter<String> e) throws Exception {
5 Log.i("lx", " subscribe: " + Thread.currentThread().getName());
6 Thread.sleep(2000);
7 e.onNext("a");
8 e.onComplete();
9 }
10 }).subscribe(new Observer<String>() {
11 @Override
12 public void onSubscribe(Disposable d) {
13 Log.i("lx", " onSubscribe: " + Thread.currentThread().getName());
14 }
15 @Override
16 public void onNext(String str) {
17 Log.i("lx", " onNext: " + Thread.currentThread().getName());
18 }
19 @Override
20 public void onError(Throwable e) {
21 Log.i("lx", " onError: " + Thread.currentThread().getName());
22 }
23 @Override
24 public void onComplete() {
25 Log.i("lx", " onComplete: " + Thread.currentThread().getName());
26 }
27 });
28 }
运行结果:
1 com.rxjava2.android.samples I/lx: onSubscribe: main
2 com.rxjava2.android.samples I/lx: subscribe: main
3 com.rxjava2.android.samples I/lx: onNext: main
4 com.rxjava2.android.samples I/lx: onComplete: main
因为此方法笔者是在main线程中调用的,所以没有进行线程调度的情况下,所有方法都运行在main线程中。但我们知道Android的UI线程是不能做网络操作,也不能做耗时操作,所以一般我们把网络或耗时操作都放在非UI线程中执行。接下来我们就来感受下Rxjava强大的线程调度能力。
1 private void doSomeWork() {
2 Observable.create(new ObservableOnSubscribe<String>() {
3 @Override
4 public void subscribe(ObservableEmitter<String> e) throws Exception {
5 Log.i("lx", " subscribe: " + Thread.currentThread().getName());
6 Thread.sleep(2000);
7 e.onNext("a");
8 e.onComplete();
9 }
10 }).subscribeOn(Schedulers.io()) //增加了这一句
11 .subscribe(new Observer<String>() {
12 @Override
13 public void onSubscribe(Disposable d) {
14 Log.i("lx", " onSubscribe: " + Thread.currentThread().getName());
15 }
16 @Override
17 public void onNext(String str) {
18 Log.i("lx", " onNext: " + Thread.currentThread().getName());
19 }
20 @Override
21 public void onError(Throwable e) {
22 Log.i("lx", " onError: " + Thread.currentThread().getName());
23 }
24 @Override
25 public void onComplete() {
26 Log.i("lx", " onComplete: " + Thread.currentThread().getName());
27 }
28 });
29 }
运行结果:
1 com.rxjava2.android.samples I/lx: onSubscribe: main
2 com.rxjava2.android.samples I/lx: subscribe: RxCachedThreadScheduler-1
3 com.rxjava2.android.samples I/lx: onNext: RxCachedThreadScheduler-1
4 com.rxjava2.android.samples I/lx: onComplete: RxCachedThreadScheduler-1
只增加了subscribeOn
这一句代码, 就发生如此神奇的现象,除了onSubscribe方法还运行在main线程(订阅发生的线程)其它方法全部都运行在一个名为RxCachedThreadScheduler-1的线程中。我们来看看rxjava是怎么完成这个线程调度的。
线程调度subscribeOn
首先我们先分析下Schedulers.io()
这个东东。
1 @NonNull
2 public static Scheduler io() {
3 return RxJavaPlugins.onIoScheduler(IO); // hook function
4 // 等价于
5 return IO;
6 }
再看看IO是什么, IO是个static变量,初始化的地方是
1 IO = RxJavaPlugins.initIoScheduler(new IOTask()); // 又是hook function
2 // 等价于
3 IO = callRequireNonNull(new IOTask());
4 // 等价于
5 IO = new IOTask().call();
继续看看IOTask
1 static final class IOTask implements Callable<Scheduler> {
2 @Override
3 public Scheduler call() throws Exception {
4 return IoHolder.DEFAULT;
5 // 等价于
6 return new IoScheduler();
7 }
8 }
代码层次很深,为了便于记忆,我们再回顾一下:
1 Schedulers.io()等价于 new IoScheduler()
2
3 // Schedulers.io()等价于
4 @NonNull
5 public static Scheduler io() {
6 return new IoScheduler();
7 }
好了,排除了其他干扰代码,接下来看看IoScheduler()是什么东东了
IoScheduler看名称就知道是个IO线程调度器,根据代码注释得知,它就是一个用来创建和缓存线程的线程池。看到这个豁然开朗了,原来Rxjava就是通过这个调度器来调度线程的,至于具体怎么实现我们接着往下看
1 public IoScheduler() {
2 this(WORKER_THREAD_FACTORY);
3 }
4
5 public IoScheduler(ThreadFactory threadFactory) {
6 this.threadFactory = threadFactory;
7 this.pool = new AtomicReference<CachedWorkerPool>(NONE);
8 start();
9 }
10
11 @Override
12 public void start() {
13 CachedWorkerPool update = new CachedWorkerPool(KEEP_ALIVE_TIME, KEEP_ALIVE_UNIT, threadFactory);
14 if (!pool.compareAndSet(NONE, update)) {
15 update.shutdown();
16 }
17 }
18
19 CachedWorkerPool(long keepAliveTime, TimeUnit unit, ThreadFactory threadFactory) {
20 this.keepAliveTime = unit != null ? unit.toNanos(keepAliveTime) : 0L;
21 this.expiringWorkerQueue = new ConcurrentLinkedQueue<ThreadWorker>();
22 this.allWorkers = new CompositeDisposable();
23 this.threadFactory = threadFactory;
24
25 ScheduledExecutorService evictor = null;
26 Future<?> task = null;
27 if (unit != null) {
28 evictor = Executors.newScheduledThreadPool(1, EVICTOR_THREAD_FACTORY);
29 task = evictor.scheduleWithFixedDelay(this, this.keepAliveTime, this.keepAliveTime, TimeUnit.NANOSECONDS);
30 }
31 evictorService = evictor;
32 evictorTask = task;
33 }
从上面的代码可以看出,new IoScheduler()
后Rxjava会创建CachedWorkerPool
的线程池,同时也创建并运行了一个名为RxCachedWorkerPoolEvictor
的清除线程,主要作用是清除不再使用的一些线程。
但目前只创建了线程池并没有实际的thread,所以Schedulers.io()
相当于只做了线程调度的前期准备。
OK,终于可以开始分析Rxjava是如何实现线程调度的。回到Demo来看subscribeOn()
方法的内部实现:
1 public final Observable<T> subscribeOn(Scheduler scheduler) {
2 ObjectHelper.requireNonNull(scheduler, "scheduler is null");
3 return RxJavaPlugins.onAssembly(new ObservableSubscribeOn<T>(this, scheduler));
4 }
很熟悉的代码RxJavaPlugins.onAssembly
,上一篇已经分析过这个方法,就是个hook function, 等价于直接return new ObservableSubscribeOn<T>(this, scheduler);
, 现在知道了这里的scheduler其实就是IoScheduler。
跟踪代码进入ObservableSubscribeOn
,
可以看到这个ObservableSubscribeOn 继承自Observable,并且扩展了一些属性,增加了scheduler。 各位看官,这不就是典型的装饰模式嘛,Rxjava中大量用到了装饰模式,后面还会经常看到这种wrap类。
上篇文章我们已经知道了Observable.subscribe()
方法最终都是调用了对应的实现类的subscribeActual
方法。我们重点分析下subscribeActual
:
1 @Override
2 public void subscribeActual(final Observer<? super T> s) {
3 final SubscribeOnObserver<T> parent = new SubscribeOnObserver<T>(s);
4
5 // 没有任何线程调度,直接调用的,所以下游的onSubscribe方法没有切换线程,
6 //本文demo中下游就是观察者,所以我们明白了为什么只有onSubscribe还运行在main线程
7 s.onSubscribe(parent);
8
9 parent.setDisposable(scheduler.scheduleDirect(new SubscribeTask(parent)));
10 }
SubscribeOnObserver
也是装饰模式的体现, 是对下游observer
的一个wrap
,只是添加了Disposable
的管理。
接下来分析最重要的scheduler.scheduleDirect(new SubscribeTask(parent))
1 // 这个类很简单,就是一个Runnable,最终运行上游的subscribe方法
2 final class SubscribeTask implements Runnable {
3 private final SubscribeOnObserver<T> parent;
4
5 SubscribeTask(SubscribeOnObserver<T> parent) {
6 this.parent = parent;
7 }
8
9 @Override
10 public void run() {
11 source.subscribe(parent);
12 }
13 }
14 @NonNull
15 public Disposable scheduleDirect(@NonNull Runnable run, long delay, @NonNull TimeUnit unit) {
16 // IoSchedular 中的createWorker()
17 final Worker w = createWorker();
18 // hook decoratedRun=run;
19 final Runnable decoratedRun = RxJavaPlugins.onSchedule(run);
20 // decoratedRun的wrap,增加了Dispose的管理
21 DisposeTask task = new DisposeTask(decoratedRun, w);
22 // 线程调度
23 w.schedule(task, delay, unit);
24
25 return task;
26 }
回到IoSchedular
1 public Worker createWorker() {
2 // 工作线程是在此时创建的
3 return new EventLoopWorker(pool.get());
4 }
5
6 public Disposable schedule(@NonNull Runnable action, long delayTime, @NonNull TimeUnit unit) {
7 if (tasks.isDisposed()) {
8 // don't schedule, we are unsubscribed
9 return EmptyDisposable.INSTANCE;
10 }
11 // action 中就包含上游subscribe的runnable
12 return threadWorker.scheduleActual(action, delayTime, unit, tasks);
13 }
最终线程是在这个方法内调度并执行的。
1 public ScheduledRunnable scheduleActual(final Runnable run, long delayTime, @NonNull TimeUnit unit, @Nullable DisposableContainer parent) {
2 // decoratedRun = run, 包含上游subscribe方法的runnable
3 Runnable decoratedRun = RxJavaPlugins.onSchedule(run);
4
5 // decoratedRun的wrap,增加了dispose的管理
6 ScheduledRunnable sr = new ScheduledRunnable(decoratedRun, parent);
7
8 if (parent != null) {
9 if (!parent.add(sr)) {
10 return sr;
11 }
12 }
13
14 // 最终decoratedRun被调度到之前创建或从线程池中取出的线程,
15 // 也就是说在RxCachedThreadScheduler-x运行
16 Future<?> f;
17 try {
18 if (delayTime <= 0) {
19 f = executor.submit((Callable<Object>)sr);
20 } else {
21 f = executor.schedule((Callable<Object>)sr, delayTime, unit);
22 }
23 sr.setFuture(f);
24 } catch (RejectedExecutionException ex) {
25 if (parent != null) {
26 parent.remove(sr);
27 }
28 RxJavaPlugins.onError(ex);
29 }
30
31 return sr;
32 }
至此我们终于明白了Rxjava是如何调度线程并执行的,通过subscribeOn方法将上游生产事件的方法运行在指定的调度线程中。
1 com.rxjava2.android.samples I/lx: onSubscribe: main
2 com.rxjava2.android.samples I/lx: subscribe: RxCachedThreadScheduler-1
3 com.rxjava2.android.samples I/lx: onNext: RxCachedThreadScheduler-1
4 com.rxjava2.android.samples I/lx: onComplete: RxCachedThreadScheduler-1
从上面的运行结果来看,因为上游生产者已被调度到RxCachedThreadScheduler-1
线程中,同时发射事件并没有切换线程,所以发射后消费事件的onNext onErro onComplete
也在RxCachedThreadScheduler-1
线程中。
总结
Schedulers.io()
等价于new IoScheduler()。
new IoScheduler()
Rxjava
创建了线程池,为后续创建线程做准备,同时创建并运行了一个清理线程RxCachedWorkerPoolEvictor
,定期执行清理任务。subscribeOn()
返回一个ObservableSubscribeOn
对象,它是Observable
的一个装饰类,增加了scheduler
。- 调用
subscribe()
方法,在这个方法调用后,subscribeActual()
被调用,才真正执行了IoSchduler
中的createWorker()
创建线程并运行,最终将上游Observable
的subscribe()
方法调度到新创建的线程中运行。
现在了解了被观察者执行线程是如何被调度到指定线程中执行的,但很多情况下,我们希望观察者(事件下游)处理事件最好在UI线程执行,比如更新UI操作等。下面分析下游何时调度,如何调度由于篇幅问题。
三、Rxjava如何对观察者线程进行调度
简单的例子
1 private void doSomeWork() {
2 Observable.create(new ObservableOnSubscribe<String>() {
3 @Override
4 public void subscribe(ObservableEmitter<String> e) throws Exception {
5 Log.i("lx", " subscribe: " + Thread.currentThread().getName());
6 e.onNext("a");
7 e.onComplete();
8 }
9 }).subscribeOn(Schedulers.io())
10 .observeOn(AndroidSchedulers.mainThread())
11 .subscribe(new Observer<String>() {
12 @Override
13 public void onSubscribe(Disposable d) {
14 Log.i("lx", " onSubscribe: " + Thread.currentThread().getName());
15 }
16 @Override
17 public void onNext(String str) {
18 Log.i("lx", " onNext: " + Thread.currentThread().getName());
19 }
20 @Override
21 public void onError(Throwable e) {
22 Log.i("lx", " onError: " + Thread.currentThread().getName());
23 }
24 @Override
25 public void onComplete() {
26 Log.i("lx", " onComplete: " + Thread.currentThread().getName());
27 }
28 });
29 }
看看运行结果:
1 com.rxjava2.android.samples I/lx: onSubscribe: main
2 com.rxjava2.android.samples I/lx: subscribe: RxCachedThreadScheduler-1
3 com.rxjava2.android.samples I/lx: onNext: main
4 com.rxjava2.android.samples I/lx: onComplete: main
从结果可以看出,事件的生产线程运行在RxCachedThreadScheduler-1中,而事件的消费线程则被调度到了main线程中。关键代码是因为这句.observeOn(AndroidSchedulers.mainThread())
。 下面我们着重分析下这句代码都做了哪些事情。
AndroidSchedulers.mainThread()
先来看看AndroidSchedulers.mainThread()
是什么?贴代码
1 /** A {@link Scheduler} which executes actions on the Android main thread. */
2 public static Scheduler mainThread() {
3 return RxAndroidPlugins.onMainThreadScheduler(MAIN_THREAD);
4 }
注释已经说的很明白了,是一个在主线程执行任务的scheduler,接着看
1 private static final Scheduler MAIN_THREAD = RxAndroidPlugins.initMainThreadScheduler(
2 new Callable<Scheduler>() {
3 @Override public Scheduler call() throws Exception {
4 return MainHolder.DEFAULT;
5 }
6 });
7
8 public static Scheduler initMainThreadScheduler(Callable<Scheduler> scheduler) {
9 if (scheduler == null) {
10 throw new NullPointerException("scheduler == null");
11 }
12 Function<Callable<Scheduler>, Scheduler> f = onInitMainThreadHandler;
13 if (f == null) {
14 return callRequireNonNull(scheduler);
15 }
16 return applyRequireNonNull(f, scheduler);
17 }
代码很简单,这个AndroidSchedulers.mainThread()
想当于new HandlerScheduler(new Handler(Looper.getMainLooper()))
,原来是利用Android
的Handler
来调度到main
线程的。
我们再看看HandlerScheduler
,它与我们上节分析的IOScheduler
类似,都是继承自Scheduler
,所以AndroidSchedulers.mainThread()
其实就是是创建了一个运行在main thread
上的scheduler。
好了,我们再回过头来看observeOn
方法。
observeOn
1 public final Observable<T> observeOn(Scheduler scheduler) {
2 return observeOn(scheduler, false, bufferSize());
3 }
4
5 public final Observable<T> observeOn(Scheduler scheduler, boolean delayError, int bufferSize) {
6 ObjectHelper.requireNonNull(scheduler, "scheduler is null");
7 ObjectHelper.verifyPositive(bufferSize, "bufferSize");
8 return RxJavaPlugins.onAssembly(new ObservableObserveOn<T>(this, scheduler, delayError, bufferSize));
9 }
10
重点是这个new ObservableObserveOn
,看名字是不是有种似成相识的感觉,还记得上篇的ObservableSubscribeOn
吗? 它俩就是亲兄弟,是继承自同一个父类。
重点还是这个方法,我们前文已经提到了,Observable的subscribe方法最终都是调用subscribeActual
方法。下面看看这个方法的实现:
1 @Override
2 protected void subscribeActual(Observer<? super T> observer) {
3 // scheduler 就是前面提到的 HandlerScheduler,所以进入else分支
4 if (scheduler instanceof TrampolineScheduler) {
5 source.subscribe(observer);
6 } else {
7 // 创建 HandlerWorker
8 Scheduler.Worker w = scheduler.createWorker();
9 // 调用上游Observable的subscribe,将订阅向上传递
10 source.subscribe(new ObserveOnObserver<T>(observer, w, delayError, bufferSize));
11 }
12 }
从上面代码可以看到使用了ObserveOnObserver
类对observer
进行装饰,好了,我们再来看看ObserveOnObserver
。
我们已经知道了,事件源发射的事件,是通过observer的onNext
,onError
,onComplete
发射到下游的。所以看看ObserveOnObserver
的这三个方法是如何实现的。
由于篇幅问题,我们只分析onNext
方法,onError
和onComplete
方法有兴趣的同学可以自己分析下。
1 @Override
2 public void onNext(T t) {
3 if (done) {
4 return;
5 }
6
7 // 如果是非异步方式,将上游发射的时间加入到队列
8 if (sourceMode != QueueDisposable.ASYNC) {
9 queue.offer(t);
10 }
11 schedule();
12 }
13
14 void schedule() {
15 // 保证只有唯一任务在运行
16 if (getAndIncrement() == 0) {
17 // 调用的就是HandlerWorker的schedule方法
18 worker.schedule(this);
19 }
20 }
21
22 @Override
23 public Disposable schedule(Runnable run, long delay, TimeUnit unit) {
24 if (run == null) throw new NullPointerException("run == null");
25 if (unit == null) throw new NullPointerException("unit == null");
26
27 if (disposed) {
28 return Disposables.disposed();
29 }
30
31 run = RxJavaPlugins.onSchedule(run);
32
33 ScheduledRunnable scheduled = new ScheduledRunnable(handler, run);
34
35 Message message = Message.obtain(handler, scheduled);
36 message.obj = this; // Used as token for batch disposal of this worker's runnables.
37
38 handler.sendMessageDelayed(message, Math.max(0L, unit.toMillis(delay)));
39
40 // Re-check disposed state for removing in case we were racing a call to dispose().
41 if (disposed) {
42 handler.removeCallbacks(scheduled);
43 return Disposables.disposed();
44 }
45
46 return scheduled;
47 }
schedule
方法将传入的run
调度到对应的handle
所在的线程来执行,这个例子里就是有main
线程来完成。 再回去看看前面传入的run
吧。
回到ObserveOnObserver
中的run
方法:
1 @Override
2 public void run() {
3 // 此例子中代码不会进入这个分支,至于这个drainFused是什么,后面章节再讨论。
4 if (outputFused) {
5 drainFused();
6 } else {
7 drainNormal();
8 }
9 }
10
11 void drainNormal() {
12 int missed = 1;
13
14 final SimpleQueue<T> q = queue;
15 final Observer<? super T> a = actual;
16
17 for (;;) {
18 if (checkTerminated(done, q.isEmpty(), a)) {
19 return;
20 }
21
22 for (;;) {
23 boolean d = done;
24 T v;
25
26 try {
27 // 从队列中queue中取出事件
28 v = q.poll();
29 } catch (Throwable ex) {
30 Exceptions.throwIfFatal(ex);
31 s.dispose();
32 q.clear();
33 a.onError(ex);
34 worker.dispose();
35 return;
36 }
37 boolean empty = v == null;
38
39 if (checkTerminated(d, empty, a)) {
40 return;
41 }
42
43 if (empty) {
44 break;
45 }
46 //调用下游observer的onNext将事件v发射出去
47 a.onNext(v);
48 }
49
50 missed = addAndGet(-missed);
51 if (missed == 0) {
52 break;
53 }
54 }
55 }
至此我们明白了RXjava是如何调度消费者线程了。
消费者线程调度流程概括
Rxjava调度消费者现在的流程,以observeOn(AndroidSchedulers.mainThread())
为例。
-
AndroidSchedulers.mainThread()
先创建一个包含handler
的Scheduler
, 这个handler
是主线程的handler
。 -
observeOn
方法创建ObservableObserveOn
,它是上游Observable
的一个装饰类,其中包含前面创建的Scheduler
和bufferSize
等. - 当订阅方法
subscribe
被调用后,ObservableObserveOn
的subscribeActual
方法创建Scheduler.Worker
并调用上游的subscribe
方法,同时将自身接收的参数'observer'用装饰类ObserveOnObserver
装饰后传递给上游。 - 当上游调用被
ObserveOnObserver
的onNext
、onError
和onComplete
方法时,ObserveOnObserver
将上游发送的事件通通加入到队列queue
中,然后再调用scheduler
将处理事件的方法调度到对应的线程中(本例会调度到main thread)。 处理事件的方法将queue
中保存的事件取出来,调用下游原始的observer再发射出去。 - 经过以上流程,下游处理事件的消费者线程就运行在了
observeOn
调度后的thread中。
总结
经过前面两节的分析,我们已经明白了Rxjava是如何对线程进行调度的。
- Rxjava的
subscribe
方法是由下游一步步向上游进行传递的。会调用上游的subscribe
,直到调用到事件源。
如: source.subscribe(xxx);
而上游的source
往往是经过装饰后的Observable
, Rxjava就是利用ObservableSubscribeOn
将subscribe
方法调度到了指定线程运行,生产者线程最终会运行在被调度后的线程中。但多次调用subscribeOn
方法会怎么样呢? 我们知道因为subscribe
方法是由下而上传递的,所以事件源的生产者线程最终都只会运行在第一次执行subscribeOn
所调度的线程中,换句话就是多次调用subscribeOn
方法,只有第一次有效。
- Rxjava发射事件是由上而下发射的,上游的
onNext
、onError
、onComplete
方法会调用下游传入的observer的对应方法。往往下游传递的observer对象也是经过装饰后的observer对象。Rxjava就是利用ObserveOnObserver
将执行线程调度后,再调用下游对应的onNext
、onError
、onComplete
方法,这样下游消费者就运行再了指定的线程内。 那么多次调用observeOn
调度不同的线程会怎么样呢? 因为事件是由上而下发射的,所以每次用observeOn
切换完线程后,对下游的事件消费都有效,比如下游的map操作符。最终的事件消费线程运行在最后一个observeOn
切换后线程中。 - 另外通过源码可以看到
onSubscribe
运行在subscribe
的调用线程中,这个就不具体分析了。