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 实现并行求和的例子，展示如何控制任务完成时机。
import java.util.concurrent.CountedCompleter;
import java.util.concurrent.ForkJoinPool;
<p>public class SumCompleter extends CountedCompleter<Long> {
private final int[] array;
private final int lo, hi;
private Long result;</p><pre class='brush:java;toolbar:false;'>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> 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 能让你更精细地控制并行任务的生命周期，尤其在复杂依赖结构中优势明显。
