RxJava2源码解析

基础解析

咱们看下RxJava最简单的写法java

Observable.create(new ObservableOnSubscribe<String>() {
            @Override
            public void subscribe(ObservableEmitter<String> emitter) throws Exception {
                emitter.onNext("1");
                emitter.onComplete();
            }
        }).subscribe(new Observer<String>() {
            @Override
            public void onSubscribe(Disposable d) {
                System.out.println("onSubscribe");
            }

            @Override
            public void onNext(String s) {
                System.out.println(s);
            }

            @Override
            public void onError(Throwable e) {
                System.out.println("onError"+e.getLocalizedMessage());
            }

            @Override
            public void onComplete() {
                System.out.println("onComplete");
            }
        })

很简单的3个步骤:微信

  1. 建立 Observable :被观察者
  2. 建立 Observer :观察者
  3. 经过 subscribe() 方法创建订阅关系

一个个来看ide

被观察者的建立

public static <T> Observable<T> create(ObservableOnSubscribe<T> source) {
    ObjectHelper.requireNonNull(source, "source is null");
    //建立了一个ObservableCreate类,里面包装了咱们传入的source参数
    return RxJavaPlugins.onAssembly(new ObservableCreate<T>(source));
}

public final class ObservableCreate<T> extends Observable<T> {
    final ObservableOnSubscribe<T> source;

    public ObservableCreate(ObservableOnSubscribe<T> source) {
        this.source = source;
    }

观察者的建立

这里很简单,只是经过new方法生成了一个简单的Observer对象。函数

订阅

订阅是经过subscribe方法来执行的,咱们来跟踪一下,这个方法是属于Observable类的oop

public final void subscribe(Observer<? super T> observer) {
    //校验观察者不为空
    ObjectHelper.requireNonNull(observer, "observer is null");
    try {
        observer = RxJavaPlugins.onSubscribe(this, observer);

        ObjectHelper.requireNonNull(observer, "Plugin returned null Observer");
        //调用subscribeActual方法,而后入参是observer(被观察者)。这个方法是抽象方法,具体的实现是交给子类的
        subscribeActual(observer);
    } catch (NullPointerException e) { // NOPMD
        throw e;
    } catch (Throwable e) {
        Exceptions.throwIfFatal(e);
        // can't call onError because no way to know if a Disposable has been set or not
        // can't call onSubscribe because the call might have set a Subscription already
        RxJavaPlugins.onError(e);

        NullPointerException npe = new NullPointerException("Actually not, but can't throw other exceptions due to RS");
        npe.initCause(e);
        throw npe;
    }
}
    /**
     * Operator implementations (both source and intermediate) should implement this method that
     * performs the necessary business logic.
     * <p>There is no need to call any of the plugin hooks on the current Observable instance or
     * the Subscriber.
     * @param observer the incoming Observer, never null
     */
    protected abstract void subscribeActual(Observer<? super T> observer);

最终经过 subscribeActual(observer) 来实现功能,而这个方法是有具体的子类去实现的。从第一步中咱们经过Observable.create()来生成的被观察者。里面最终的生成的是 ObservableCreate 这个类。也就是说,这个subscribeActual(observer) 方法是由 ObservableCreate 这个类去实现的,咱们去里面找一下。测试

@Override
protected void subscribeActual(Observer<? super T> observer) {
    //这里将咱们传入的被观察者进行了一层封装,里面实现了ObservableEmitter<T>, Disposable等接口->装饰者模式
    CreateEmitter<T> parent = new CreateEmitter<T>(observer);
    //调用被观察者的onSubscribe方法(这里很神奇,调起者是observer,而不是被订阅者,是为了兼容Rxajva1么?)
    observer.onSubscribe(parent);
    try {
        //这里的source就是咱们本身写的那个ObservableOnSubscribe了,调用了里面的subscriber方法,而后参数是封装后的观察者。
        source.subscribe(parent);
    } catch (Throwable ex) {
        Exceptions.throwIfFatal(ex);
        parent.onError(ex);
    }
}

Observable.create(new ObservableOnSubscribe<String>() {
            //看到了哈,实际是执行的这个方法,这里面的emitter是咱们封装以后的CreateEmitter,那么这里面的onNext(),onComplete()又是谁呢?
            @Override
            public void subscribe(ObservableEmitter<String> emitter) throws Exception {
                emitter.onNext("1");
                emitter.onComplete();
            }
        })

咱们如今回到咱们封装生成的 CreateEmitter 这个类ui

static final class CreateEmitter<T>
extends AtomicReference<Disposable>
implements ObservableEmitter<T>, Disposable {
    private static final long serialVersionUID = -3434801548987643227L;
    final Observer<? super T> observer;
    //定义的观察者
    CreateEmitter(Observer<? super T> observer) {
        this.observer = observer;
    }
    
    @Override
    public void onNext(T t) {
        if (t == null) {
            onError(new NullPointerException("onNext called with null. Null values are generally not allowed in 2.x operators and sources."));
            return;
        }
        //调用的是观察者的onNext()方法
        if (!isDisposed()) {
            observer.onNext(t);
        }
    }

    @Override
    public void onComplete() {
        if (!isDisposed()) {
            try {
                //调用的是观察者的onComplete()方法
                observer.onComplete();
            } finally {
                //执行完onComplete()方法后要取消订阅
                dispose();
            }
        }
    }
    .....
}

到这里为知,最简单的一个流程基本已经走通了。。this

高级用法

线程切换

下层切换

RxJava中咱们使用的最多的应该就是进行线程切换了吧?经过 observeOn() 方法来进行线程的随意切换,舒舒服服,不再用进行恶心的线程处理了。线程

Observable.create(new ObservableOnSubscribe<String>() {
            @Override
            public void subscribe(ObservableEmitter<String> emitter) throws Exception {
                emitter.onNext("1");
                emitter.onComplete();
            }
        }).observeOn(Schedulers.io())

observeOn() 方法是属于Observable这个类的。咱们跟踪进去这个方法去看看。code

public final Observable<T> observeOn(Scheduler scheduler, boolean delayError, int bufferSize) {
        //进行空校验
        ObjectHelper.requireNonNull(scheduler, "scheduler is null");
        ObjectHelper.verifyPositive(bufferSize, "bufferSize");
        return RxJavaPlugins.onAssembly(new ObservableObserveOn<T>(this, scheduler, delayError, bufferSize));
    }

这里建立了一个 ObservableObserveOn 对象,因此和以前基础里面将的同样,当调用 subscribe() 方法的时候,会先调用观察者的 onSubscribe() 方法,而后经过subscribe的层层处理,调用这个被观察者里面的 subscribeActual() 方法。

@Override
protected void subscribeActual(Observer<? super T> observer) {
    if (scheduler instanceof TrampolineScheduler) {//若是传入的scheduler是TrampolineScheduler,那么线程不须要切换,直接调用subscribe方法便可
        source.subscribe(observer);
    } else {
        //根据传入的scheduler,建立Worker
        Scheduler.Worker w = scheduler.createWorker();
        //将传入的observer进行包装,包装为ObserveOnObserver类。
        source.subscribe(new ObserveOnObserver<T>(observer, w, delayError, bufferSize));
    }
}

这里能够依据基础篇的进行整理一下,这里将观察者进行了一层包装,也就是咱们的观察者由原来的observaer变为了ObserveOnObserver对象。而被观察者仍是以前的ObservableCreate(注意,这里只是依据基础中.create()建立的类,因此是ObservableCreate,若是是其余方式建立的被观察者,那么这里可能就是另外一个具体的实现类了),并未改变。以前咱们讲过,当调用subscribe方法的onNext(),onComplete()方法,实际上是调用的观察者的方法。咱们如今看一下ObserveOnObserver的onNext和onComplete方法又是作了什么神奇的操做。

@Override
public void onNext(T t) {
    if (done) {//若是已经完成,直接返回
        return;
    }
    if (sourceMode != QueueDisposable.ASYNC) {
        //将onNext的数据放入队列queue
        queue.offer(t);
    }
    //进行线程切换
    schedule();
}

void schedule() {
    if (getAndIncrement() == 0) {
        //调用了worker的方法,这里经过调用线程池,调用了自身的run方法
        worker.schedule(this);
    }
}

这里咱们使用的是IO线程,那么在 scheduler.createWorker() 中的生成worker时

@NonNull
@Override
public Worker createWorker() {
    return new EventLoopWorker(pool.get());
}

那么跟到这个类里面的 schedule 方法

@Override
public Disposable schedule(@NonNull Runnable action, long delayTime, @NonNull TimeUnit unit) {
    if (tasks.isDisposed()) {
        // don't schedule, we are unsubscribed
        return EmptyDisposable.INSTANCE;
    }
    //这里调用了线程worker的scheduleActual方法,并把Runable对象传进去
    return threadWorker.scheduleActual(action, delayTime, unit, tasks);
}

public ScheduledRunnable scheduleActual(final Runnable run, long delayTime, @NonNull TimeUnit unit, @Nullable DisposableContainer parent) {
        //留下钩子
        Runnable decoratedRun = RxJavaPlugins.onSchedule(run);
        ScheduledRunnable sr = new ScheduledRunnable(decoratedRun, parent);
        ....
        Future<?> f;
        try {
            if (delayTime <= 0) {
                //在线程池中调用封装以后的Runnable
                f = executor.submit((Callable<Object>)sr);
            } else {
                f = executor.schedule((Callable<Object>)sr, delayTime, unit);
            }
            sr.setFuture(f);
        } catch (RejectedExecutionException ex) {
            if (parent != null) {
                parent.remove(sr);
            }
            RxJavaPlugins.onError(ex);
        }
        return sr;
    }

能够看到,其实最终是经过线程池调用了 ObserveOnObserver 自己,这个类实现了 Runnable 接口,咱们看一下run方法里面作了什么。

@Override
public void run() {
    if (outputFused) {
        drainFused();
    } else {
        drainNormal();
    }
}
//具体的操做
void drainNormal() {
     int missed = 1;
     //被观察者onNext发送的数据队列
     final SimpleQueue<T> q = queue;
     //实际的观察者
     final Observer<? super T> a = downstream;
     for (;;) {
         //检测是否有异常信息
         if (checkTerminated(done, q.isEmpty(), a)) {
             return;
         }
         //遍历
         for (;;) {
             boolean d = done;
             T v;
             //取出队列中的数据
             try {
                 v = q.poll();
             } catch (Throwable ex) {
                 //发生异常,则直接调用dispose()和onError()方法
                 Exceptions.throwIfFatal(ex);
                 disposed = true;
                 upstream.dispose();
                 q.clear();
                 a.onError(ex);
                 worker.dispose();
                 return;
             }
             ....
             //调用实际的观察者的onNext()方法
             a.onNext(v);
         }
         ...
     }
 }

由于这个操做最终是在scheduler.createWorker()建立的地方进行了处理,才实现了对于以后代码处理都在io线程中进行了调用。从而实现线程的切换功能。这里咱们对以前的测试代码流程作一个总结。

先看一下对于观察者的onSubscribe()方法的调用流程:

企业微信截图_15789953155616

这里面咱们本身定义的观察者经过subscribe()方法层层往上调用,最后调用了咱们定义的被观察者里面的onSubscribe方法,再一层层的往下调用,最后到咱们本身定义的onSubscribe()方法,里面不多有线程的切换处理,因此这段代码在哪儿执行,那么这段代码在那里执行,这个onSubscribe()方法就是在哪一个线程执行。

继续,咱们看一下onNext()方法

企业微信截图_1578995357300

上层切换

除了 observeOn 方法来处理咱们操做流的下层线程处理以外,咱们也能够经过 subscribeOn 方法来进行对上层流的线程处理。

测试用代码:

Observable.create(new ObservableOnSubscribe<String>() {
            @Override
            public void subscribe(ObservableEmitter<String> emitter) throws Exception {
                emitter.onNext("1");
                emitter.onComplete();
            }
}).subscribeOn(Schedulers.io())

如今咱们跟踪进 subscribeOn 方法

public final Observable<T> subscribeOn(Scheduler scheduler) {
    ObjectHelper.requireNonNull(scheduler, "scheduler is null");
    //
    return RxJavaPlugins.onAssembly(new ObservableSubscribeOn<T>(this, scheduler));
}

这里看到,跟咱们基础篇里面的 create() 方法有殊途同归之妙,这里面生成了一个ObservableSubscribeOn类,这个类也是继承Observable类的,咱们跟踪进去看一下。

public final class ObservableSubscribeOn<T> extends AbstractObservableWithUpstream<T, T> {
    final Scheduler scheduler;

    public ObservableSubscribeOn(ObservableSource<T> source, Scheduler scheduler) {
        super(source);
        this.scheduler = scheduler;
    }

    @Override
    public void subscribeActual(final Observer<? super T> observer) {
        final SubscribeOnObserver<T> parent = new SubscribeOnObserver<T>(observer);
        //调用订阅者的onSubscribe方法,这里的线程还未进行切换
        observer.onSubscribe(parent);
        //进行线程的切换处理
        //1.创造一个SubscribeTask的Runable方法
        //2.经过scheduler的scheduleDirect进行线程的切换
        //3.经过parent.setDisposable来进行Disposable的切换
        parent.setDisposable(scheduler.scheduleDirect(new SubscribeTask(parent)));
    }

看起来是否是很像?在基础篇咱们知道了,这个 subscribeActual 方法里面的参数就是咱们的观察者。

咱们看一下里面和以前分析所不一样的地方,也就是线程的切换

final class SubscribeTask implements Runnable {
    ...
    @Override
    public void run() {
        //source是咱们上一层的被观察者,parent是包装以后的观察者.
        //因此会在相关的worker里面调用source的subscribe方法,
        //即上层的数据调用已经在woker里面了(若是是IoScheduler,那么这里就是在RxCachedThreadScheduler线程池调用了这个方法 )
        source.subscribe(parent);
    }
}

而后看一下这里面最重要的 scheduler.scheduleDirect 这个方法

@NonNull
    public Disposable scheduleDirect(@NonNull Runnable run) {
        return scheduleDirect(run, 0L, TimeUnit.NANOSECONDS);
    }
    @NonNull
    public Disposable scheduleDirect(@NonNull Runnable run, long delay, @NonNull TimeUnit unit) {
        //建立一个Worker,这个是有具体的实现类来实现的,好比咱们的IOScheduler,ImmediateThinScheduler等,具体要看咱们切换传参
        final Worker w = createWorker();
        final Runnable decoratedRun = RxJavaPlugins.onSchedule(run);
        DisposeTask task = new DisposeTask(decoratedRun, w);
        w.schedule(task, delay, unit);
        return task;
    }

这里咱们对上层切换的流程作一个总结:当调用 subscribeOn 方法的时候,会在建立的调度器中来执行被观察者的执行代码,从而实现了对上层的线程切换功能。

先看一下测试代码中的onNext()方法的调用流程:

企业微信截图_15789953986768

汇总

其实对于线程的切换,主要是根据里面传递的线程切换函数,将上游或者下游的代码在指定的线程里面去执行来实现。

image

本文由 开了肯 发布!

相关文章
相关标签/搜索