Android主流三方库源码分析：RxJava，android基础

m0_60667406

于 2024-04-10 02:59:27 发布

阅读量820

点赞数 13

分类专栏： 2024年程序员学习文章标签： android rxjava echarts

本文链接：https://blog.csdn.net/m0_60667406/article/details/137575827

版权

2024年程序员学习专栏收录该内容

277 篇文章 0 订阅

订阅专栏

public static boolean isDisposed(Disposable d) {

// 1

return d == DISPOSED;

}

public static boolean set(AtomicReference field, Disposable d) {

for (;😉 {

Disposable current = field.get();

if (current == DISPOSED) {

if (d != null) {

d.dispose();

}

return false;

}

// 2

if (field.compareAndSet(current, d)) {

if (current != null) {

current.dispose();

}

return true;

}

…

public static boolean dispose(AtomicReference field) {

Disposable current = field.get();

Disposable d = DISPOSED;

if (current != d) {

// …

current = field.getAndSet(d);

if (current != d) {

if (current != null) {

current.dispose();

}

return true;

}

return false;

}

…

}

复制代码

DisposableHelper是一个枚举类，内部只有一个值即DISPOSED, 从上面的分析可知它就是用来标记事件流被切断（废弃）状态的。先看到注释2和注释3处的代码field.compareAndSet(current, d)和field.getAndSet(d)，这里使用了原子引用AtomicReference内部包装的CAS方法处理了标志Disposable的并发读写问题。最后看到注释3处，将我们传入的CreateEmitter这个原子引用类保存的Dispable状态和DisposableHelper内部的DISPOSED进行比较，如果相等，就证明数据流被切断了。为了更进一步理解Disposed的作用，再来看看CreateEmitter中剩余的关键方法。

2.4.6、CreateEmitter

@Override

public void onNext(T t) {

…

// 1

if (!isDisposed()) {

observer.onNext(t);

}

@Override

public void onError(Throwable t) {

if (!tryOnError(t)) {

// 2

RxJavaPlugins.onError(t);

}

@Override

public boolean tryOnError(Throwable t) {

…

// 3

if (!isDisposed()) {

try {

observer.onError(t);

} finally {

// 4

dispose();

}

return true;

}

return false;

}

@Override

public void onComplete() {

// 5

if (!isDisposed()) {

try {

observer.onComplete();

} finally {

// 6

dispose();

}

复制代码

在注释1、3、5处，onNext()和onError()、onComplete()方法首先都会判断事件流是否被切断，如果事件流此时被切断了，那么onNext()和onComplete()则会退出方法体，不做处理，onError()则会执行到RxJavaPlugins.onError(t)这句代码，内部会直接抛出异常，导致崩溃。如果事件流没有被切断，那么在onError()和onComplete()内部最终会调用到注释4、6处的这句dispose()代码，将事件流进行切断，由此可知，onError()和onComplete()只能调用一个，如果先执行的是onComplete()，再调用onError()的话就会导致异常崩溃。

三、RxJava的线程切换

首先给出RxJava线程切换的例子：

Observable.create(new ObservableOnSubscribe() {

@Override

public voidsubscribe(ObservableEmitteremitter) throws Exception {

emitter.onNext(“1”);

emitter.onNext(“2”);

emitter.onNext(“3”);

emitter.onComplete();

}

})

.subscribeOn(Schedulers.io())

.observeOn(AndroidSchedulers.mainThread())

.subscribe(new Observer() {

@Override

public void onSubscribe(Disposable d) {

Log.d(TAG, “onSubscribe”);

}

@Override

public void onNext(String s) {

Log.d(TAG, "onNext : " + s);

}

@Override

public void onError(Throwable e) {

Log.d(TAG, "onError : " +e.toString());

}

@Override

public void onComplete() {

Log.d(TAG, “onComplete”);

}

});

复制代码

可以看到，RxJava的线程切换主要分为subscribeOn()和observeOn()方法，首先，来分析下subscribeOn()方法。

1、subscribeOn(Schedulers.io())

在Schedulers.io()方法中，我们需要先传入一个Scheduler调度类，这里是传入了一个调度到io子线程的调度类，我们看看这个Schedulers.io()方法内部是怎么构造这个调度器的。

2、Schedulers#io()

static final Scheduler IO;

…

public static Scheduler io() {

// 1

return RxJavaPlugins.onIoScheduler(IO);

}

static {

…

// 2

IO = RxJavaPlugins.initIoScheduler(new IOTask());

}

static final class IOTask implements Callable {

@Override

public Scheduler call() throws Exception {

// 3

return IoHolder.DEFAULT;

}

static final class IoHolder {

// 4

static final Scheduler DEFAULT = new IoScheduler();

}

复制代码

Schedulers这个类的代码很多，这里我只拿出有关Schedulers.io这个方法涉及的逻辑代码进行讲解。首先，在注释1处，同前面分析的订阅流程的处理一样，只是一个处理hook的逻辑，最终返回的还是传入的这个IO对象。再看到注释2处，在Schedulers的静态代码块中将IO对象进行了初始化，其实质就是新建了一个IOTask的静态内部类，在IOTask的call方法中，也就是注释3处，可以了解到使用了静态内部类的方式把创建的IOScheduler对象给返回出去了。绕了这么大圈子，Schedulers.io方法其实质就是返回了一个IOScheduler对象。

3、Observable#subscribeOn()

public final Observable subscribeOn(Scheduler scheduler) {

…

return RxJavaPlugins.onAssembly(new ObservableSubscribeOn(this, scheduler));

}

复制代码

在subscribeOn()方法里面，又将ObservableCreate包装成了一个ObservableSubscribeOn对象。我们关注到ObservableSubscribeOn类。

4、ObservableSubscribeOn

public final class ObservableSubscribeOn extends AbstractObservableWithUpstream<T, T> {

final Scheduler scheduler;

public ObservableSubscribeOn(ObservableSource source, Scheduler scheduler) {

// 1

super(source);

this.scheduler = scheduler;

}

@Override

public void subscribeActual(final Observer<? super T> observer) {

// 2

final SubscribeOnObserver parent = new SubscribeOnObserver(observer);

// 3

observer.onSubscribe(parent);

// 4

parent.setDisposable(scheduler.scheduleDirect(new SubscribeTask(parent)));

}

…

}

复制代码

首先，在注释1处，将传进来的source和scheduler保存起来。接着，等到实际订阅的时候，就会执行到这个subscribeActual方法，在注释2处，将我们自定义的Observer包装成了一个SubscribeOnObserver对象。在注释3处，通知观察者订阅了被观察者。在注释4处，内部先创建了一个SubscribeTask对象，来看看它的实现。

5、ObservableSubscribeOn#SubscribeTask

final class SubscribeTask implements Runnable {

private final SubscribeOnObserver parent;

SubscribeTask(SubscribeOnObserver parent) {

this.parent = parent;

}

@Override

public void run() {

source.subscribe(parent);

}

复制代码

SubscribeTask是ObservableSubscribeOn的内部类，它实质上就是一个任务类，在它的run方法中会执行到source.subscribe(parent)的订阅方法，这个source其实就是我们在ObservableSubscribeOn构造方法中传进来的ObservableCreate对象。接下来看看scheduler.scheduleDirect()内部的处理。

6、Scheduler#scheduleDirect()

public Disposable scheduleDirect(@NonNull Runnable run) {

return scheduleDirect(run, 0L, TimeUnit.NANOSECONDS);

}

public Disposable scheduleDirect(@NonNull Runnable run, long delay, @NonNull TimeUnit unit) {

// 1

final Worker w = createWorker();

// 2

final Runnable decoratedRun = RxJavaPlugins.onSchedule(run);

// 3

DisposeTask task = new DisposeTask(decoratedRun, w);

// 4

w.schedule(task, delay, unit);

return task;

}

复制代码

这里最后会执行到上面这个scheduleDirect()重载方法。首先，在注释1处，会调用createWorker()方法创建一个工作者对象Worker，它是一个抽象类，这里的实现类就是IoScheduler，下面，我们看看IoScheduler类的createWorker()方法。

6.1、IOScheduler#createWorker()

final AtomicReference pool;

…

public IoScheduler(ThreadFactory threadFactory) {

this.threadFactory = threadFactory;

this.pool = new AtomicReference(NONE);

start();

}

…

@Override

public Worker createWorker() {

// 1

return new EventLoopWorker(pool.get());

}

static final class EventLoopWorker extends Scheduler.Worker {

…

EventLoopWorker(CachedWorkerPool pool) {

this.pool = pool;

this.tasks = new CompositeDisposable();

// 2

this.threadWorker = pool.get();

}

复制代码

首先，在注释1处调用了pool.get()这个方法，pool是一个CachedWorkerPool类型的原子引用对象，它的作用就是用于缓存工作者对象Worker的。然后，将得到的CachedWorkerPool传入新创建的EventLoopWorker对象中。重点关注一下注释2处，这里将CachedWorkerPool缓存的threadWorker对象保存起来了。

下面，我们继续分析3.6处代码段的注释2处的代码，这里又是一个关于hook的封装处理，最终还是返回的当前的Runnable对象。在注释3处新建了一个切断任务DisposeTask将decoratedRun和w对象包装了起来。最后在注释4处调用了工作者的schedule()方法。下面我们来分析下它内部的处理。

6.2、IoScheduler#schedule()

@Override

public Disposable schedule(@NonNull Runnableaction, long delayTime, @NonNull TimeUnit unit){

…

return threadWorker.scheduleActual(action,delayTime, unit, tasks);

}

复制代码

内部调用了threadWorker的scheduleActual()方法，实际上是调用到了父类NewThreadWorker的scheduleActual()方法，我们继续看看NewThreadWorker的scheduleActual()方法中做的事情。

6.3、NewThreadWorker#scheduleActual()

public NewThreadWorker(ThreadFactory threadFactory) {

executor = SchedulerPoolFactory.create(threadFactory);

}

@NonNull

public ScheduledRunnable scheduleActual(final Runnable run, long delayTime, @NonNull TimeUnit unit, @Nullable DisposableContainer parent) {

Runnable decoratedRun = RxJavaPlugins.onSchedule(run);

// 1

ScheduledRunnable sr = new ScheduledRunnable(decoratedRun, parent);

if (parent != null) {

if (!parent.add(sr)) {

return sr;

}

Future<?> f;

try {

// 2

if (delayTime <= 0) {

// 3

f = executor.submit((Callable)sr);

} else {

// 4

f = executor.schedule((Callable)sr, delayTime, unit);

}

sr.setFuture(f);

} catch (RejectedExecutionException ex) {

if (parent != null) {

parent.remove(sr);

}

RxJavaPlugins.onError(ex);

}

return sr;

}

复制代码

在NewThreadWorker的scheduleActual()方法的内部，在注释1处首先会新建一个ScheduledRunnable对象，将Runnable对象和parent包装起来了，这里parent是一个DisposableContainer对象，它实际的实现类是CompositeDisposable类，它是一个保存所有事件流是否被切断状态的容器，其内部的实现是使用了RxJava自己定义的一个简单的OpenHashSet类进行存储。最后注释2处，判断是否设置了延迟时间，如果设置了，则调用线程池的submit()方法立即进行线程切换，否则，调用schedule()方法进行延时执行线程切换。

7、为什么多次执行subscribeOn()，只有第一次有效？

从上面的分析，我们可以很容易了解到被观察者被订阅时是从最外面的一层（ObservableSubscribeOn）通知到里面的一层（ObservableOnSubscribe），当连续执行了到多次subscribeOn()的时候，其实就是先执行倒数第一次的subscribeOn()方法，直到最后一次执行的subscribeOn()方法，这样肯定会覆盖前面的线程切换。