美文网首页
Rxjava2.0 Schedulers. io() 源码解析

Rxjava2.0 Schedulers. io() 源码解析

作者: Android程序员老鸦 | 来源:发表于2021-07-08 17:37 被阅读0次

这篇文章来看看rxjava指定线程的时候,那些线程是怎么创建出来的,先来看Schedulers.io(),看看这个io线程的创建:

public final class Schedulers {
    ...
    @NonNull
    static final Scheduler IO;
    static {
        //RxJavaPlugins类是rxjava里很强大的工具类,之前我们多次看到它的身影
        //这里又用到了它来创建Scheduler对象
        IO = RxJavaPlugins.initIoScheduler(new IOTask());
     }

     public static Scheduler io() {
        //RxJavaPlugins的onIoScheduler()方法,这种方法看着又是一个包装,没有包装则直接返回入参IO自身
        return RxJavaPlugins.onIoScheduler(IO);
     }
 //静态内部类IOTask 
 static final class IOTask implements Callable<Scheduler> {
        @Override
        public Scheduler call() throws Exception {
            return IoHolder.DEFAULT;
        }
    }
 //静态内部类IoHolder 
  static final class IoHolder {
        //静态变量DEFAULT ,最终是在这里创建的IoScheduler
        static final Scheduler DEFAULT = new IoScheduler();
    }
    ...
}

再来看看RxJavaPlugins里的initIoScheduler()方法:

    public static Scheduler initIoScheduler(@NonNull Callable<Scheduler> defaultScheduler) {
        ObjectHelper.requireNonNull(defaultScheduler, "Scheduler Callable can't be null");
        Function<? super Callable<Scheduler>, ? extends Scheduler> f = onInitIoHandler;
        if (f == null) {
            //onInitIoHandler不另外设置则为空,callRequireNonNull方法返回的就是IoScheduler
            return callRequireNonNull(defaultScheduler);
        }
        return applyRequireNonNull(f, defaultScheduler);
    }
    static Scheduler callRequireNonNull(@NonNull Callable<Scheduler> s) {
        try {
              //s.call()不为空则返回s.call(),也就是IoScheduler
            return ObjectHelper.requireNonNull(s.call(), "Scheduler Callable result can't be null");
        } catch (Throwable ex) {
            throw ExceptionHelper.wrapOrThrow(ex);
        }
    }

绕来绕去,最终Schedulers.io()拿到的Scheduler就是IoScheduler,由上篇文章可知subscribeOn()方法后最终会这样用到IoScheduler(里面的scheduleDirect()方法就是开了线程在执行Runnable):

//这个scheduler就是IoScheduler,SubscribeTask就是个ruannable,run()方法运行了Observable的subscribe()方法
parent.setDisposable(scheduler.scheduleDirect(new SubscribeTask(parent)));

先看看scheduleDirect()方法,这个方法是父类Scheduler里的:

public abstract class Scheduler {
  ...
  @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) {
        //createWorker()是抽象方法,需要看子类的实现
        final Worker w = createWorker();
        //返回装饰过的runable,其实就是返回了自身run
        final Runnable decoratedRun = RxJavaPlugins.onSchedule(run);
        //继续封装成DisposeTask ,decoratedRun的装饰类
        DisposeTask task = new DisposeTask(decoratedRun, w);
        //执行了task方法
        w.schedule(task, delay, unit);

        return task;
    }
  ...
}

看看IoScheduler的createWorker()方法:

public final class IoScheduler extends Scheduler {
    ...
    @Override
    public Worker createWorker() {
        //新建了一个EventLoopWorker对象
        return new EventLoopWorker(pool.get());
    }
  //静态内部类
   static final class EventLoopWorker extends Scheduler.Worker {
        private final CompositeDisposable tasks;
        private final CachedWorkerPool pool;
        private final ThreadWorker threadWorker;

        final AtomicBoolean once = new AtomicBoolean();

        EventLoopWorker(CachedWorkerPool pool) {
            this.pool = pool;
            this.tasks = new CompositeDisposable();
            this.threadWorker = pool.get();
        }

        @Override
        public void dispose() {
            if (once.compareAndSet(false, true)) {
                tasks.dispose();

                // releasing the pool should be the last action
                pool.release(threadWorker);
            }
        }

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

        @NonNull
        @Override
        //最终线程是在这里执行的
        public Disposable schedule(@NonNull Runnable action, long delayTime, @NonNull TimeUnit unit) {
            //解除订阅了,返回EmptyDisposable
            if (tasks.isDisposed()) {
                // don't schedule, we are unsubscribed
                return EmptyDisposable.INSTANCE;
            }
            //最后靠的是ThreadWorker(线程工作者) 的scheduleActual()(实际安排执行的方法)
            return threadWorker.scheduleActual(action, delayTime, unit, tasks);
        }
    }
  //静态内部类ThreadWorker,继承自NewThreadWorker,NewThreadWorker很重要
  static final class ThreadWorker extends NewThreadWorker {
        //封装了一个截至时间,主要的方法还是在父类
        private long expirationTime;

        ThreadWorker(ThreadFactory threadFactory) {
            super(threadFactory);
            this.expirationTime = 0L;
        }

        public long getExpirationTime() {
            return expirationTime;
        }

        public void setExpirationTime(long expirationTime) {
            this.expirationTime = expirationTime;
        }
    }
    ...
}

看看父类NewThreadWorker :

public class NewThreadWorker extends Scheduler.Worker implements Disposable {
    ...
    private final ScheduledExecutorService executor;

    public NewThreadWorker(ThreadFactory threadFactory) {
        executor = SchedulerPoolFactory.create(threadFactory);
    }
    public ScheduledRunnable scheduleActual(final Runnable run, long delayTime, @NonNull TimeUnit unit, @Nullable DisposableContainer parent) {
        //返回自身run
        Runnable decoratedRun = RxJavaPlugins.onSchedule(run);
      //装饰decoratedRun,ScheduledRunnable 便于管理生命周期,这里不深究,本质还是runnable
        ScheduledRunnable sr = new ScheduledRunnable(decoratedRun, parent);
      
        if (parent != null) {
            if (!parent.add(sr)) {
                return sr;
            }
        }

        Future<?> f;
        try {
            if (delayTime <= 0) {
                //熟悉的身影,executor是个线程池,最终在这里开辟了异步线程来执行subscribe()方法,
                //从而达到异步线程的目的,submit()立马执行
                f = executor.submit((Callable<Object>)sr);
            } else {
              //用schedule()来执行延迟线程
                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;
    }
   ...
}

上面executor的创建:

    public static ScheduledExecutorService create(ThreadFactory factory) {
        final ScheduledExecutorService exec = Executors.newScheduledThreadPool(1, factory);//核心线程为1的线程池
        if (PURGE_ENABLED && exec instanceof ScheduledThreadPoolExecutor) {
            ScheduledThreadPoolExecutor e = (ScheduledThreadPoolExecutor) exec;
            POOLS.put(e, exec);
        }
        return exec;
    }
io线程开启总结:
IoScheduler更多的还是做一些协调作用,真正用到的是 NewThreadWorker 类的线程池来开辟线程执行异步操作。

下篇分析AndroidSchdulers.mainThread(),看看它是怎么切换到android主线程的。

相关文章

网友评论

      本文标题:Rxjava2.0 Schedulers. io() 源码解析

      本文链接:https://www.haomeiwen.com/subject/ixhtpltx.html