ForkJoinPool是Java 7引入的专为分而治之设计的线程池,基于工作窃取机制提升CPU利用率,适用于递归拆分的计算密集型任务。
ForkJoinPool 是 Java 7 引入的一个特殊的线程池,专门用于实现“分而治之”的算法。它基于 分支合并框架(Fork/Join Framework),适用于可以递归拆分成小任务的计算密集型任务。通过工作窃取(work-stealing)机制,空闲线程可以从其他线程的任务队列中“窃取”任务执行,提高 CPU 利用率。
ForkJoinPool 的设计目标是高效处理可分割的大任务。主要流程如下:
使用 ForkJoinPool 需要了解以下核心类:
下面以计算一个大数组元素总和为例,展示如何使用 RecursiveTask 实现分支合并。
1. 定义递归任务类
当数组长度大于阈值时,拆成两部分分别计算;否则直接求和。
import java.util.concurrent.RecursiveTask;public class SumTask extends RecursiveTask
{ private final long[] array; private final int start, end; private static final int THRESHOLD = 1000; // 拆分阈值 public SumTask(long[] array, int start, int end) { this.array = array; this.start = start; this.end = end; } @Override protected Long compute() { if (end - start zuojiankuohaophpcn= THRESHOLD) { // 小任务:直接计算 long sum = 0; for (int i = start; i zuojiankuohaophpcn end; i++) { sum += array[i]; } return sum; } else { // 大任务:拆分 int mid = (start + end) / 2; SumTask left = new SumTask(array, start, mid); SumTask right = new SumTask(array, mid, end); left.fork(); // 异步提交左任务 long rightResult = right.compute(); // 当前线程执行右任务 long leftResult = left.join(); // 等待左任务完成并获取结果 return leftResult + rightResult; } }}
2. 提交任
务到 ForkJoinPool
创建线程池并提交任务,获取最终结果。
import java.util.concurrent.ForkJoinPool;public class ForkJoinExample { public static void main(String[] args) { long[] array = new long[100_000]; for (int i = 0; i < array.length; i++) { array[i] = i + 1; }
ForkJoinPool pool = new ForkJoinPool(); try { SumTask task = new SumTask(array, 0, array.length); long result = pool.invoke(task); // 执行任务 System.out.println("总和: " + result); } finally { pool.shutdown(); // 关闭线程池 } }}
最佳实践与注意事项
ForkJoinPool.commonPool(),避免重复创建线程池。基本上就这些。ForkJoinPool 特别适合递归类并行计算,比如归并排序、快速排序、树遍历等场景。掌握它的使用方式,能有效提升复杂计算任务的性能。