一文秒懂 Java ForkJoinPool

作者简介：大家好，我是smart哥，前中兴通讯、美团架构师，现某互联网公司CTO

联系qq：184480602，加我进群，大家一起学习，一起进步，一起对抗互联网寒冬

学习必须往深处挖，挖的越深，基础越扎实！

阶段1、深入多线程

阶段2、深入多线程设计模式

阶段3、深入juc源码解析

阶段4、深入jdk其余源码解析

阶段5、深入jvm源码解析

码哥源码部分

码哥讲源码-原理源码篇【2024年最新大厂关于线程池使用的场景题】

码哥讲源码【炸雷啦！炸雷啦！黄光头他终于跑路啦！】

码哥讲源码-【jvm课程前置知识及c/c++调试环境搭建】

​​​​​​码哥讲源码-原理源码篇【揭秘join方法的唤醒本质上决定于jvm的底层析构函数】

码哥源码-原理源码篇【Doug Lea为什么要将成员变量赋值给局部变量后再操作？】

码哥讲源码【你水不是你的错,但是你胡说八道就是你不对了！】

码哥讲源码【谁再说Spring不支持多线程事务，你给我抽他！】

终结B站没人能讲清楚红黑树的历史，不服等你来踢馆！

打脸系列【020-3小时讲解MESI协议和volatile之间的关系，那些将x86下的验证结果当作最终结果的水货们请闭嘴】   

ForkJoinPool 是Java 7 中引入的 fork/join 框架的核心之一。它解决了一个常见的问题： 如何在递归中生成多个任务。因为，即使是使用一个简单的 ThreadPoolExecutor ，也会在不断的递归中快速耗尽线程。因为每个任务或子任务都需要自己的线程来运行。

在 fork/join 框架中，任何任务都可以生成 ( fork ) 多个子任务并使用 join() 方法等待它们的完成。fork/join 框架的好处是它不会为每个任务或子任务创建新线程，而是实现了 工作窃取 ( Work Stealing ) 算法。关于 fork/join 框架的详细信息，你可以访问我们的 一文秒懂 Java Fork/Join。

接下来，我们看一个使用 ForkJoinPool 遍历节点树并计算所有叶值之和的简单示例。在这个示例中，树是一个由节点，int 值和一组子节点组成。

    static class TreeNode {
    
        int value;
    
        Set<TreeNode> children;
    
        TreeNode(int value, TreeNode... children) {
            this.value = value;
            this.children = Sets.newHashSet(children);
        }
    }

创建了树 TreeNode 之后，如果我们想要并行地对树中的所有值求和，我们需要实现一个 RecursiveTask<Integer> 接口。每个任务都接收自己的节点，并将其值添加到其子节点的值之和上。

要计算子节点值的总和，任务实现执行以下操作

1. 将子节点集合转换为流 ( stream )
2. 映射前面操作中创建的流，为每个元素创建一个新的 CountingTask
3. 通过 fork 执行每个子任务
4. 通过在每个 fork 任务上调用 join() 方法来收集结果
5. 使用 ·Collectors.summingInt` 收集器对结果求和

    public static class CountingTask extends RecursiveTask<Integer> {
    
        private final TreeNode node;
    
        public CountingTask(TreeNode node) {
            this.node = node;
        }
    
        @Override
        protected Integer compute() {
            return node.value + node.children.stream()
              .map(childNode -> new CountingTask(childNode).fork())
              .collect(Collectors.summingInt(ForkJoinTask::join));
        }
    }

在树上运行计算的代码非常简单：

    TreeNode tree = new TreeNode(5,
      new TreeNode(3), new TreeNode(2,
        new TreeNode(2), new TreeNode(8)));
    
    ForkJoinPool forkJoinPool = ForkJoinPool.commonPool();
    int sum = forkJoinPool.invoke(new CountingTask(tree));