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;}@Overrideprotected Integer compute() {// TODO Auto-generated method stubint 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 blocke.printStackTrace();}}}

 

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