CompletionService的测试代码如下所示:
final Random random = new Random();
ExecutorService executorService = Executors.newFixedThreadPool(10);
CompletionService<String>completionService = new ExecutorCompletionService<String>(executorService);
for(int i = 0 ; i < 100 ; i++) {
final int num = i;
completionService.submit(new Callable<String>() {
public String call() {
try {
Thread.sleep((random.nextLong()) & 5000);
} catch (InterruptedException e) {
e.printStackTrace();
}
return "a" + num;
}
});
}
for(int i = 0 ; i < 100 ; i++) {
Future<String> f = completionService.take();
System.out.println(f.get());
executorService.shutdown();
我们通过ExecutorService提交的callble任务在拿到Future后调用get方法时过去执行结果时,如果这个任务还没有完成则会阻塞等待返回结果,如果使用的是CompletionService则调用take方法取到的都是完成的任务的Future对象,自然调用get方法时就不会阻塞了。
CompletionService实际上就是我们前面创建的ExecutorService的包装,将提交的Callable任务进行包装,CompletionService的submit方法代码如下:
public Future<V> submit(Callable<V> task) {
if (task == null) throw new NullPointerException();
RunnableFuture<V> f = newTaskFor(task);
executor.execute(new QueueingFuture(f));
return f;
}
QueueingFuture将task进行包装,实际上线程池执行的时QueueingFuture的run方法,该方法就是调用玩task.call方法获取到返回值Future之后将任务放到
CompletionService自己维护的一个队列中,这样子后面通过completionService.take()方法获取到的Future自然就是已经执行完成的任务,巧妙的完成了装换。