package com.mmall.concurrency.example.aqs;
import lombok.extern.slf4j.Slf4j;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.Future;
import java.util.concurrent.RecursiveTask;
@Slf4j
public class ForkJoinTaskExample extends RecursiveTask<Integer> {//把大任务拆分成若干个独立小任务
public static final int threshold = 2;
private int start;
private int end;
public ForkJoinTaskExample(int start, int end) {
this.start = start;
this.end = end;
}
@Override
protected Integer compute() {
int sum = 0;
//如果任务足够小就计算任务
boolean canCompute = (end - start) <= threshold;
if (canCompute) {
for (int i = start; i <= end; i++) {
sum += i;
}
} else {//这个不是递归执行,而是根据递归分成小任务,这种递归与执行子任务的后面代码没关系,主线程不会进入递归里按照递归执行
//如果任务大于阈值,就分裂成两个子任务计算
int middle = (start + end) / 2;
ForkJoinTaskExample leftTask = new ForkJoinTaskExample(start, middle);//右边分成若干个个独立任务,并没有按照递归思路计算,而是根据递归分好任务后加入到每个线程的双端队列中吧
ForkJoinTaskExample rightTask = new ForkJoinTaskExample(middle + 1, end);//左边同上;假如左边10个小任务,2个线程,那么每个线程需要计算5个任务,允许窃取任务
//执行子任务
leftTask.fork();//即便1个线程执行5个任务,这5个任务的结果也不会合并
rightTask.fork();
//等待任务执行结束合并其结果
int leftResult = leftTask.join();//假如左边两个线程一共执行了10个任务,那么这10个任务在这句代码合并结果
int rightResult = rightTask.join();
//合并子任务
sum = leftResult + rightResult;
}
return sum;
}
public static void main(String[] args) {
ForkJoinPool forkJoinPool = new ForkJoinPool();
//生成一个计算任务,计算1+2+3+4
ForkJoinTaskExample task = new ForkJoinTaskExample(1, 100);
//执行一个大任务
Future<Integer> result = forkJoinPool.submit(task);
try {
log.info("result:{}", result.get());
} catch (Exception e) {
log.error("exception", e);
}
}
}