ForkJoin 框架是Java7提供的用于并行执行任务的框架,是把一个大任务分割成若干个小任务,最终汇总每个小任务结果后得到大任务结果的
框架。要想使用ForkJoin框架,必须首先创建一个ForkJoin任务,称之为ForkJoinTask,提供Fork和Join操作的机制,不需要直接继承
ForkJoinTask类,只需要继承它的子类。子类有两个:
RecursiveAction: 用于返回没有结果的任务
RecursiveTask:用于返回有结果的任务
ForkJoinPool:ForkJoinTask任务需要通过ForkJoinPool来执行
下面代码:以1+2+3+4 为例子
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.Future;
import java.util.concurrent.RecursiveTask;
public class CountTask extends RecursiveTask<Integer>{
//阀值 用于两个数的相加 a+b
private static final int THRESHOLD = 2;
private int start;
private int end;
public CountTask(int start, int end){
this.start = start;
this.end = end;
}
@Override
protected Integer compute() {
System.out.println("Come in ");
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;
CountTask leftTask = new CountTask(start, middle);
CountTask rightTask = new CountTask(middle+1, end);
//执行子任务
leftTask.fork();
rightTask.fork();
//等待子任务自行完 并得到其结果
int leftResuklt = leftTask.join();
int rightResult = rightTask.join();
//合并子任务
sum = leftResuklt + rightResult;
}
return sum;
}
public static void main(String[] args) {
ForkJoinPool forkJoinPool = new ForkJoinPool();
//生成一个计算任务
CountTask task = new CountTask(1,4);
//执行一个任务
Future<Integer> result = forkJoinPool.submit(task);
try {
//检查任务是否已经抛出异常或者已经被取消了
if(task.isCompletedAbnormally())
{
System.out.println(task.getException());
}
System.out.println(result.get());
} catch (Exception e) {
e.printStackTrace();
}
}
}