CountedCompleter是ForkJoinTask的抽象子类，通过维护待完成计数器控制任务完成时机，适用于需协调多个子任务的场景。其核心在于pending count机制：调用addToPendingCount增加计数，tryComplete减少计数并传播完成状态，当计数归零时触发onCompletion方法执行汇总操作。示例中并行求和任务在拆分时增加计数，子任务完成后自动递减，最终在onCompletion中合并结果。该机制灵活支持树形处理、图遍历等复杂依赖结构，但需确保计数管理正确，避免提前完成或死锁。
在Java中，CountedCompleter 是 ForkJoinPool 提供的一个特殊类型的 ForkJoinTask，它允许你自定义任务的完成条件，而不仅仅依赖于任务自身的执行结束。这在需要多个子任务协同完成、或某个任务需等待其他任务完成后才触发回调时非常有用。
什么是CountedCompleter？
CountedCompleter 是 ForkJoinTask 的抽象子类，它引入了一个“待完成计数器”（pending count）。每当一个子任务启动或某个依赖任务被注册，你可以手动增加这个计数器。只有当计数器归零时，任务的 onCompletion 方法才会被调用，表示该任务真正完成。
与普通的 RecursiveAction 或 RecursiveTask 相比，CountedCompleter 更灵活，适用于树形结构处理、图遍历、流水线聚合等场景。
核心方法说明
你需要重点关注以下几个方法：
立即学习“Java免费学习笔记（深入）”；
compute()：主逻辑入口，通常在这里拆分任务或执行实际工作。onCompletion(CountedCompleter>)：当 pending count 变为0时调用，用于汇总结果或触发后续动作。tryComplete()：将当前任务的 pending count 减1，并向父任务传播完成状态。addToPendingCount(int)：增加待完成计数，常用于 fork 子任务前。
使用示例：计算数组和（带回调）
下面是一个使用 CountedCompleter 实现并行求和的例子，展示如何控制任务完成时机。
AppMall应用商店
AI应用商店，提供即时交付、按需付费的人工智能应用服务
56 查看详情
import java.util.concurrent.CountedCompleter;import java.util.concurrent.ForkJoinPool;public class SumCompleter extends CountedCompleter {private final int[] array;private final int lo, hi;private Long result;
public SumCompleter(CountedCompleter parent, int[] array, int lo, int hi) { super(parent); this.array = array; this.lo = lo; this.hi = hi;}public void compute() { if (hi - lo <= 1000) { // 小数据直接计算 long sum = 0; for (int i = lo; i < hi; i++) { sum += array[i]; } result = sum; tryComplete(); // 减少 pending count 并通知父任务 } else { int mid = (lo + hi) / 2; addToPendingCount(1); // 增加一个待完成子任务 new SumCompleter(this, array, lo, mid).fork(); new SumCompleter(this, array, mid, hi).fork(); tryComplete(); // 当前任务本身也算完成（结构已拆分） }}public void onCompletion(CountedCompleter caller) { // 所有子任务完成后执行 if (result == null) { result = 0L; // 遍历所有子任务结果 for (SumCompleter child = rawResultOf(getFirstChild()); child != null; child = rawResultOf(child.getNextSibling())) { result += child.result; } }}@SuppressWarnings("unchecked")private static T rawResultOf(CountedCompleter c) { return c != null ? (T)c.getRawResult() : null;}public static void main(String[] args) { int[] data = new int[10000]; for (int i = 0; i < data.length; i++) data[i] = 1; SumCompleter task = new SumCompleter(null, data, 0, data.length); ForkJoinPool.commonPool().invoke(task); System.out.println("Sum: " + task.getResult());}}
在这个例子中：
每次 fork 子任务前调用 addToPendingCount(1)。子任务完成后会自动减少父任务的 pending count。tryComplete() 触发完成检查，若计数归零则执行 onCompletion。最终结果通过 getResult() 获取。
适用场景与注意事项
CountedCompleter 特别适合以下情况：
任务之间存在依赖关系，不能简单地 fork/join。需要在所有子任务完成后执行汇总或清理操作。构建异步回调链，类似 CompletableFuture 的组合行为，但在 ForkJoin 框架内。
需要注意：
必须正确管理 pending count，漏掉 addToPendingCount 会导致任务提前完成。避免在 compute 中无限递归，应设置合理的阈值。getRawResult() 返回的是 result 字段，需确保线程安全写入。
基本上就这些。掌握 CountedCompleter 能让你更精细地控制并行任务的生命周期，尤其在复杂依赖结构中优势明显。
以上就是Java中如何使用CountedCompleter处理ForkJoin任务的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/288085.html
微信扫一扫 
支付宝扫一扫 
