一,概述
相比传统的线程池ExecuteService,ForkJoinPool的优势在于能采用分治算法、工作窃取算法高效利用CPU资源,如下图
Fork即拆分,Join即合并,
通过将大任务拆分成多个小任务,在多个线程中执行后,合并结果即可得到大任务的结果,经典的例子有归并排序、超大数组求和等。
ForkJoinPool主要mode如下:
ForkJoinPool:ForkJoin线程池。
ForkJoinTask:ForkJoin#FutureTask。
ForkJoinThread:ForkJoin特殊线程,用于工作窃取。
二,简单实例
public class Main {
private static class BigTask extends java.util.concurrent.ForkJoinTask<Long> {
private long result;
@Override
public Long getRawResult() {
return result;
}
@Override
protected void setRawResult(Long value) {
result = value;
}
@Override
protected boolean exec() {
if (canCompute()) {
result = compute();
} else {
result = forkTask();
}
return true;
}
private long forkTask() {
BigTask bigTask1 = new BigTask();
BigTask bigTask2 = new BigTask();
//拆分多个任务
bigTask1.fork();
bigTask2.fork();
//合并任务结果
return bigTask1.join() + bigTask2.join();
}
private long compute() {
//已经是最小单位,单线程计算
return 10;
}
private boolean canCompute() {
//是否可再拆分?
return false;
}
}
;
public static void main(String[] args) throws ExecutionException, InterruptedException {
//创建线程池
ForkJoinPool pool = new ForkJoinPool();
BigTask bigTask = new BigTask();
pool.submit(bigTask);
System.out.println("result =" + bigTask.get());
}
}
三,简单原理
1,工作窃取
每个Thread都有对应一个workqueue,这区别于传统线程池只有一个Queue的概念。
意思是多个ForkJoinTask可能会被某一个ForkJoinThread执行,如果某ForkJoinThread#workQueue为空,则会去ForkThreadPool找某ForkJoinThread中还未处理的任务,一般是从队列尾窃取,因此称为工作窃取,能大幅利用多核CPU资源。
2,拆分任务
使用ForkJoinTask#fork api,在ForkJoinThread环境下,直接加入当前线程工作队列,
3,执行任务
ForkJoinTask被调度执行时,会调用doExec方法,子类需重写exec方法实现分治任务
4,合并任务结果
可以理解为没有异常的get方法,