Spring Boot @Async 场景下如何等待所有线程执行完成后执行后续流程

首先要明确：返回void的@Async方法无法追踪执行状态，你需要先调整异步方法的返回类型，再通过收集任务句柄的方式等待全部完成，以下是两种常用实现方案：

方案1：基于 Spring 原生Future实现（兼容性最好） #

• 调整标注@Async的方法返回值为Future<Void>（不需要返回执行结果的场景），示例代码：

@Async
public Future<Void> executeServerCmd(String serverIp, String cmd) {
    // 原有连接服务器、执行命令的业务逻辑
    return new AsyncResult<>(null);
}

• for循环调用异步方法时，收集所有返回的Future实例到列表：

List<Future<Void>> taskList = new ArrayList<>();
for (String ip : serverIps) {
    Future<Void> task = asyncService.executeServerCmd(ip, targetCmd);
    taskList.add(task);
}

• 遍历所有Future调用get()方法阻塞等待，全部返回后即可执行后续流程：

for (Future<Void> task : taskList) {
    try {
        // 可指定超时时间避免单节点故障阻塞全流程，比如 task.get(30, TimeUnit.SECONDS)
        task.get();
        // 可在此处处理单台服务器的执行成功逻辑
    } catch (InterruptedException | ExecutionException e) {
        // 处理单台服务器执行异常逻辑，比如记录错误日志、标记执行失败
        e.printStackTrace();
    }
}
// 此处所有异步任务已全部执行完成，直接写后续处理逻辑即可

方案2：基于CompletableFuture实现（更灵活，适合复杂任务编排场景） #

• 调整@Async方法返回值为CompletableFuture<Void>：

@Async
public CompletableFuture<Void> executeServerCmd(String serverIp, String cmd) {
    // 原有连接服务器、执行命令的业务逻辑
    return CompletableFuture.completedFuture(null);
}

• 收集所有CompletableFuture实例，调用allOf组合所有任务后等待完成：

List<CompletableFuture<Void>> taskList = new ArrayList<>();
for (String ip : serverIps) {
    CompletableFuture<Void> task = asyncService.executeServerCmd(ip, targetCmd);
    taskList.add(task);
}

CompletableFuture<Void> allTasks = CompletableFuture.allOf(taskList.toArray(new CompletableFuture[0]));
// 阻塞等待所有任务执行完成
allTasks.join();
// 后续处理逻辑

注意：如果使用自定义线程池执行@Async任务，需要确保线程池核心线程数足够覆盖并发连接的服务器数量，避免任务排队导致执行耗时过长。

内容的提问来源于stack exchange，提问作者WILSON CHACÓN