Fork/Join框架

一、什么是Fork/Join框架

     Fork/Join框架是Java 7提供的一个用于并行执行任务的框架,是一个把大任务分割成若干个小任务,最终汇总每个小任务结果后得到大任务结果的框架。Fork/Join框架要完成两件事情:

  1.任务分割:首先Fork/Join框架需要把大的任务分割成足够小的子任务,如果子任务比较大的话还要对子任务进行继续分割

  2.执行任务并合并结果:分割的子任务分别放到双端队列里,然后几个启动线程分别从双端队列里获取任务执行。子任务执行完的结果都放在另外一个队列里,启动一个线程从队列里取数据,然后合并这些数据。

二、Fork/Join框架的设计

在Java的Fork/Join框架中,使用两个类完成上述操作

1、ForkJoinTask:我们要使用Fork/Join框架,首先需要创建一个ForkJoin任务。该类提供了在任务中执行fork和join的机制。通常情况下我们不需要直接集成ForkJoinTask类,只需要继承它的子类,Fork/Join框架提供了两个子类:

  • RecursiveAction:用于没有返回结果的任务
  • RecursiveTask:用于有返回结果的任务

2、ForkJoinPool:ForkJoinTask需要通过ForkJoinPool来执行。

      任务分割出的子任务会添加到当前工作线程所维护的双端队列中,进入队列的头部。当一个工作线程的队列里暂时没有任务时,它会随机从其他工作线程的队列的尾部获取一个任务(工作窃取算法)。

三、工作窃取算法

      工作窃取算法是指某个线程从其他队列里窃取任务来执行。

      通俗理解:干完活的线程与其等着,不如帮助其他线程干活,于是它就去其他线程队列中窃取一个任务来执行。

      为了减少窃取任务线程与被窃取线程之间的造成竞争,使用双端队列,被窃取任务线程从双端队列的头部拿任务执行,而窃取任务的线程从双端队列的尾部拿任务执行。
 * 优点:充分利用线程进行并行计算,减少了线程间的竞争。
 * 缺点:当双端队列里只有一个任务时,存在竞争。该算法会消耗过多的系统资源,创建了多个线程和双端队列。

四、使用Fork/Join框架

          通过一个简单的需求来使用Fork/Join框架,需求是:计算1+2+3+4的结果

          分析:设置分割的阈值是2,由于是4个数字相加,所以Fork/Join框架会把这个任务fork分成两个子任务,子任务一负责计算1+2、子任务二负责计算3+4,然后再join两个子任务的结果。因为有返回结果的任务,所以必须继承RecursiveTask类。

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

public class CountTask extends RecursiveTask<Integer>{
	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() {
		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 leftResult = leftTask.join();
			int rightResult = rightTask.join();
			//合并子任务
			sum = leftResult + rightResult;
		}
		return sum;
	}
	public static void main(String[] args) {
		ForkJoinPool forkJoinPool = new ForkJoinPool();
		//生成一个计算任务,负责计算1+2+3+4
		CountTask task = new CountTask(1, 4);
		//执行一个任务
		Future<Integer> result = forkJoinPool.submit(task);
		try {
			System.out.println(result.get());
		}catch(InterruptedException e){			
		}catch(ExecutionException e){			
		}
	}	
}

      通过这个例子,我们进一步了解ForkJoinTask,ForkJoinTask与一般任务的主要区别在于它需要实现compute方法,在这个方法里,首先需要判断任务是否足够小,如果足够小就直接执行任务。如果不足够小,就必须分割成两个子任务,每个子任务在调用fork方法时,又会进入compute方法,看看当前子任务是否需要继续分割成子任务,如果不需要继续分割,则执行当前子任务并返回结果。使用join方法会等待子任务执行完并得到其结果。

五、Fork/Join框架的异常处理

      ForkJoinTask在执行的时候可能会抛出异常,但是我们没办法在主线程里直接捕获异常,所以ForkJoinTask提供了isCompletedAbnormally()方法来检查任务是否已经抛出异常或已经被取消了,并且可以通过ForkJoinTask 的 getException方法获取异常。使用如下代码:

if(task.isCompletedAbnormally()){
    System.out.println(task.getException());
}

getException方法返回Throwable对象,如果任务被取消了则返回CancellationException。如果任务没有完成或者没有抛出异常则返回null。

六、ForkJoinTask的join方法实现原理

      ForkJoinTask需要通过ForkJoinPool来执行,ForkJoinPool由ForkJoinTask数组和ForkJoinWorkerThread数组组成,ForkJoinTask数组负责将存放程序提交给ForkJoinPool,而ForkJoinWorkerThread负责执行这些任务。

(1)ForkJoinTask的Fork方法的实现原理:
  当我们调用ForkJoinTask的fork方法时,程序会把任务放在ForkJoinWorkerThread的pushTask的workQueue中,异步地执行这个任务,然后立即返回结果,代码如下:

public final ForkJoinTask<V> fork() {
       
       ((ForkJoinWorkerThread)Thread.currentThread()).pushTask(this);

        return this;
    }

 pushTask方法把当前任务存放在ForkJoinTask数组队列里。然后再调用ForkJoinPool的signalWork()方法唤醒或创建一个工作线程来执行任务。

(2)join方法的主要作用是阻塞当前线程并等待获取结果。让我们一起看看ForkJoinTask的join方法的实现,代码如下:

public final V join(){
    if(doJoin()!= NORMAL)
        return reportResult();
    else
        return getRawResult();
}

private V reportResult(){
    int s;
    Throwable ex;
    if((s==status) == CANCELLED) 
        throw new CanclelationException();
    if(s==EXCEPTION && (ex = getThrowableException())!= null)  
       UNSAFE.throwException(ex);
    return getRawResult();
}

      它首先调用doJoin方法,通过doJoin()方法得到当前任务的状态来判断返回什么结果,任务状态有4种:已完成(NORMAL)、被取消(CANCELLED)、信号(SIGNAL)和出现异常(EXCEPTIONAL)。

  •   如果任务状态是已完成,则直接返回任务结果。
  •   如果任务状态是被取消,则直接抛出CancellationException
  •   如果任务状态是抛出异常,则直接抛出对应的异常

  让我们分析一下doJoin方法的实现:

       在doJoin()方法里,首先通过查看任务的状态,看任务是否已经执行完成,如果执行完成,则直接返回任务状态;如果没有执行完,则从任务数组里取出任务并执行。如果任务顺利执行完成,则设置任务状态为NORMAL,如果出现异常,则记录异常,并将任务状态设置为EXCEPTIONAL。

八、性能分析

      因为Fork/Join框架执行的时候,需要不断地拆分任务,又不断地合并任务结果,这些也是需要消耗一点时间的,而任务不大可以很快就能执行完的情况下,它的fork和join操作无疑占据整个程序运行时间比例比较大。当任务越大时,Fork/Join框架的优势就越明显。

      拆分的临界值的设定对Fork/Join框架执行的速度也很关键,不能太大,也不能太小,太大了没有充分拆分任务压榨CPU,太小了就会拆分过度,浪费太多时间在拆分上。

借鉴书籍和博客:《java并发编程的艺术》、https://blog.csdn.net/TimHeath/article/details/71307834

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值