简介
ForkJoinPool的优势在于,可以充分利用多cpu,多核cpu的优势,把一个任务拆分成多个“小任务”,把多个“小任务”放到多个处理器核心上并行执行,当多个“小任务”执行完成之后,再将这些执行结果合并起来即可。
适用场景:最适合的是计算密集型的任务,如果存在 I/O,线程间同步,sleep() 等会造成线程长时间阻塞的情况。
代码
static int[] nums=new int[1000000];
static final int MAX_NUM=500;
static Random r=new Random(); static {
for (int i = 0; i < nums.length; i++) {
nums[i]=r.nextInt(MAX_NUM);
}
System.out.println("nums总和为:-->"+ Arrays.stream(nums).sum());
} /**
* 继承RecursiveTask有返回值
*
*
**/
public static class AddTaskRet extends RecursiveTask<Long>{ private static final long serialVersionUID = 1L;
int start, end; AddTaskRet(int s, int e) {
start = s;
end = e;
} @Override
protected Long compute() {
//如果计算的数小于500,则一个线程进行计算即可
if(end-start <= MAX_NUM) {
long sum = 0L;
for(int i=start; i<end; i++){
sum += nums[i];
}
return sum;
}
//大于500,这里分成两个线程去进行计算
int middle = start + (end-start)/2;
AddTaskRet subTask1 = new AddTaskRet(start, middle);
AddTaskRet subTask2 = new AddTaskRet(middle, end);
//执行两个任务
subTask1.fork();
subTask2.fork();
//把两个小任务累加的结果合并起来
return subTask1.join() + subTask2.join();
}
} /**
* 继承RecursiveAction,无返回值
*
*
**/
public static class AddTask extends RecursiveAction{
int start, end;
AddTask(int s, int e) {
start = s;
end = e;
} @Override
protected void compute() {
if(end-start <= MAX_NUM) {
long sum = 0L;
for(int i=start; i<end; i++){
sum += nums[i];
}
System.out.println("from:" + start + " to:" + end + " = " + sum);
} else { int middle = start + (end-start)/2; AddTask subTask1 = new AddTask(start, middle);
AddTask subTask2 = new AddTask(middle, end);
subTask1.fork();
subTask2.fork();
}
}
} /**
* @Description:
* @Author: zhuyang
* @Date: 2022/1/17 19:36
* @return: void
**/
public void poolDemo(){
ForkJoinPool fjp = new ForkJoinPool();
AddTaskRet task = new AddTaskRet(0, nums.length);
//进行执行任务
fjp.execute(task);
//获取结果
long result = task.join();
System.out.println("多任务分片执行结果为---->"+result);
}