CompletionService讲解
应用场景
当向Executor提交多个任务并且希望获得它们在完成之后的结果,如果用FutureTask,可以循环获取task,并调用get方法去获取task执行结果,但是如果task还未完成,获取结果的线程将阻塞直到task完成,由于不知道哪个task优先执行完毕,使用这种方式效率不会很高。在jdk5时候提出接口CompletionService,它整合了Executor和BlockingQueue的功能,可以更加方便在多个任务执行时获取到任务执行结果。
案例
需求:不使用求和公式,计算从1到100000000相加的和。
分析设计:需求指明不能使用求和公式,只能循环依次相加,为了提高效率,我们可以将1到100000000的数分为n段由n个task执行,执行结束后merge结果求最后的和。
声明task执行载体,线程池executor;
声明CompletionService,通过参数指定执行task的线程池,存放已完成状态task的阻塞队列,队列默认为基于链表结构的阻塞队列LinkedBlockingQueue;
调用submit方法提交task;
调用take方法获取已完成状态task。
CompletionService源码分析
CompletionService接口提供五个方法:
Future<V> submit(Callable<V> task)
提交Callable类型的task;Future<V> submit(Runnable task, V result)
提交Runnable类型的task;Future<V> take() throws InterruptedException
获取并移除已完成状态的task,如果目前不存在这样的task,则等待;Future<V> poll()
获取并移除已完成状态的task,如果目前不存在这样的task,返回null;Future<V> poll(long timeout, TimeUnit unit) throws InterruptedException
获取并移除已完成状态的task,如果在指定等待时间内不存在这样的task,返回null。
接下来我们来看看CompletionService接口的具体实现:ExecutorCompletionService。
ExecutorCompletionService实现分析
executor:执行task的线程池,创建CompletionService必须指定;
aes:主要用于创建待执行task;
completionQueue:存储已完成状态的task,默认是基于链表结构的阻塞队列LinkedBlockingQueue。
参数校验,不符合条件的task抛出异常,程序结束;
将Callable类型或者Runnable类型的task构造成FutureTask;
把构造好的FutureTask交由线程池executor执行。
看到这里可能大家会比较疑惑了,task调用submit方法可以提交,完成的task是什么时候被加入到completionQueue里的呢?
take在获取并移除已完成状态的task时,如果目前暂时不存在这样的task,等待,直到存在这样的task;
poll在获取并移除已完成状态的task时,如果目前暂时不存在这样的task,不等待,直接返回null。
Last updated