ForkJoin分支合并框架

一、简介

从JDK1.7开始,Java提供ForkJoin框架用于并行执行任务,它的思想就是讲一个大任务分割成若干小任务,最终汇总每个小任务的结果得到这个大任务的结果。

1、ForkJoinPool

既然任务是被逐渐的细化的,那就需要把这些任务存在一个池子里面,这个池子就是ForkJoinPool,它与其它的ExecutorService区别主要在于它使用“工作窃取“,那什么是工作窃取呢?

一个大任务会被划分成无数个小任务,这些任务被分配到不同的队列,这些队列有些干活干的块,有些干得慢。于是干得快的,一看自己没任务需要执行了,就去隔壁的队列里面拿去任务执行。

2、ForkJoinTask

ForkJoinTask就是ForkJoinPool里面的每一个任务。他主要有两个子类:RecursiveAction和RecursiveTask。然后通过fork()方法去分配任务执行任务,通过join()方法汇总任务结果,

这就是整个过程的运用。他有两个子类,使用这两个子类都可以实现我们的任务分配和计算。

(1)RecursiveAction 一个递归无结果的ForkJoinTask(没有返回值)

(2)RecursiveTask 一个递归有结果的ForkJoinTask(有返回值)

ForkJoinPool由ForkJoinTask数组和ForkJoinWorkerThread数组组成,ForkJoinTask数组负责存放程序提交给ForkJoinPool的任务,而ForkJoinWorkerThread数组负责执行这些任务。

下面我们就来看看如何去使用。

class MyTake extends RecursiveTask<Integer> {

    private static final Integer ADJUST_VALUE = 10;
    private int begin;
    private int end;
    private int result;

    public MyTake(int begin, int end) {
        super();
        this.begin = begin;
        this.end = end;
    }

    // 任务小到不能计算,就直接计算
    @Override
    protected Integer compute() {
        if ((end - begin) <= ADJUST_VALUE) {
            for (int i = begin; i <= end; i++) {
                result = result + i;
            }
        } else {
            int moddle = (end + begin) / 2;
            MyTake take1 = new MyTake(begin, moddle);
            MyTake take2 = new MyTake(moddle + 1, end);
            take1.fork();
            take2.fork();
            result = take1.join() + take2.join();
        }

        return result;
    };
}

/*
 * 分支合并框架 ForkJoinPoll ForkJoinTask RecursiveTask
 */
public class ForkJoinDemo {

    public static void main(String[] args) throws InterruptedException, ExecutionException {
        MyTake myTake = new MyTake(0, 100);

        ForkJoinPool forkJoinPool = new ForkJoinPool();
        ForkJoinTask<Integer> forkJoinTask = forkJoinPool.submit(myTake);
        System.out.println(forkJoinTask.get());
        forkJoinPool.shutdown();

    }
}
原文地址:https://www.cnblogs.com/fengyangcai/p/12911265.html