Java JUC工具类--ForkJoin

ForkJoin

• Fork/Join框架是JAVA7提供的一个用于并行执行任务的框架，是一个把大任务分割成若干个小任务，最终汇总每个小任务结果后得到大任务结果的框架

 

Fork/Join中两个重要的类

• ForkJoinTask：使用该框架，需要创建一个ForkJoin任务，它提供在任务中执行fork和join操作的机制。
• 一般情况下，我们并不需要直接继承ForkJoinTask类，只需要继承它的子类，它的子类有两个：
• RecursiveAction:用于没有返回结果的任务
• RecursiveTask:用于有返回结果的任务
• ForkJoinPool：任务ForkJoinTask需要通过ForkJoinPool来提交任务

package com.example.core.juc;

import java.util.concurrent.*;

public class UseForkJoin extends RecursiveTask<Integer> {//执行计算并要求返回结果的需要继承RecursiveTask<Integer>
    //RecursiveTask<Integer>其中Integer为计算的类型
    private static final int THRESHOLD = 2;//阈值，也就是拆分的原则
    private int start;
    private int end;

    public UseForkJoin(int start,int end){
        this.start = start;
        this.end = end;
    }

    @Override
    protected Integer compute(){
        int sum = 0; //1+2+3+......+100
        boolean canCompute = (end - start) <= THRESHOLD;

        if(canCompute){
            for(int i = start;i<=end;i++){
                sum += i;
            }
        }else{
            //如果，任务数大于阈值的情况，就进行拆分 fork操作 然后去join
            //1 + 100/2 = 50
            int middle = (start + end)/2;
            UseForkJoin leftTask = new UseForkJoin(start,middle);
            UseForkJoin rightTask = new UseForkJoin(middle+1,end);
            //执行左右两边的任务
            leftTask.fork();
            rightTask.fork();
            //等待任务执行完成之后，进行获取结果
            int leftResult = leftTask.join();
            int rightResult = rightTask.join();
            sum = leftResult + rightResult;
        }
        return sum;
    }

    public static void main(String[] args) throws InterruptedException, ExecutionException {
        ForkJoinPool pool = new ForkJoinPool();
        UseForkJoin ufj = new UseForkJoin(1,100);
        Future<Integer> result = pool.submit(ufj);
        System.out.println("最终的结果是:"+result.get());
    }
}
/*
output:
最终的结果是:5050
 */