小明要吃冰淇淋之RxJava:lift原理

静默加载

发布于 2020-05-29 11:09:24

3200

发布于 2020-05-29 11:09:24

前言

我们在使用 RxJava 的时候必不可少的会使用一些基础的操作符， map 、 flatMap 、 filter 等。这些都是对 Obserbable 进行各种各样的变化，我们查看其实现都能看到一个叫做 lift 的操作符。

lift源码

我们先看源码（rxjava:1.3.0），再通过一个例子来熟悉、掌握。

public interface Func1<T, R> extends Function {
    R call(T t);
}
//注意操作符的入参和出参都是Subscriber
public interface Operator<R, T> extends Func1<Subscriber<? super R>, Subscriber<? super T>> {
    // cover for generics insanity
}

public final <R> Observable<R> lift(final Operator<? extends R, ? super T> operator) {
    //创建新的Observable，OnSubscribeLift代替原理的onSubscribe
    return unsafeCreate(new OnSubscribeLift<T, R>(onSubscribe, operator));
}

public final class OnSubscribeLift<T, R> implements OnSubscribe<R> {
    //上一层的Observable的onSubscribe
    final OnSubscribe<T> parent;
    //需要变换的操作符
    final Operator<? extends R, ? super T> operator;
    public OnSubscribeLift(OnSubscribe<T> parent, Operator<? extends R, ? super T> operator) {
        this.parent = parent;
        this.operator = operator;
    }
    @Override
    public void call(Subscriber<? super R> o) {
        try {
            //调用操作符的call方法，operator.call
            Subscriber<? super T> st = RxJavaHooks.onObservableLift(operator).call(o);
            try {
                st.onStart();
                parent.call(st);
            } catch (Throwable e) {
                Exceptions.throwIfFatal(e);
                st.onError(e);
            }
        } catch (Throwable e) {
            Exceptions.throwIfFatal(e);
            o.onError(e);
        }
    }
}

源码解读

以上为 lift 操作相关的代码，我们先不管泛型中的 R 、 T 之类的，这种情况下根据泛型去推到容易让其更加的复杂。所以我们先简化一下代码：

public <R> Observable<R> lift(Operator<? extends R, ? super T> operator) {
    return Observable.create(new OnSubscribe<R>() {
        @Override
        public void call(Subscriber subscriber) {
            Subscriber newSubscriber = operator.call(subscriber);
            newSubscriber.onStart();
            onSubscribe.call(newSubscriber);
        }
    });
}