import java.util.ArrayList;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.RecursiveTask;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.ForkJoinTask;
public class Tets {//计算数列的和, 需要返回结果,可以定义任务继承 RecursiveTaskprivate static class CountTask extends RecursiveTask<Long> {private static final int THRESHOLD = 10000; //定义数据规模的阈值,允许计算10000 个数内的和,超过该阈值的数列就需要分解private static final int TASKNUM = 100; //定义每次把大任务分解为 100 个小任务private long start; //计算数列的起始值private long end; //计算数列的结束值public CountTask(long start, long end) {this.start = start;this.end = end;}//重写 RecursiveTask 类的 compute()方法,计算数列的结果@Overrideprotected Long compute() {long sum = 0 ; //保存计算的结果
//判断任务是否需要继续分解,如果当前数列 end 与 start 范围的数超过阈值THRESHOLD,就需要继续分解if ( end - start < THRESHOLD){
//小于阈值可以直接计算for (long i = start ; i <= end; i++){sum += i;}}else { //数列范围超过阈值,需要继续分解
//约定每次分解成 100 个小任务,计算每个任务的计算量long step = (start + end ) / TASKNUM;
//start = 0 , end = 200000, step = 2000, 如果计算[0,200000]范围内数列的// 和, 把该范围的数列分解为 100 个小任务,每个任务计算 2000 个数即可
//注意,如果任务划分的层次很深,即 THRESHOLD 阈值太小,每个任务的计//               算量很小,层次划分就会很深,可能出现两种情况:一是系统内的线程数量会越积越多,导致性//               能下降严重; 二是分解次数过多,方法调用过多可能会导致栈溢出
//创建一个存储任务的集合ArrayList<CountTask> subTaskList = new ArrayList<>();long pos = start; //每个任务的起始位置for (int i = 0; i < TASKNUM; i++) {long lastOne = pos + step; //每个任务的结束位置
//调整最后一个任务的结束位置if ( lastOne > end ){lastOne = end;}
//创建子任务CountTask task = new CountTask(pos, lastOne);
//把任务添加到集合中subTaskList.add(task);
//调用 for()提交子任务task.fork();
//调整下个任务的起始位置pos += step + 1;}
//等待所有的子任务结束后,合并计算结果for (CountTask task : subTaskList) {sum += task.join(); //join()会一直等待子任务执行完毕返回执行结果}}return sum;}}public static void main(String[] args) {
//创建 ForkJoinPool 线程池ForkJoinPool forkJoinPool = new ForkJoinPool();
//创建一个大的任务CountTask task = new CountTask(0L, 200000L);
//把大任务提交给线程池ForkJoinTask<Long> result = forkJoinPool.submit(task);try {Long res = result.get(); //调用任务的 get()方法返回结果System.out.println("计算数列结果为:" + res);} catch (InterruptedException e) {e.printStackTrace();} catch (ExecutionException e) {e.printStackTrace();}
//验证long s = 0L;for (long i = 0; i <= 200000 ; i++) {s += i;}System.out.println(s);}
}