CompletableFuture——聚合未来以快速失败

CompletableFuture -- Aggregate Future to Fail Fast

我一直在使用 CompletableFuture.allOf(...) 助手来创建聚合期货,只有当它们的复合期货被标记为完成时才会成为 "done",即:

CompletableFuture<?> future1 = new CompletableFuture<>();
CompletableFuture<?> future2 = new CompletableFuture<>();
CompletableFuture<?> future3 = new CompletableFuture<>();

CompletableFuture<?> future = CompletableFuture.allOf(future1, future2, future3);

我希望此功能略有变化,其中总期货在市场上完成时:

在后一种情况下,聚合 future 应该(例外地)立即完成,而不必等待其他 futures 完成,即 fail-fast.

为了与 CompletableFuture.allOf(...) 对比说明这一点,请考虑以下内容:

// First future completed, gotta wait for the rest of them...
future1.complete(null);
System.out.println("Future1 Complete, aggregate status: " + future.isDone());

// Second feature was erroneous! I'd like the aggregate to now be completed with failure
future2.completeExceptionally(new Exception());
System.out.println("Future2 Complete, aggregate status: " + future.isDone());

// Finally complete the third future, that will mark the aggregate as done
future3.complete(null);
System.out.println("Future3 Complete, aggregate status: " + future.isDone());

使用 allOf(...),此代码产生:

Future1 Complete, aggregate status: false
Future2 Complete, aggregate status: false
Future3 Complete, aggregate status: true

而我的替代聚合实现将 return "true" 在 Feature2 完成后,考虑到它是一个例外。


我在 Java 标准库中找不到任何可以帮助我实现此目的的实用程序,这感觉很奇怪......因为它是一个相对普通的用例。

看看 CompletableFuture.allOf(...) 的实现,很明显这些场景背后的逻辑相当复杂。我不愿意自己写这个,我想知道是否有其他选择?

虽然在语法上不如 CompletableFuture.allOf(...) 方法好,但 thenCompose(...) 似乎可以提供解决方案:

CompletableFuture<?> future = future1.thenCompose((f) -> future2).thenCompose((f) -> future3);

这将产生所需的:

Future1 Complete, aggregate status: false
Future2 Complete, aggregate status: true
Future3 Complete, aggregate status: true

这可以包含在一个辅助方法中,该方法会为调用者提供一些语法上的细节:

private static CompletableFuture<?> composed(CompletableFuture<?> ... futures) {

    // Complete when ALL the underlying futures are completed
    CompletableFuture<?> allComplete = CompletableFuture.allOf(futures);

    // Complete when ANY of the underlying futures are exceptional
    CompletableFuture<?> anyException = new CompletableFuture<>();
    for (CompletableFuture<?> completableFuture : futures) {
        completableFuture.exceptionally((t) -> {
            anyException.completeExceptionally(t);
            return null;
        });
    }

    // Complete when either of the above are satisfied
    return CompletableFuture.anyOf(allComplete, anyException);
}

允许:

CompletableFuture<?> future = composed(future1, future2, future3);

您可以通过创建一个 allOf 和一个 anyOf,然后将它们组合成第二个 anyOf 来做到这一点。