CompletableFuture异常链：任一子任务异常如何立即触发handle回调

CompletableFuture.allOf() 的默认行为是等待所有传入的子任务全部完成（包含正常完成、异常完成两种状态），才会将自身标记为完成状态。只要还有任意一个子Future未完成，哪怕已经有其他子Future异常结束，allOf返回的汇总Future也不会触发完成回调，这就是异常场景下handle方法不执行的核心原因。

要实现「任意一个Future异常完成时立刻触发回调，所有Future正常完成时也触发回调」的效果，可以在原有allOf逻辑基础上，给每个子Future单独绑定异常监听，只要监听到任意一个子Future异常，就主动完成汇总Future，无需等待剩余任务。

可直接复用的工具方法 #

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionException;
import java.util.function.BiConsumer;

public class CompletableFutureUtils {
    public static CompletableFuture<Void> allOfOrFailFast(CompletableFuture<?>... futures) {
        // 保留原有allOf逻辑，处理所有任务正常完成的场景
        CompletableFuture<Void> allFuture = CompletableFuture.allOf(futures);
        
        // 给每个子Future加异常监听：任意一个异常，立刻让汇总Future异常完成
        BiConsumer<Object, Throwable> exceptionHandler = (res, ex) -> {
            if (ex != null) {
                // 解包CompletionException，抛出原始业务异常
                Throwable realEx = ex instanceof CompletionException ? ex.getCause() : ex;
                allFuture.completeExceptionally(realEx);
            }
        };
        for (CompletableFuture<?> future : futures) {
            future.whenComplete(exceptionHandler);
        }
        
        return allFuture;
    }
}

注：CompletableFuture的状态变更操作是幂等的，哪怕后续有多个任务先后触发异常、或是剩余任务后续全部正常完成，已经进入完成状态的汇总Future不会重复触发回调，不存在线程安全问题。

改造后的业务代码 #

把原有嵌套调用allOf的逻辑替换成上述工具方法即可：

CompletableFuture<Void> a1 = new CompletableFuture<>();
CompletableFuture<Void> a2 = new CompletableFuture<>();
CompletableFuture<Void> a3 = new CompletableFuture<>();
CompletableFuture<Void> b = new CompletableFuture<>();

// 传入所有需要监听的Future，不需要手动嵌套allOf
CompletableFuture<Void> allWithFailFast = CompletableFutureUtils.allOfOrFailFast(a1, a2, a3, b);

allWithFailFast.handle((result, exception) -> {
    System.out.println("Test");
    // 可通过exception是否为null区分触发场景
    if (exception != null) {
        System.out.println("触发原因：存在任务异常，异常信息：" + exception.getMessage());
    } else {
        System.out.println("触发原因：所有任务正常完成");
    }
    return result;
});

效果验证 #

• 正常场景：按顺序执行a1.complete(null);a2.complete(null);a3.complete(null);b.complete(null);，所有任务完成后会打印Test和正常完成提示，和原有逻辑表现完全一致。
• 异常快速触发场景：执行a1.completeExceptionally(new IllegalStateException("任务a1执行失败"));，不需要等a2、a3、b完成，会立刻打印Test和异常触发提示，符合预期。

• 快速触发异常回调后，未完成的子任务不会被自动中断，会继续在后台执行，只是不再阻塞后续回调逻辑。如果需要终止未执行的任务，可以在异常分支里遍历所有Future调用cancel(true)方法。
• 不能只给子Future加异常回调就丢弃原有allOf逻辑，否则会出现所有任务正常完成时回调不触发的问题。

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