如何让CompletableFuture的thenApply抛出的异常传递至上层？

你的代码存在两个核心问题导致异常无法传递到上层，以及潜在的空指针风险，以下是修复后的代码和说明：

核心问题分析 #

1. 异步任务未串联：computeResult中调用doSomething()后，没有将其返回的CompletionStage与主流程串联，而是直接返回null，导致doSomething的异常完全被孤立，上层无法感知。
2. 非法返回值：doSomething的else分支返回null，会导致调用方在调用thenApply时抛出NullPointerException。

修复后的代码 #

核心业务函数 #

public CompletionStage<Result> computeResult() {
    return CompletableFuture.allOf(a, b)
        // 使用thenCompose串联返回CompletionStage的异步操作
        .thenCompose(ignore -> {
            final var aValue = a.join();
            final var bValue = b.join();

            // 将doSomething的结果与buildResult串联，纳入主流程
            return doSomething()
                .thenApply(ignoreVoid -> buildResult());
        });
}

执行校验的私有函数 #

private CompletionStage<Void> doSomething() {
    if (someCondition) {
        return anotherService.shouldPerformTask()
            .thenApply(shouldPerform -> {
                // Do something
                if (someCheck) {
                    throw new RuntimeException("message");
                }
                return null;
            });
    }
    // 返回一个正常完成的空结果，避免空指针
    return CompletableFuture.completedFuture(null);
}

修复说明 #

• 使用thenCompose替代thenApply：thenCompose专门用于处理返回CompletionStage的异步操作，能将子任务的结果和异常无缝传递到主流程中，而thenApply仅适用于同步转换。
• 避免返回null：用CompletableFuture.completedFuture(null)返回一个已完成的空CompletionStage，确保后续的thenApply调用不会触发空指针。
• 异常传递逻辑：当doSomething的thenApply中抛出RuntimeException时，该异常会被自动包装为CompletionException，上层调用join()时会抛出这个异常，完全符合你的单元测试断言预期。

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