ForkJoinPool浅析

一，概述

相比传统的线程池ExecuteService，ForkJoinPool的优势在于能采用分治算法、工作窃取算法高效利用CPU资源，如下图

Fork即拆分，Join即合并，

通过将大任务拆分成多个小任务，在多个线程中执行后，合并结果即可得到大任务的结果，经典的例子有归并排序、超大数组求和等。

ForkJoinPool主要mode如下：

ForkJoinPool：ForkJoin线程池。

ForkJoinTask：ForkJoin#FutureTask。

ForkJoinThread：ForkJoin特殊线程，用于工作窃取。

二，简单实例

public class Main {

    private static class BigTask extends java.util.concurrent.ForkJoinTask<Long> {
        private long result;

        @Override
        public Long getRawResult() {
            return result;
        }

        @Override
        protected void setRawResult(Long value) {
            result = value;
        }

        @Override
        protected boolean exec() {
            if (canCompute()) {
                result = compute();
            } else {
                result = forkTask();
            }
            return true;
        }

        private long forkTask() {
            BigTask bigTask1 = new BigTask();
            BigTask bigTask2 = new BigTask();

            //拆分多个任务
            bigTask1.fork();
            bigTask2.fork();

            //合并任务结果
            return bigTask1.join() + bigTask2.join();
        }

        private long compute() {
            //已经是最小单位，单线程计算
            return 10;
        }

        private boolean canCompute() {
            //是否可再拆分？
            return false;
        }
    }

    ;

    public static void main(String[] args) throws ExecutionException, InterruptedException {
        //创建线程池
        ForkJoinPool pool = new ForkJoinPool();


        BigTask bigTask = new BigTask();

        pool.submit(bigTask);

        System.out.println("result =" + bigTask.get());
    }
}

三，简单原理

1，工作窃取

每个Thread都有对应一个workqueue，这区别于传统线程池只有一个Queue的概念。

意思是多个ForkJoinTask可能会被某一个ForkJoinThread执行，如果某ForkJoinThread#workQueue为空，则会去ForkThreadPool找某ForkJoinThread中还未处理的任务，一般是从队列尾窃取，因此称为工作窃取，能大幅利用多核CPU资源。

2，拆分任务

使用ForkJoinTask#fork api，在ForkJoinThread环境下，直接加入当前线程工作队列，

3，执行任务

ForkJoinTask被调度执行时，会调用doExec方法，子类需重写exec方法实现分治任务

4，合并任务结果

可以理解为没有异常的get方法，