Java 结合两个期货作为主备

发布于08月25日

我有两个进程应该并行启动，如果第一个进程失败，我应该使用第二个进程结果.我想出了以下代码.有没有什么明智的方法来简化它？

final CompletableFuture<Data> mainFut = mainProcess(...);
final CompletableFuture<Data> backupFut = backupProcess(...);
final CompletableFuture<Data> resultFut = new CompletableFuture<>();
mainFut.handle((r, t) -> {
    if (t == null) {
        resultFut.complete(r); // main process is fine, use its result
    } else {
        LOGGER.error("Main process failed, using backup", t);
        backupFut.handle((rb, tb) -> {
            if (tb == null) {
                resultFut.complete(rb); //backup is fine, use its result
            } else {
                resultFut.completeExceptionally(tb); //backup failed too
            }
            return null;
        });
    }
    return null;
});
return resultFut;

推荐答案

您没有使用handle的结果，这将非常适合您的用例，例如

final CompletableFuture<Data> mainFut = mainProcess(...);
final CompletableFuture<Data> backupFut = backupProcess(...);

return mainFut.handle((r,t) -> t == null? r: backupFut.join());

如果您希望避免潜在的阻止join()呼叫，您可以使用

final CompletableFuture<Data> mainFut = mainProcess(...);
final CompletableFuture<Data> backupFut = backupProcess(...);

return mainFut.handle((r1, t1) -> t1 == null? mainFut: backupFut)
    .thenCompose(Function.identity());

备份操作的结果将取代主操作的异常，但当两者都失败时，您可以将原始异常包括在异常结果中，如下所示

return mainFut.handle((r1, t1) -> t1 == null? mainFut:
        backupFut.whenComplete((r2, t2) -> { if(t2 != null) t2.addSuppressed(t1); }))
    .thenCompose(Function.identity());