RxJava之create操作符源码解析

转载请以链接形式标明出处:
本文出自:103style的博客

本文基于 RxJava 2.x 版本

create操作符例子:
Observable
        .create(new ObservableOnSubscribe<Object>() {
            @Override
            public void subscribe(ObservableEmitter<Object> emitter) throws Exception {

            }
        })
        .subscribe(new Observer<Object>() {
            @Override
            public void onSubscribe(Disposable d) {

            }

            @Override
            public void onNext(Object o) {

            }

            @Override
            public void onError(Throwable e) {

            }

            @Override
            public void onComplete() {

            }
        });

首先我们看create 方法:

public static <T> Observable<T> create(ObservableOnSubscribe<T> source) {
    ObjectHelper.requireNonNull(source, "source is null");
    return RxJavaPlugins.onAssembly(new ObservableCreate<T>(source));
}

RxJavaPlugins 类的 onAssembly 方法:

static volatile Function<? super Observable, ? extends Observable> onObservableAssembly;

public static <T> Observable<T> onAssembly(@NonNull Observable<T> source) {
    Function<? super Observable, ? extends Observable> f = onObservableAssembly;
    if (f != null) {
        return apply(f, source);
    }
    return source;
}

在源码中查看引用可知 onObservableAssembly 只有在测试的时候才不为 null
所以Observable.create(ObservableOnSubscribe<T> source)实际上就是返回了 ObservableCreate对象


ObservableCreate 类,可以看到 ObservableCreateObservable 的子类,并实现了父类的 subscribeActual 方法。

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

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

    @Override
    protected void subscribeActual(Observer<? super T> observer) {...}
    ...
}

然后我们看subscribe方法: 实际上是调用了 Observable 的抽象方法 subscribeActual(observer);

public final void subscribe(Observer<? super T> observer) {
    ...
    subscribeActual(observer);
    ...
}

protected abstract void subscribeActual(Observer<? super T> observer);

又因为 create操作符返回的 ObservableCreateObservable 的子类,
所以实际上调用的是ObservableCreatesubscribeActual(observer);

具体可阅读 Observable subscribe流程介绍


ObservableCreatesubscribeActual(observer)方法:

@Override
protected void subscribeActual(Observer<? super T> observer) {
    CreateEmitter<T> parent = new CreateEmitter<T>(observer);
    observer.onSubscribe(parent);

    try {
        source.subscribe(parent);
    } catch (Throwable ex) {
        Exceptions.throwIfFatal(ex);
        parent.onError(ex);
    }
}
  • 首先创建了 CreateEmitter对象,
  • 然后调用了 subscribe 方法传进来的 Observer 对象的 onSubscribe() 方法
  • 然后调用了create 操作符 传进来的 ObservableOnSubscribe 对象的 subscribe(ObservableEmitter<T> emitter)方法

因为 CreateEmitter 类实现了 ObservableEmitter<T>Disposable 接口,
所以我们可以在 create 操作符 传进来的 ObservableOnSubscribe 对象的 subscribe(ObservableEmitter<T> emitter)方法里调用onNextonErroronComplete等方法。

static final class CreateEmitter<T>
extends AtomicReference<Disposable>
implements ObservableEmitter<T>, Disposable {
    ...
    final Observer<? super T> observer;

    CreateEmitter(Observer<? super T> observer) {
        this.observer = observer;
    }
    ...
}

ObservableEmitter 接口:

public interface ObservableEmitter<T> extends Emitter<T> {
    void setDisposable(@Nullable Disposable d);
    void setCancellable(@Nullable Cancellable c);
    boolean isDisposed();
    ObservableEmitter<T> serialize();
    boolean tryOnError(@NonNull Throwable t);
}

public interface Emitter<T> {
    void onNext(@NonNull T value);
    void onError(@NonNull Throwable error);
    void onComplete();
}

Disposable 接口:

public interface Disposable {
    void dispose();
    boolean isDisposed();
}

因为CreateEmitter 又重写了onNextonErroronComplete等方法。
所以 create 操作符 传进来的 ObservableOnSubscribe 对象的 subscribe(ObservableEmitter<T> emitter)方法里调用onNextonErroronComplete等方法实际上调用了 CreateEmitteronNextonErroronComplete等方法。

CreateEmitteronNextonErroronComplete方法:

    @Override
    public void onNext(T t) {
        if (t == null) {
            onError(new NullPointerException("..."));
            return;
        }
        if (!isDisposed()) {
            observer.onNext(t);
        }
    }

    @Override
    public void onError(Throwable t) {
        if (!tryOnError(t)) {
            RxJavaPlugins.onError(t);
        }
    }

    @Override
    public boolean tryOnError(Throwable t) {
        if (t == null) {
            t = new NullPointerException("...");
        }
        if (!isDisposed()) {
            try {
                observer.onError(t);
            } finally {
                dispose();
            }
            return true;
        }
        return false;
    }

    @Override
    public void onComplete() {
        if (!isDisposed()) {
            try {
                observer.onComplete();
            } finally {
                dispose();
            }
        }
    }
  • onNextonError传进来的值做了空判断。
  • 如果 !isDisposed() 则继续执行 observer 对象的 onNextonErroronComplete等方法。
    ( observer 对象为 create操作符 之后的 subscribe()方法传进来的 Observer<T> 对象)
  • 并在 onCompleteonError 方法最后执行 dispose() 方法。

接下来我们来看 CreateEmitterdispose()isDisposed()方法

    @Override
    public void dispose() {
        DisposableHelper.dispose(this);
    }

    @Override
    public boolean isDisposed() {
        return DisposableHelper.isDisposed(get());
    }

继续看 get()方法,看下面代码可知 get() 返回的是一个 Disposable 对象

static final class CreateEmitter<T>
extends AtomicReference<Disposable>
implements ObservableEmitter<T>, Disposable {...}


public class AtomicReference<V> implements Serializable {
    private volatile V value;

    public AtomicReference(V var1) {
        this.value = var1;
    }

    public AtomicReference() {
    }

    public final V get() {
        return this.value;
    }

继续看 DisposableHelperisDisposed(Disposable d)dispose(AtomicReference<Disposable> field)方法:

public enum DisposableHelper implements Disposable {
    /**
     * The singleton instance representing a terminal, disposed state, don't leak it.
     */
    DISPOSED
    ;

    public static boolean isDisposed(Disposable d) {
        return d == DISPOSED;
    }
    ...
    public static boolean dispose(AtomicReference<Disposable> 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;
    }
    ...
}
  • isDisposed(Disposable d) 则是判断 d 是否和枚举值 DISPOSED 相等。
  • dispose(AtomicReference<Disposable> field) 方法即是 将 CreateEmitterisDisposed() 中调用 get() 获取的对象赋值为 DisposableHelper 的枚举值 DISPOSED
    所以调用dispose(AtomicReference<Disposable> field)方法后, isDisposed(Disposable d)即返回true

CreateEmittersetDisposable(Disposable d)setCancellable(Cancellable c)

static final class CreateEmitter<T> 
        extends AtomicReference<Disposable>
        implements ObservableEmitter<T>, Disposable {
    ...
    @Override
    public void setDisposable(Disposable d) {
        DisposableHelper.set(this, d);
    }
    ...
    @Override
    public void setCancellable(Cancellable c) {
        setDisposable(new CancellableDisposable(c));
    }
    ...
}

get()方法返回的 Disposable需要在 setDisposable 或者 setCancellable 设置。
所以如果没有调用这两个方法,get()方法返回的值为 null

所以 isDisposed(Disposable d)true.

dispose(AtomicReference<Disposable> field) 方法中因为 currentnull, 所以直接返回 false


如果我们在create操作符中调用了 setDisposable 或者 setCancellable 方法,如下:

Observable
        .create(new ObservableOnSubscribe<Object>() {
            @Override
            public void subscribe(ObservableEmitter<Object> emitter) throws Exception {
                emitter.setCancellable(new Cancellable() {
                    @Override
                    public void cancel() throws Exception {}
                });
                 //or
                emitter.setDisposable(new Disposable() {
                    @Override
                    public void dispose() { }

                    @Override
                    public boolean isDisposed() {
                        return false;
                    }
                }); 
            }
        })
        .subscribe(new Observer<Object>() {...});

emitter.setCancellable 最后也是调用了 setDisposable(new CancellableDisposable(c)); 方法。
所以emitter.setDisposable or emitter.setCancellable 都是通过 DisposableHelper.set(this, d); 去赋值 CreateEmitter 中的 value 值,我们可以通过 上述的get()获取。


DisposableHelper.set(this, d)

public static boolean set(AtomicReference<Disposable> field, Disposable d) {
    for (;;) {
        Disposable current = field.get();  
        if (current == DISPOSED) {
            if (d != null) {
                d.dispose();
            }
            return false;
        }
        if (field.compareAndSet(current, d)) {
            if (current != null) {
                current.dispose();
            }
            return true;
        }
    }
}
  • 首先获取当前的 valuecurrent, 由上面的分析我们得知 默认为 null
    所以直接走到 field.compareAndSet(current, d),表示更新 CreateEmittervalued ,返回 true则表示 传递的参数 currentdnot equal
    又因为currentnull。所以直接 return true

  • 当我们调用了 setDisposable 或者 setCancellable之后, 再次调用 setDisposable 或者 setCancellable
    此时 current 的值则不为 null
    如果在此之前调用过 dispose()方法,则 current 即为 DISPOSED。所以再次 setDisposable 则无效。
    否则 比较 当前的 value 是否和 d 相等,如果不相等 field.compareAndSet(current, d)则返回 true,更新 CreateEmittervalued,并释放 current

所以我们多次调用 setDisposable 或者 setCancellable ,如果中间没有调用 dispose(); ,后一次设置会覆盖前面一次设置,最后有效的为最后一次设置。

以上

参考文章

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

推荐阅读更多精彩内容