public int countLines(List<Path> javaFiles) throws Exception {
// 根据处理器数量创建线程池。虽然多线程并不保证能够提升性能,但适量地
// 开线程一般可以从系统骗取更多资源。
ExecutorService es = Executors.newFixedThreadPool(
Runtime.getRuntime().availableProcessors() * 2);
// 使用 ExecutorCompletionService 内建的阻塞队列。
CompletionService cs = new ExecutorCompletionService(es);
// 按文件向 CompletionService 提交任务。
for (final Path javaFile : javaFiles) {
cs.submit(new Callable<Integer>() {
@Override
public Integer call() throws Exception {
// 略去计算单个文件行数的代码。
return countLines(javaFile);
}
});
}
try {
int loc = 0;
int size = javaFiles.size();
for (int i = 0; i < size; i++) {
// take 方法等待下一个结果并返回 Future 对象。不直接返回计算结果是为了
// 捕获计算时可能抛出的异常。
// poll 不等待,有结果就返回一个 Future 对象,否则返回 null。
loc += cs.take().get();
}
return loc;
} finally {
// 关闭线程池。也可以将线程池提升为字段以便重用。
// 如果任务线程(Callable#call)能响应中断,用 shutdownNow 更好。
es.shutdown();
}
}
CompletionService
最新推荐文章于 2024-10-13 23:51:46 发布