Java Spring中@Async异步树任务执行完成前主线程等待方案咨询

方案1：CountDownLatch固定计数实现 #

提前统计整棵树的所有节点总数，初始化相同计数的CountDownLatch实例，在每个节点任务的执行逻辑末尾（包括异常分支）调用countDown()方法，主线程直接调用await()方法阻塞等待即可。

• 适用场景：树结构固定，执行过程中不会动态新增节点
• 代码示例：

// 预先统计整棵树总节点数
int totalNodes = calcAllNodeCount(root);
CountDownLatch taskLatch = new CountDownLatch(totalNodes);

// 启动根节点异步任务，每个节点执行完成后必须调用taskLatch.countDown()
asyncExecuteRoot(root, taskLatch);

// 主线程阻塞等待，可设置超时避免永久挂起
taskLatch.await(60, TimeUnit.MINUTES);

// 所有任务完成后继续执行主流程

方案2：嵌套CompletableFuture链式组装（推荐） #

不需要提前统计节点数，每个节点的异步方法返回CompletableFuture，当前节点执行完成后触发所有子节点执行，将所有子节点的CompletableFuture用allOf合并后作为当前节点方法的返回值。最终根节点返回的CompletableFuture会在整棵树所有节点全部完成后才进入完成状态，主线程直接调用join()等待即可。

• 优势：完全符合无感知调用要求，不需要额外维护全局计数，主流程保持在主线程执行，天然支持动态生成子节点的场景
• 代码示例：

@Async
public CompletableFuture<Void> executeTaskNode(Node node) {
    try {
        // 执行当前节点的业务逻辑
        runNodeTask(node);
    } catch (Exception e) {
        // 按需求处理异常，不抛出即可避免中断分支
        log.error("节点执行失败", e);
    }
    
    // 触发所有子节点执行，收集子节点的Future
    List<CompletableFuture<Void>> childFutureList = node.getChildren()
            .stream()
            .map(this::executeTaskNode)
            .toList();
    
    // 返回合并后的Future，所有子节点完成后当前节点的Future才会完成
    return CompletableFuture.allOf(childFutureList.toArray(new CompletableFuture[0]));
}

// 主线程调用逻辑
executeTaskNode(rootNode).join();
// 此处开始执行所有任务完成后的主流程

方案3：Phaser动态计数实现 #

如果树节点会在执行过程中动态新增，可使用Phaser替代CountDownLatch，每个节点启动前注册一个相位，执行完成后抵达相位，主线程等待相位归零即可。

内容的提问来源于stack exchange，提问作者Sjoerd Schepers