一、介绍
题中说的fork join框架主要是指java中的ava.util.concurrent.ForkJoinPool类。该类的使用场景特别适合用于“分而治之”,递归计算的算法;
二、用法
使用RecursiveTask实现一个累加的功能,使用分而治之的思想,实现分段求和后汇总:
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.RecursiveTask;
class SumTask extends RecursiveTask<Integer> {
private static final long serialVersionUID = -5141593245600458833L;
private Integer start = 0;
private Integer end = 0;
public SumTask(int start, int end) {
this.start = start;
this.end = end;
}
@Override
protected Integer compute() {
if (end - start < 100) {
// 小于100时直接返回结果
int sumResult = 0;
for (int i = start; i <= end; i++) {
sumResult += i;
}
return sumResult;
} else {
// 大于一百时进行分割
int middle = (end + start) / 2;
SumTask leftSum = new SumTask(this.start, middle);
SumTask rightSum = new SumTask(middle, this.end);
leftSum.fork();
rightSum.fork();
return leftSum.join() + rightSum.join();
}
}
}
public class ForkJoinPoolTester {
public static void main(String[] args) {
ForkJoinPool forkJoinPool = new ForkJoinPool(10);
SumTask sumTask = new SumTask(1, 1000);
forkJoinPool.submit(sumTask);
System.out.print("result:" + sumTask.join());
}
}
ForkJoinPool 实现了工作窃取算法来提高 CPU 的利用率。每个线程都维护了一个双端队列,用来存储需要执行的任务。工作窃取算法允许空闲的线程从其它线程的双端队列中窃取一个任务来执行。窃取的任务必须是最晚的任务,避免和队列所属线程发生竞争。例如下图中,Thread2 从 Thread1 的队列中拿出最晚的 Task1 任务,Thread1 会拿出 Task2 来执行,这样就避免发生竞争。但是如果队列中只有一个任务时还是会发生竞争。