AbstractExecutorService源码分析
AbstractExecutorService源码分析
AbeJeffrey 发表于7个月前
AbstractExecutorService源码分析
  • 发表于 7个月前
  • 阅读 20
  • 收藏 2
  • 点赞 0
  • 评论 0
摘要: AbstractExecutorService实现了ExecutorService,提供了很多默认的操作,本文将结合源码分析这些操作的实现。

AbstractExecutorService实现了ExecutorService,它为ExecutorService中的执行方法提供了默认实现。该类使用 newTaskFor 返回的RunnableFuture实现了submit、invokeAny 和 invokeAll 方法。下面做详细分析。

    protected <T> RunnableFuture<T> newTaskFor(Runnable runnable, T value) {
        return new FutureTask<T>(runnable, value);
    }

为给定任务和默认值返回一个RunnableFuture。关于RunnableFuture可参见https://my.oschina.net/7001/blog/873089

    protected <T> RunnableFuture<T> newTaskFor(Callable<T> callable) {
        return new FutureTask<T>(callable);
    }

为给定可调用任务返回一个RunnableFuture。

submit

    public Future<?> submit(Runnable task) {
        if (task == null) throw new NullPointerException();
        RunnableFuture<Void> ftask = newTaskFor(task, null);
        execute(ftask);//由子类提供实现
        return ftask;
    }

提交并执行Runnable任务,并返回Future,该 Future 的 get 方法在成功完成时将会返回null。

    public <T> Future<T> submit(Runnable task, T result) {
        if (task == null) throw new NullPointerException();
        RunnableFuture<T> ftask = newTaskFor(task, result);
        execute(ftask);
        return ftask;
    }

提交并执行Runnable任务,并返回Future,该 Future 的 get 方法在成功完成时将会返回给定的结果。

   public <T> Future<T> submit(Callable<T> task) {
        if (task == null) throw new NullPointerException();
        RunnableFuture<T> ftask = newTaskFor(task);
        execute(ftask);
        return ftask;
    }

提交并执行Callable任务,并返回Future,该 Future 的 get 方法在成功完成时将会返回该任务的结果。

invokeAny

执行给定的任务列表,如果某个任务已成功完成(未抛出异常)则立马返回其结果。一旦正常或异常返回后,则取消尚未完成的任务。

   //执行列表中的任务,成功完成一个任务则返回。
   public <T> T invokeAny(Collection<? extends Callable<T>> tasks)
        throws InterruptedException, ExecutionException {
        try {
            return doInvokeAny(tasks, false, 0);
        } catch (TimeoutException cannotHappen) {
            assert false;
            return null;
        }
    }
    //行列表中的任务,成功完成一个任务或达到超时间隔则返回。
    public <T> T invokeAny(Collection<? extends Callable<T>> tasks,
                           long timeout, TimeUnit unit)
        throws InterruptedException, ExecutionException, TimeoutException {
        return doInvokeAny(tasks, true, unit.toNanos(timeout));
    }

由invokeAny源码可知,其主要机制在doInvokeAny中实现,源码如下:

    private <T> T doInvokeAny(Collection<? extends Callable<T>> tasks,
                              boolean timed, long nanos)
        throws InterruptedException, ExecutionException, TimeoutException {
        if (tasks == null)
            throw new NullPointerException();
        int ntasks = tasks.size();
        if (ntasks == 0)
            throw new IllegalArgumentException();
        ArrayList<Future<T>> futures = new ArrayList<Future<T>>(ntasks);
        ExecutorCompletionService<T> ecs =
            new ExecutorCompletionService<T>(this);

        try {
            // ee用于记录异常,当不能获得任何结果时抛出记录的最后一个异常
            ExecutionException ee = null;
            final long deadline = timed ? System.nanoTime() + nanos : 0L;
            Iterator<? extends Callable<T>> it = tasks.iterator();

            // 启动一个任务,再渐进式启动其他任务,然后将任务数量减1,活动数量计1
            futures.add(ecs.submit(it.next()));
            --ntasks;
            int active = 1;

            for (;;) {
                Future<T> f = ecs.poll();//获取队头元素
                if (f == null) {//队列为空
                    if (ntasks > 0) {
                        //当没有任务完成,且任务队列还有任务未执行时,将一直进入此代码块直到提交完所有任务
                        --ntasks;
                        futures.add(ecs.submit(it.next()));
                        ++active;
                    }
                    else if (active == 0)
                        break;
                    else if (timed) {//只有提交完所有任务,且无任务完成时,且超时timed为true
                        f = ecs.poll(nanos, TimeUnit.NANOSECONDS);
                        if (f == null)//等待指定时间还没有任务完成则抛出异常
                            throw new TimeoutException();
                        nanos = deadline - System.nanoTime();
                    }
                    else
                        f = ecs.take();//不需要超时则阻塞直到有任务完成
                }
                if (f != null) {//队列中已有任务执行完成
                    --active;
                    try {
                        return f.get();
                    } catch (ExecutionException eex) {
                        ee = eex;
                    } catch (RuntimeException rex) {
                        ee = new ExecutionException(rex);
                    }
                }
            }

            if (ee == null)//说明没有任务正常执行完成返回结果
                ee = new ExecutionException();
            throw ee;

        } finally {
            for (int i = 0, size = futures.size(); i < size; i++)//取消剩余任务
                    futures.get(i).cancel(true);
        }
    }

doInvokeAny中使用ExecutorCompletionService来管理执行完成的任务,ExecutorCompletionService内部使用LinkedBlockingQueue作为完成队列,任务执行完后自动将结果Future加入到队列中。关于ExecutorCompletionService的具体实现可参考https://my.oschina.net/7001/blog/874638

invokeAll

执行给定的任务列表,当所有任务完成时才返回保持任务状态和结果的 Future 列表。下面是具体实现:

    public <T> List<Future<T>> invokeAll(Collection<? extends Callable<T>> tasks)
        throws InterruptedException {
        if (tasks == null)
            throw new NullPointerException();
        List<Future<T>> futures = new ArrayList<Future<T>>(tasks.size());
        boolean done = false;
        try {
            for (Callable<T> t : tasks) {//提交所有任务到线程池中执行
                RunnableFuture<T> f = newTaskFor(t);
                futures.add(f);
                execute(f);
            }
            for (Future<T> f : futures) {
                if (!f.isDone()) {//确保所有任务均已完成
                    try {
                        f.get();//不止需要确保任务完成,且确认是正常完成,异常完成该方法则会抛出异常
                    } catch (CancellationException ignore) {
                    } catch (ExecutionException ignore) {
                    }
                }
            }
            done = true;
            return futures;
        } finally {
            if (!done)
                for (Future<T> f : futures)
                    f.cancel(true);
        }
    }

该实现不考虑超时,下面的实现将考虑超时,若发生超时则会立即返回。源码如下:

    public <T> List<Future<T>> invokeAll(Collection<? extends Callable<T>> tasks,
                                         long timeout, TimeUnit unit)
        throws InterruptedException {
        if (tasks == null)
            throw new NullPointerException();
        long nanos = unit.toNanos(timeout);
        ArrayList<Future<T>> futures = new ArrayList<Future<T>>(tasks.size());
        boolean done = false;
        try {
            for (Callable<T> t : tasks)
                futures.add(newTaskFor(t));

            final long deadline = System.nanoTime() + nanos;
            final int size = futures.size();

            for (int i = 0; i < size; i++) {
                execute((Runnable)futures.get(i));
                nanos = deadline - System.nanoTime();
                if (nanos <= 0L)//发生超时则立即返回
                    return futures;
            }

            for (int i = 0; i < size; i++) {
                Future<T> f = futures.get(i);
                if (!f.isDone()) {
                    if (nanos <= 0L)
                        return futures;
                    try {
                        f.get(nanos, TimeUnit.NANOSECONDS);
                    } catch (CancellationException ignore) {
                    } catch (ExecutionException ignore) {
                    } catch (TimeoutException toe) {
                        return futures;
                    }
                    nanos = deadline - System.nanoTime();
                }
            }
            done = true;
            return futures;
        } finally {
            if (!done)
                for (int i = 0, size = futures.size(); i < size; i++)//取消正在执行的任务
                    futures.get(i).cancel(true);
        }
    }

综上所述,AbstractExecutorService为ExecutorService提供了默认实现,这里分析AbstractExecutorService的源码以方便后续介绍ThreadPoolExecutor的原理。

欢迎指出本文有误的地方,转载请注明原文出处https://my.oschina.net/7001/blog/875419

共有 人打赏支持
粉丝 22
博文 36
码字总数 89060
×
AbeJeffrey
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: