RxJava 原理分析（重启）

以前写过 RxJava 源码分析，但是是直接对着源码分析流程，分析的也比较简单，这次从原理角度分析。 RxJava 是异步，网上介绍的都是订阅者模式，其实就是观察者模式的变种，我们先分析下 java 中自带的观察者接口 Observer 和被观察者 Observable。

这两个类的代码比较简单，Observable 中有个集合，集合中存放的是 Observer，Observable 有个notifyObservers()方法，一旦调用，就会遍历集合，执行 Observer 对象的 update() 方法。在使用中，我们可以写一个类 ObservableManager 继承 Observable，把ObservableManager做成单例，对外暴露一个封装了 notifyObservers() 的方法，供外部调用；我们在其它业务需要的地方实现 Observer 接口，通过 addObserver() 方法把它的实现类添加到 ObservableManager 中
，这样，当我们在一个地方用 ObservableManager发送了一条消息，在B地方，添加到集合中的 Observer 都能收到消息，这就是观察者模式。

RxJava 则是一个变种，它里面把发送消息的这一块，抽取了个接口出来，下面写个简单的例子，是 RxJava 的简化版。Observer 是观察者，与传统的相比，这里面有三个回调方法；Emitter 是用来发送消失，说白了就是调用 Observer 的，所以它里面的方法与 Observer 相对应；传统的观察者模式中，被观察者对外提供一个方法，这个方法直接遍历观察者，发送信息，在 RxJava 中，由于使用 Emitter 发送消息，我们不能直接创建它，需要让RxJava把它作为参数回调出来，让开发者拿到 Emitter 来调用它的方法，我们对 Emitter 进行一层封装，暴露给使用者，这里用 ObservableOnSubscribe 来实现这个功能；最后就是被观察者了，ObservableCreate 中传入一个 ObservableOnSubscribe，把它当做一个成员变量，在添加观察者的方法中，马上使用 ObservableOnSubscribe 来执行它的回调，其中的形参 Emitter 是我们自己封装的一个实现类，具体代码如下

public interface Observer<T> {
    //收到消息
    void onNext(T t);
    //出错的回调
    void onError(Throwable e);
    //从订阅到消息全部发送成功
    void onComplete();
}

public interface Emitter<T> {
    //发送消息
    void onNext(T t);
    //发送异常
    void onError(Throwable e);
    //发送完成的信号
    void onComplete();
}

public interface ObservableOnSubscribe<T> {
    //为每个订阅的观察者提供一个能进行消息发送的功能，这里是给调用的地方使用的
    void subscribe(Emitter<T> emitter);
}

以上三个接口，下面是重点

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

    //创造被观察者的实例
    public static<T> ObservableCreate create(ObservableOnSubscribe<T> source){
        return new ObservableCreate<>(source);
    }

    public void subscribe(Observer observer) {
        subscribeActual(observer);
    }

    protected void subscribeActual(Observer observer) {
        CreateEmitter<T> parent = new CreateEmitter<T>(observer);
        source.subscribe(parent);
    }


    static final class CreateEmitter<T> implements Emitter<T>{
        Observer<T> observer;
        public CreateEmitter(Observer<T> observer) {
            this.observer = observer;
        }
        @Override
        public void onNext(T t) {
            observer.onNext(t);
        }
        @Override
        public void onError(Throwable e) {
            observer.onError(e);
        }
        @Override
        public void onComplete() {
            observer.onComplete();
        }
    }
}

写个简单的调用例子

    void test1(){
        ObservableCreate.create(new ObservableOnSubscribe<String>() {
            @Override
            public void subscribe(Emitter<String> emitter) {
                // 触发点
                emitter.onNext("a");
                emitter.onNext("b");
                emitter.onComplete();
            }
        }).subscribe(new Observer<String>() {
            @Override
            public void onNext(String s) {
                System.out.println("onNext:  " + s);
            }

            @Override
            public void onError(Throwable e) {

            }

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

这样，本例子中的 emitter 其实就是 ObservableCreate 中的 CreateEmitter对象；ObservableCreate 中的 source，就是 test1()中的 new ObservableOnSubscribe<String>()对象，这样看起来，test1() 中先是给成员变量source赋值，执行 subscribe() 方法时，就触发了 ObservableCreate 中的 subscribeActual(Observer observer) 方法，此时的 observer 就是 test1() 中创建的 Observer<String> 对象，subscribeActual(Observer observer) 中的第二
行代码 source.subscribe(parent) 执行的是 test1()中加注释 //触发点 的位置。这种写法是观察者的变种，是相应代码，执行的顺序是从上往下的。如果还感觉晕的可以自己打断点试试。

单看上述代码，基本无价值，无外乎就是使用了接口，进行了逻辑调整，上述只是个基础，现在开始深入一点，为了方便，我把 ObservableCreate 的代码进行抽取

public abstract class Observable<T> {

    public void subscribe(Observer observer) {
        subscribeActual(observer);
    }
    protected abstract void subscribeActual(Observer<T> observer);
    
    //创造被观察者的实例
    public static<T> ObservableCreate create(ObservableOnSubscribe<T> source){
        return new ObservableCreate<>(source);
    }

}

public class ObservableCreate <T> extends Observable<T> {
    private ObservableOnSubscribe<T> source;
    public ObservableCreate(ObservableOnSubscribe<T> source){
        this.source = source;
    }

    @Override
    public void subscribe(Observer observer) {
        super.subscribe(observer);
    }

    @SuppressWarnings("unchecked")
    @Override
    protected void subscribeActual(Observer observer) {
        CreateEmitter<T> parent = new CreateEmitter<T>(observer);
        source.subscribe(parent);
    }


    static final class CreateEmitter<T> implements Emitter<T>{
        Observer<T> observer;
        public CreateEmitter(Observer<T> observer) {
            this.observer = observer;
        }
        @Override
        public void onNext(T t) {
            observer.onNext(t);
        }
        @Override
        public void onError(Throwable e) {
            observer.onError(e);
        }
        @Override
        public void onComplete() {
            observer.onComplete();
        }
    }
}

这样的代码和之前是一样的，只不过进行了抽取，对于 Observable 其实还可以抽取一下，

public interface ObservableSource<T> {
    void subscribe(Observer<T> observer);
}

public abstract class Observable<T> implements ObservableSource{

    @Override
    public void subscribe(Observer observer) {
        subscribeActual(observer);
    }

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

    //创造被观察者的实例
    public static<T> ObservableCreate create(ObservableOnSubscribe<T> source){
        return new ObservableCreate<>(source);
    }

}

如果把它放进去，此时简化版的 RxJava 有六个类了。这六个类能干嘛呢？它还是只能执行 test1() 中的功能，如果想扩展功能怎么办，比如 RxJava 中的 map() 对数据进行转换。定义个新的数据转换接口 Function，新的被观察者类型 ObservableMap，以及在 Observable 中新添加一个创建 ObservableMap 对象的方法，注意为非静态的

public class ObservableMap<T,R> extends Observable<R>{

    final ObservableSource<T> source;
    final Function<T,R> function;

    public ObservableMap(ObservableSource<T> source, Function<T, R> function) {
        this.source = source;
        this.function = function;
    }

    @Override
    protected void subscribeActual(Observer<R> observer) {
        source.subscribe(new MapObserver<>(observer,function));
    }


    static final class MapObserver<T,R> implements Observer<T>{
        final Function<T,R> mapper;
        final Observer<R> actual;

        public MapObserver(Observer<R> actual, Function<T, R> mapper) {
            this.actual = actual;
            this.mapper = mapper;
        }
        @Override
        public void onError(Throwable e) {
            actual.onError(e);
        }
        @Override
        public void onComplete() {
            actual.onComplete();
        }
        @Override
        public void onNext(T t) {
            R apply = mapper.apply(t);
            actual.onNext(apply);
        }
    }

}

public abstract class Observable<T> implements ObservableSource{

    @Override
    public void subscribe(Observer observer) {
        subscribeActual(observer);
    }

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

    //创造被观察者的实例
    public static<T> ObservableCreate create(ObservableOnSubscribe<T> source){
        return new ObservableCreate<>(source);
    }

    //创建被观察者
    @SuppressWarnings("unchecked")
    public<R> Observable<R> map(Function<T,R> function){
        return new ObservableMap<>(this,function);
    }

}

写个例子，运行下，大家可以看下打印的值，发现还是从上往下执行

    void test2(){
        Observable.create(new ObservableOnSubscribe<String>() {
            @Override
            public void subscribe(Emitter<String> emitter) {
                emitter.onNext("a");
                emitter.onComplete();
            }
        }).map(new Function<String, String>() {
            @Override
            public String apply(String s) {
                return s +" 111";
            }
        }).subscribe(new Observer<String>() {
            @Override
            public void onNext(String s) {
                System.out.println("onNext:  " + s);
            }

            @Override
            public void onError(Throwable e) {

            }

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

ObservableMap 中有两个成员变量 ObservableSource 和 Function，这里用到了装饰者模式，意思是它里面可以调用 ObservableSource 的方法，我们看看 test2() 中的代码，Observable.create() 返回对象记为 A，此时调用了 map() 方法，返回对象记为 B，那么它俩之间就产生了关联

    public<R> Observable<R> map(Function<T,R> function){
        return new ObservableMap<>(this,function);
    }

看看 map() 类，A 是 ObservableCreate 类型，B 是 ObservableMap 类型，并且把 A 当做参数传到了 B 中，此时 B 中的成员变量 ObservableSource 和 Function 就是 A 和 test2() 中 map() 方法对应的 new Function<String, String> 匿名接口对象。再往下的 subscribe() 方法执行的是对象 B 的方法，注意了 

    @Override
    protected void subscribeActual(Observer<R> observer) {
        source.subscribe(new MapObserver<>(observer,function));
    }

此时 source 是 A， MapObserver 把数据转换器 function 和观察者 observer 封装了一层，然后供 source 调用，MapObserver 实现了 Observer，它里面就是个中转站，第一反应，代理或者装饰模式，装饰比着代理，更侧重于功能的扩展，这里是装饰模式。 此时是把逻辑交割给了 A，也就是 ObservableCreate 类，执行它的 subscribeActual()方法

    @Override
    protected void subscribeActual(Observer observer) {
        CreateEmitter<T> parent = new CreateEmitter<T>(observer);
        source.subscribe(parent);
    }

此时，subscribeActual(Observer observer) 的形参 observer 就是 ObservableMap 传过来的 MapObserver 对象，到这里就接到第一阶段的分析了，source 是 test2() 中的第一行代码 create() 方法中的 ObservableOnSubscribe 对象，subscribeActual() 方法触发了 source.subscribe(parent)，此时看 test2() 中代码 emitter.onNext("a")，emitter 是 subscribeActual(Observer observer) 方法中 CreateEmitter 对象，也就是执行它的 onNext(T t)，这就是个代理，执行的是
observer.onNext(t)，上面分析了 observer 是 ObservableMap 传过来的 MapObserver 对象，看看MapObserver的onNext(t) 方法，此时 t 的值是字符串 "a"

    @Override
    public void onNext(T t) {
        R apply = mapper.apply(t);
        actual.onNext(apply);
    }

此时，mapper.apply(t) 执行了数据转换，对应 test2() 中的 new Function<String, String>() 对象中的方法，此时返回的值 apply 是字符串 "a 111" ，然后调用了actual.onNext(apply)，这里 actual 是 MapObserver 构造方法传入的观察者 observer，也就是 test2() 中的 new Observer<String>() 对象，此时执行的是它的 onNext(apply)方法，到此本次分析结束。

如果我们在上面方法基础上，再提那家一个 map 转换呢？

   void test3(){
        Observable.create(new ObservableOnSubscribe<String>() {
            @Override
            public void subscribe(Emitter<String> emitter) {
                emitter.onNext("a");
                emitter.onComplete();
            }
        }).map(new Function<String, String>() {
            @Override
            public String apply(String s) {
                return s +" 111";
            }
        }).map(new Function<String, Integer>() {
            @Override
            public Integer apply(String s) {
                return s.length();
            }
        }).subscribe(new Observer<Integer>() {
            @Override
            public void onNext(Integer i) {
                System.out.println("onNext:  " + i);
            }

            @Override
            public void onError(Throwable e) {

            }

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

注意了，数据是从上往下，数据是一层层转换，然后传给下一层。 还按照上面的分析，create() 方法创建了 ObservableCreate 对象 A，map(new Function<String, String>()) 创建了 ObservableMap 对象 B，map(new Function<String, Integer>()) 创建了 ObservableMap 对象 C，对象 C 执行了 subscribe(Observer observer) 方法，分析后，会发现和上面的逻辑一样，对象包裹关系是 A-->B-->C 一层包裹一层，当C执行 subscribe(Observer observer) 方法时，会把它本身的 observer 和 function 封装成一个新的观察者类，然后把它作为参数调用B的 subscribe(Observer observer) 方法，此时B也是ObservableMap类型，相同的操作再重复一遍，然后调用A的subscribe(Observer observer) 方法，到这里，随时到达顶点了，这里触发了发射器的回调方法，然后开始了test3()中从上往下传值，先是执行 emitter.onNext("a") 代码，此时在A中CreateEmitter对象中的observer，是B通过方法传过来的，所以就调用B对象的onNext(T t) 方法，这个方法是字符串的拼接功能，

    @Override
    public void onNext(T t) {
        R apply = mapper.apply(t);
        actual.onNext(apply);
    }
        
    // mapper 对象的实例
    new Function<String, String>() {
        @Override
        public String apply(String s) {
            return s +" 111";
        }
    }

        
R apply = mapper.apply(t) ，获得新的值后，然后又调用了 actual.onNext(apply) 这行代码，此时 actual 是C传进来的观察者，又回到了数据转换的地方

    @Override
    public void onNext(T t) {
        R apply = mapper.apply(t);
        actual.onNext(apply);
    }
        
    // mapper 对象的实例
    new Function<String, Integer>() {
        @Override
        public Integer apply(String s) {
            return s.length();
        }

这一步获取了字符串的长度，把字符串转换为了Integer类型数据，actual 就是 test3() 中的观察者了，所以就打印了日志。

看到这，其实就是 A-->B-->C，这是包裹关系，最后一个被观察者执行了 subscribe() 方法后，ObservableMap 类型的对象，都把自己的属性 Function 和 observer 给封装成新的 observer，然后传给上一层，此时调用关系就是 C-->B-->A；到达A，也就是 ObservableCreate 类型了，它内部没有用装饰模式包装一个 Observable，算是到了代码的开端的地方，从这里，开始了代码的调用，此时是调用是一层调用一层，A-->B-->C，每层都把数据传递给下一层。关于这种代码结构，
有人认为是适配器模式，但我个人觉得里面更像是个责任链模式。从这里也发现个规律，subscribeActual() 方法是数据开始的地方，用于封装该方法传形参Observer而创建的新 Observer 中的 onNext(T t) 方法，则是接收数据的地方。看到这，是否有什么联想？ 关于这里说这么多，其实是为了线程切换做准备的。既然一个地方触发下一步，一个接收上面传递的数据，那么我们在这两个地方是不是可以切换线程了？

我们可以仿照 ObservableMap ，也用装饰模式，来写两个类，一个是子线程，一个是android的UI线程

public class ObservableIo<T> extends Observable<T>{

    private static ExecutorService executorService = Executors.newCachedThreadPool();
    final ObservableSource<T> source;

    public ObservableIo(ObservableSource<T> source) {
        this.source = source;
    }

    @Override
    protected void subscribeActual(final Observer<T> observer) {
        executorService.submit(new Runnable() {
            @Override
            public void run() {
                source.subscribe(new IoObserver<>(observer));
            }
        });
    }


    static final class IoObserver<T> implements Observer<T>{
        final Observer<T> actual;

        public IoObserver(Observer<T> actual) {
            this.actual = actual;
        }
        @Override
        public void onError(Throwable e) {
            actual.onError(e);
        }
        @Override
        public void onComplete() {
            actual.onComplete();
        }
        @Override
        public void onNext(T t) {
            actual.onNext(t);
        }
    }

}


public class ObservableUI<T> extends Observable<T>{

    final ObservableSource<T> source;

    public ObservableUI(ObservableSource<T> source) {
        this.source = source;
    }

    @Override
    protected void subscribeActual(Observer<T> observer) {
        source.subscribe(new UIObserver<>(observer));
    }


    static final class UIObserver<T> implements Observer<T>{
        final Observer<T> actual;

        public UIObserver(Observer<T> actual) {
            this.actual = actual;
        }
        @Override
        public void onError(Throwable e) {
            actual.onError(e);
        }
        @Override
        public void onComplete() {
            actual.onComplete();
        }
        @Override
        public void onNext(final T t) {
            new Handler(Looper.getMainLooper()).post(new Runnable() {
                @Override
                public void run() {
                    actual.onNext(t);
                }
            });
        }
    }

}

Observable 类中新添两个方法

    public<R> Observable<R> io(){
        return new ObservableIo<>(this);
    }

    public<R> Observable<R> ui(){
        return new ObservableUI<>(this);
    }

然后看看测试代码

    private void test4() {
        Observable.create(new ObservableOnSubscribe<String>() {
            @Override
            public void subscribe(Emitter<String> emitter) {
                Log.e("Observable", "subscribe:  " + (Looper.getMainLooper() == Looper.myLooper()));
                emitter.onNext("a");
            }
        }).map(new Function<String, String>() {
            @Override
            public String apply(String s) {
                Log.e("Observable", "apply:  " + s + "  " + (Looper.getMainLooper() == Looper.myLooper()));
                return s +" 111";
            }
        }).io()
          .ui()
          .subscribe(new Observer<String>() {
            @Override
            public void onNext(String s) {
                Log.e("Observable", "onNext:  " + s + "  " + (Looper.getMainLooper() == Looper.myLooper()));
            }

            @Override
            public void onError(Throwable e) {

            }

            @Override
            public void onComplete() {
            }
        });
    }

打印日志为

Observable: subscribe:  false
Observable: apply:  a  false
Observable: onNext:  a 111  true

上面的代码已经能实现线程切换了。这是一个比较粗糙的代码，其中 ObservableMap、ObservableIo、ObservableUI 都用到了装饰模式，那么类似的都可以抽取一个基类，它们里面的 Observer 包装类，也可以抽取基类，这样方便后期扩展。 上面的例子只是实现了 onNext() 方法的线程切换，这仅仅是个例子，RxJava 的代码有点多，直接看有点懵，这里提供了简易版的，它就是 RxJava 的原理，这个Demo看懂了，再去看 RxJava ，应该就容易理解了。