ForkAndJoin实践

Java的并发编程模型中，有个Fork-Join框架，目的是将大的任务分解成小的任务进行计算，然后再把每个小任务的计算结果汇总。
Fork / Join 框架提供了两个可供继承的子类
1，RecursiveAction 用以分解没有计算结果的任务
2，RecursiveTask    用以分解有计算结果的任务
 
接下来计算一个1加到100的程序，体会一下Fork/Join框架的使用
 
首先定义一个任务类CountTask类，因为是要计算结果，因此继承RecursiveTask类，主要是实现其compute方法，这里设置的阀值THRESHOLD值为10，表示每个任务计算的加数的个数最多为10个，接下来就是将任务根据阀值分组，并对每个分组进行计算
 

package com.luchi.thread.forkAndJoin;

import java.util.concurrent.RecursiveTask;

public class CountTask extends RecursiveTask<Integer>{

private static final int THRESHOLD=10;//设置阀值
private int start;
private int end;

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

@Override
protected Integer compute() {
// TODO Auto-generated method stub
int sum=0;
//看任务是不是在阀值以下
boolean canComputs=(end-start+1)<=THRESHOLD;
if(canComputs){
for(int i=start;i<=end;i++){
sum+=i;
}
return sum;
}else{
int size=end-start+1;
int taskNum;
if(size%THRESHOLD==0){
taskNum=size/THRESHOLD;
}else{
taskNum=size/THRESHOLD+1;
}
for(int i=1;i<=taskNum;i++){

int groupBegin=start+(i-1)*THRESHOLD;
int groupEnd=Math.min(groupBegin+THRESHOLD-1, end);
CountTask childTask=new CountTask(groupBegin, groupEnd);
childTask.fork();
int childCount=childTask.join();
sum+=childCount;
}
return sum;
}
}

}

 
 
看一下测试程序TestForkAndJoin类，fork/join需要使用ForkJoinPool容器，这个容器负责执行每个task
 

package com.luchi.thread.forkAndJoin;

import java.util.concurrent.ExecutionException;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.Future;

public class TestForkAndJoin {

public static void main(String[]args){

ForkJoinPool forkJoinPool=new ForkJoinPool();
CountTask task=new CountTask(1,100);
Future<Integer> result=forkJoinPool.submit(task);
try {
System.out.println(result.get());
} catch (InterruptedException | ExecutionException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}

}

}

 
其实这个模型和FutureTask使用方法差不多，不同的是，FutureTask是单线程使用，而ForkJoinPool里面保持了一个ForkJoinTask数组用来保存多个任务，同时ForkJoinWorkerThread用来执行任务，我的理解就是ForkJoin框架提供了一个多线程的执行环境用来执行多个任务而已，而其join方法和其他地方的join方法相同