java forkjoin 简书_Java并发系列6-Fork/Join框架

一、介绍

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

Fork：把一个大任务切分为若干子任务并行的执行。

Join：合并这些子任务的执行结果，最后得到这个大任务的结果。

Fork/Join示意图

上图中，上部的Task依赖于下部的Task执行，只有当各个子任务执行完成后，才能得到Task0的返回结果。

2、原理

分割任务。需要有一个fork类来将主任务分割成子任务，存在嵌套分隔的情况。

执行任务并合并结果。分割的子任务分别放在各自的双端队列中，之后启动几个线程分别从其所有的双端队列中获取任务执行。子任务执行完的结果都统一放在一个结果队列中，启动一个线程从此队列中获取并合并数据。

Fork/Join使用两个类来完成以上两件事情：

ForkJoinTask：使用ForkJoin框架，必须首先创建一个ForkJoin任务。它提供在任务中执行fork()和join()操作，通常情况下我们不需要直接继承ForkJoinTask类，而只需要继承它的子类，Fork/Join框架提供了以下两个子类：

RecursiveAction：用于没有返回结果的任务。

RecursiveTask ：用于有返回结果的任务。

ForkJoinPool ：ForkJoinTask需要通过ForkJoinPool来执行，任务分割出的子任务会添加到当前工作线程所维护的双端队列中，进入队列的头部。当一个工作线程的队列里暂时没有任务时，它会随机从其他工作线程的队列的尾部获取一个任务。

二、例子

通过一个例子，简单介绍fork/join框架的使用，计算任务是：计算1 + 2 + ... + 100

1、首先是继承类，由于我们需要子任务返回结果，因此选择带有返回结果的RecursiveTask。

2、继承RecursiveTask类后，需要实现其父类的抽象方法compute()。

3、任务分隔， 设定阈值为10，如果超过该阈值，则进行分隔。

public class CountTask extends RecursiveTask {

private static final int THRESHOLD = 10; // 阈值

private int start;

private int end;

public CountTask(int start, int end) {

this.start = start;

this.end = end;

}

@Override

protected Integer compute() {

boolean isNeedSplit = (end - start) > THRESHOLD;

int sum = 0;

if (!isNeedSplit) {

for (int i = start; i <= end; ++i) {

sum += i;

}

return sum;

}

// 分隔任务

int mid = (start + end) >> 1;

CountTask lowTask = new CountTask(start, mid);

CountTask highTask = new CountTask(mid + 1, end);

// 子任务执行

lowTask.fork();

highTask.fork();

// 获取子任务计算结果

return lowTask.join() + highTask.join();

}

public static void main(String[] args) throws Exception {

ForkJoinPool forkJoinPool = new ForkJoinPool();

// 创建新任务

CountTask task = new CountTask(1, 100);

// 提交任务

Future ret = forkJoinPool.submit(task);

// 输出结果

System.out.println(ret.get());

}

}