JUC系列Fork/Join框架之概览

Posted 顧棟

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了JUC系列Fork/Join框架之概览相关的知识,希望对你有一定的参考价值。

JUC系列Fork/Join框架

文章目录


Fork/Join框架是Java并发工具包中的一种可以将一个大任务拆分为很多小任务来异步执行的工具,自JDK1.7引入。

Fork/Join总览类图

主要模块

ForkJoinPool可以通过池中的ForkJoinWorkerThread来处理ForkJoinTask任务。

ForkJoinPool只接收ForkJoinTask任务(在实际使用中,也可以接收Runnable/Callable任务,但在真正运行时,也会把这些任务封装成ForkJoinTask类型的任务),RecursiveTaskForkJoinTask的子类,是一个可以递归执行的ForkJoinTaskRecursiveAction是一个无返回值的RecursiveTask,CountedCompleter 在任务完成执行后会触发执行一个自定义的钩子函数。

在实际运用中,我们一般都会继承 RecursiveTaskRecursiveActionCountedCompleter 来实现我们的业务需求,而不会直接继承ForkJoinTask类。

核心类

任务对象:

ForkJoinTask

任务管理的线程池

ForkJoinPool

执行任务的线程

ForkJoinWorkerThread

核心算法:分治算法

1.fork 1.fork 2.fork 2.fork 2.fork 2.fork 3.join 3.join 3.join 3.join 4.join 4.join mainTaskA TaskB TaskC TaskB-1 TaskB-2 TaskC-1 TaskC-2

核心算法:工作窃取法

work-stealing(工作窃取)算法: 线程池内的所有工作线程都尝试找到并执行已经提交的任务,或者是被其他活动任务创建的子任务(如果不存在就阻塞等待)。这种特性使得 ForkJoinPool 在运行多个可以产生子任务的任务,或者是提交的许多小任务时效率更高。尤其是构建异步模型的 ForkJoinPool 时,对不需要合并(join)的事件类型任务也非常适用。

在 ForkJoinPool 中,线程池中每个工作线程(ForkJoinWorkerThread)都对应一个任务队列(WorkQueue),工作线程优先处理来自自身队列的任务(LIFO或FIFO顺序,参数 mode 决定),然后以FIFO的顺序随机窃取其他队列中的任务。

使用示例

RecursiveAction使用

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.RecursiveAction;

public class SortTask extends RecursiveAction 
    final long[] array;
    final int low;
    final int high;
    private int THRESHOLD = 0;

    public SortTask(long[] array) 
        this.array = array;
        this.low = 0;
        this.high = array.length - 1;
    

    public SortTask(long[] array, int low, int high) 
        this.array = array;
        this.low = low;
        this.high = high;
    

    @Override
    protected void compute() 
        System.out.println("[" + new SimpleDateFormat("HH:mm:ss.SSS").format(new Date()) + "--" + Thread.currentThread().getName() + "] task is [low=" + low + ", high=" + high + "].");

        if (low < high) 
            int pivot = partition(array, low, high);
            SortTask left = new SortTask(array, low, pivot - 1);
            SortTask right = new SortTask(array, pivot + 1, high);
            left.fork();
            right.fork();
            left.join();
            right.join();
        
    

    private int partition(long[] array, int low, int high) 
        long x = array[high];
        int i = low - 1;
        for (int j = low; j < high; j++) 
            if (array[j] <= x) 
                i++;
                swap(array, i, j);
            
        
        swap(array, i + 1, high);
        return i + 1;
    

    private void swap(long[] array, int i, int j) 
        if (i != j) 
            long temp = array[i];
            array[i] = array[j];
            array[j] = temp;
        
    

    public static void main(String[] args) throws Exception 
        long[] array = 10, 0, 21, 44, 52, 30;
        SortTask sort = new SortTask(array);
        ForkJoinPool pool = new ForkJoinPool();
        pool.submit(sort);
        boolean flag = true;
        while (flag) 
            if (sort.isDone()) 
                StringBuilder a = new StringBuilder();
                for (int i = 0; i < sort.array.length; i++) 
                    a.append(sort.array[i]).append(" ");
                
                System.out.println("[" + new SimpleDateFormat("HH:mm:ss.SSS").format(new Date()) + "--" + Thread.currentThread().getName() + "] a is [" + a.substring(0, a.length() - 1) + "].");
                flag = false;
            
        
        pool.shutdown();
    

执行结果

[18:06:06.143--ForkJoinPool-1-worker-1] task is [low=0, high=5].
[18:06:06.143--ForkJoinPool-1-worker-2] task is [low=0, high=2].
[18:06:06.143--ForkJoinPool-1-worker-2] task is [low=0, high=1].
[18:06:06.143--ForkJoinPool-1-worker-2] task is [low=0, high=-1].
[18:06:06.143--ForkJoinPool-1-worker-2] task is [low=1, high=1].
[18:06:06.143--ForkJoinPool-1-worker-2] task is [low=3, high=2].
[18:06:06.159--ForkJoinPool-1-worker-2] task is [low=4, high=5].
[18:06:06.159--ForkJoinPool-1-worker-2] task is [low=4, high=3].
[18:06:06.159--ForkJoinPool-1-worker-2] task is [low=5, high=5].
[18:06:06.159--main] a is [0 10 21 30 44 52].

RecursiveTask使用

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.Future;
import java.util.concurrent.RecursiveTask;

public class SumTask extends RecursiveTask<Long> 
    private static final int THRESHOLD = 10;

    private long start;

    private long end;

    public SumTask(long end) 
        this(0, end);
    

    private SumTask(long start, long end) 
        this.start = start;
        this.end = end;
    

    @Override
    protected Long compute() 
        long sum = 0;
        if ((end - start) <= THRESHOLD) 
            System.out.println("[" + new SimpleDateFormat("HH:mm:ss.SSS").format(new Date()) + "--" + Thread.currentThread().getName() + "] task is [" + start + "," + end + "].");

            for (long l = start; l <= end; l++) 
                sum += l;
            
         else 
            long mid = (start + end) >>> 1;
            SumTask left = new SumTask(start, mid);
            SumTask right = new SumTask(mid + 1, end);
            left.fork();
            right.fork();
            sum = left.join() + right.以上是关于JUC系列Fork/Join框架之概览的主要内容,如果未能解决你的问题,请参考以下文章

JUC并发编程 共享模式之工具 线程池 -- Fork / Join 框架(JDK1.7 新加入的线程池实现)

Java——多线程高并发系列之Fork/Join框架简单应用

Java——多线程高并发系列之Fork/Join框架简单应用

13JUC--ForkJoinPool 分支/合并框架 工作窃取(未完成)

JUC系列ForkJoin框架设计官方说明翻译

JDK1.7之Fork/join