廖雪峰Java11多线程编程-3高级concurrent包-9Fork_Join
1. Fork/Join模式
线程池可以高效执行大量小任务:
Fork/Join线程池可以执行一种特殊的任务:
- 把一个大任务拆成多个小任务并行执行
- Fork/Join是在JDK 1.7引入的
Fork/Join模式的应用:
- java.util.Arrays.parallelSort(array):方法内部通过Fork/Join对一个大数组进行并行排序,在多核CPU上就可以大大提高排序的速度。
例如:计算一个有1000万个元素数组的和。
如果把一个大数组拆成2个数组,就可以利用双核CPU并行计算,最后把2个结果相加就是最终的结果。
如果拆成2个数组以后,每个数组仍然很大,可以进一步拆分成4个数组,我们就可以让4核CPU并行执行。
Fork/Join就是把一个大任务不断的拆成小任务,执行并行计算的一种模式。
class SumTask extends RecursiveTask<Long> {
@Override
protected Long compute(){
//把一个大任务分拆成2个子任务
SumTask subtask1 = new Sumtask(...);
SumTask subtask2 = new Sumtask(...);
//调用invokeAll()同时运行2个小任务,当2个任务都完成以后,invokeAll才返回
invokeAll(subtask1, subtask2);
//通过join()获得2个子任务的结果
Long result1 = subtask1.join();
Long result2 = subtask2.join();
return result1 + result2; //返回结果
}
}
使用Fork/Join的关键在于,在compute方法内部,我们需要把一个大任务分拆成2个小任务,然后调用invokeAll()这个方法同时运行2个小任务。
当2个任务都运行结束以后,invokeAll()才会返回。接着,我们调用join()方法获得2个小任务的执行结果,最后把2个结果相加返回。
Recursive可以不断的把自身拆成小任务并行执行
2.示例
1.创建长度为1000的随机数组成的数组,并计算和作为期望值
2.使用SumTask创建一个ForkJoinTask对象。
3.通过ForkJoinPool.commonPool()方法获得一个commonPool(),然后用invoke来执行这个任务
4.求和的任务SumTask继承制RecursiveTask,返回值是Long类型,指定常量是THREHOLD。构造方法是传入一个数组、开始索引、结束索引。
5.SumTask的compute方法:我们首先判断任务是否足够的小。是,就直接进行计算,并且返回计算的结果;否,即继续拆分,这样我们就获得了subtask1,subtask2 这两个子任务。紧接着调用invokeAll()方法,同时执行这两个子任务。然后我们用join()方法获得两个子任务的结果,相加返回。
import java.awt.*;
import java.util.Random;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.ForkJoinTask;
import java.util.concurrent.RecursiveTask;
class SumTask extends RecursiveTask<Long> {//求和的任务继承至RecursiveTask,返回值是Long类型
//执行常量THRESHOLD,用它来分解任务
static final int THRESHOLD = 250;
long[] array;
int start;
int end;
SumTask(long[] array,int start,int end){
//创建SumTask时,传入一个数组,起始结束位置的索引
this.array = array;
this.start = start;
this.end = end;
}
public Long compute(){
//执行compute方法时,任务足够小,直接计算返回执行的结果
if(end - start <= THRESHOLD){
long sum = 0;
for(int i=start;i<end;i++){
sum += this.array[i];
try{
Thread.sleep(2);
}catch (InterruptedException e){}
}
return sum;
}
//如果任务太大,就一分为二,拆成2个子任务
int middle = (start + end) / 2;
System.out.println(String.format("split %d-%d ==> %d~%d ,%d~%d",start,end,start,middle,middle,end));
SumTask subtask1 = new SumTask(this.array,start,middle);
SumTask subtaks2 = new SumTask(this.array,middle,end);
//调用invokeAll同时执行这两个任务
invokeAll(subtask1,subtaks2);
//用join获取子任务的结果
Long subresult1 = subtask1.join();
Long subresult2 = subtaks2.join();
Long result = subresult1 + subresult2;
System.out.println(String.format("result = %d + %d ==> %d",subresult1,subresult2,result));
return result;
}
}
public class ForkJoinTaskSample {
//对一个大数组进行求和
public static void main(String[] args) throws Exception{
long[] array = new long[1000]; //创建一个包含1000个元素的数组
long expectedSum = 0;
for(int i=0;i<array.length;i++){
//1.创建数组的过程中,并计算数组的和作为期望值
array[i] = random();
expectedSum += array[i];
}
System.out.println("expectedSum: "+expectedSum);
//创建一个ForkJoinTask
ForkJoinTask<Long> task = new SumTask(array,0,array.length);
Long startTime = System.currentTimeMillis();
//通过ForkJoinPool.commonPool获得一个ForkJoinPool,用invoke方法执行这个任务
Long result = ForkJoinPool.commonPool().invoke(task);
Long endTime = System.currentTimeMillis();
System.out.println("Fork/join sum: "+result+" in "+(endTime-startTime));
}
static Random random = new Random(0);
static long random(){
return random.nextInt(10000);
}
}
![](https://img2018.cnblogs.com/blog/1418970/201906/1418970-20190618200233414-624631069.png)
3. 总结:
- Fork/Join是一种基于分治的算法:分解任务+合并结果
- Fork/JoinPool线程池可以把一个大任务拆成小任务并行执行
- 任务类必须继承自RecursiveTask/RecursiveAction。
* RecursiveTask有返回值;RecursiveAction没有返回值 - 使用Fork/Join模式可以进行并行计算提高效率