Java异步代码重构：将future.get()改为CompletionStage.whenComplete()是否合理？

你的重构版本存在的问题 #

1. 线程安全风险：ArrayList不是线程安全集合，whenComplete的回调会在异步线程中执行，多个线程同时调用add方法会引发ConcurrentModificationException，或者出现元素丢失、重复的情况。
2. 结果不完整：直接返回combinedStatus时，大部分异步任务可能还未执行完毕，返回的列表内容是空或不完整的。
3. 变量笔误：回调逻辑中使用了response.result()，但方法参数是resp，应该修正为resp.result()。

正确的重构方案与惯用方式 #

场景1：需要等待所有任务完成后返回完整结果 #

如果调用方需要拿到所有任务的执行结果再继续，推荐使用CompletableFuture.allOf来实现非阻塞式的等待与结果收集，避免手动回调的并发问题：

List<CompletableFuture<Object>> futures = new ArrayList<>();
for (List<Long> batch : batches) {
    CompletableFuture<Object> future = ask(batch);
    futures.add(future);
}

// 组合所有异步任务，等待全部完成后统一收集结果
CompletableFuture<List<Object>> combinedFuture = CompletableFuture.allOf(futures.toArray(new CompletableFuture[0]))
    .thenApply(v -> {
        List<Object> result = new ArrayList<>();
        for (CompletableFuture<Object> future : futures) {
            try {
                Object resp = future.get(); // allOf完成后，get不会阻塞
                result.add(resp.result());
            } catch (Exception e) {
                // 单独处理单个任务的异常，如记录日志
                e.printStackTrace();
            }
        }
        return result;
    });

// 若调用方允许异步处理，直接返回CompletableFuture<List<Object>>
// return combinedFuture;

// 若必须同步返回结果，此处的get因allOf已完成，不会长时间阻塞
// return combinedFuture.get();

场景2：完全异步处理每个任务的结果 #

如果不需要等待所有任务完成，而是每个任务结果就绪后就处理，需要使用线程安全集合存储结果：

// 使用线程安全的CopyOnWriteArrayList避免并发修改问题
List<Object> combinedStatus = new CopyOnWriteArrayList<>();

for (List<Long> batch : batches) {
    CompletableFuture<Object> future = ask(batch);
    future.whenComplete((resp, ex) -> {
        if (ex != null) {
            // 记录异常日志
            ex.printStackTrace();
        } else {
            combinedStatus.add(resp.result());
            // 可选：每个结果就绪后执行后续业务逻辑
        }
    });
}

// 注意：此时返回combinedStatus可能为空，需调用方通过事件、回调等方式监听结果完成

异步重构的通用原则 #

• 优先使用线程安全集合：异步回调中操作集合时，避免用ArrayList这类非线程安全实现，改用CopyOnWriteArrayList、ConcurrentLinkedQueue等。
• 利用CompletableFuture组合API：allOf/anyOf用于批量任务的等待控制，thenApply/thenCompose用于链式处理结果，减少手动回调的复杂度。
• 不要返回未完成的结果集合：如果调用方需要完整结果，返回CompletableFuture<List<Object>>而非直接返回列表，让调用方决定同步等待还是继续异步流程。
• 精细化异常处理：每个异步任务的异常单独处理，避免单个任务失败影响全局，同时不要吞掉异常，至少保留日志记录。

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