如何在CompletionStage的Lambda中执行异步任务并确保其完成后返回？

我们有一个内部API internalAPI，它会执行传入Lambda中的代码，并返回CompletionStage<Boolean>类型结果。初始实现如下：

private CompletionStage<Boolean> foo1() {
    return internalAPI(param -> {
        // 第一步操作
        // 第二步操作
        return Boolean.TRUE;
    });
}

现在第二步操作被实现为异步方法secondStep()，返回CompletionStage<Boolean>类型。在foo2()的Lambda中调用它时，无法确保它在执行return Boolean.TRUE;前完成：

private CompletionStage<Boolean> foo2() {
    return internalAPI(param -> {
        // 第一步操作
        secondStep();  // <-- 如何执行并获取它的结果？
        // 若第二步执行成功则...
        return Boolean.TRUE;
    });
}

private CompletionStage<Boolean> secondStep() {
    // 执行异步操作
}

补充说明（编辑后） #

以下是internalAPI()的简要实现，它会在单个事务中执行所有步骤，若任一步骤失败则事务可回滚：

public <A> CompletableFuture<A> internalAPI(Function<Connection, A> block) {
    return CompletableFuture.supplyAsync(() -> doTransaction(block),
            new HttpExecutionContext(databaseContext).current());
}

public <T> T doTransaction(Function<Connection, T> block) {
    Transaction tx = Ebean.beginTransaction();
    try (Connection sql2oConnection = sql.open()) {
        spliceConnections(sql2oConnection, tx.getConnection());
        T ret = block.apply(sql2oConnection);
        Ebean.commitTransaction();
        return ret;
    } finally {
        Ebean.endTransaction();
    }
}

核心问题分析 #

internalAPI的事务逻辑是同步执行传入的block代码块：执行完block.apply()后立即提交事务。如果在block中调用异步的secondStep()，会导致事务在secondStep完成前就提交，既无法等待异步结果，也会破坏事务的原子性（异步操作失败时无法回滚已提交的事务）。

方案1：将secondStep改为同步方法 #

这是最稳妥的方案，能保证整个流程在同一个事务中同步执行：

private CompletionStage<Boolean> foo2() {
    return internalAPI(param -> {
        // 第一步操作
        boolean secondResult = secondStep(); // 改为同步调用，获取结果
        if (secondResult) {
            // 第二步执行成功后的逻辑
        }
        return Boolean.TRUE;
    });
}

// 修改为同步方法
private boolean secondStep() {
    // 执行原异步操作的同步版本
}

方案2：同步等待异步secondStep完成（仅当无法修改secondStep时使用） #

如果必须保留secondStep的异步特性，需要在block中强制等待异步任务完成。但需注意：异步任务必须绑定到当前事务上下文，否则其操作不会纳入事务管理，失败时无法回滚。

private CompletionStage<Boolean> foo2() {
    return internalAPI(param -> {
        // 第一步操作
        // 同步等待异步任务完成，获取结果
        boolean secondResult = secondStep()
                .toCompletableFuture()
                .join(); // 阻塞当前线程，直到任务完成
        
        if (secondResult) {
            // 第二步执行成功后的逻辑
        }
        return Boolean.TRUE;
    });
}

注意：使用join()会阻塞当前事务线程，可能影响系统性能；若异步任务未绑定到当前事务上下文，其操作不受事务控制，风险较高。

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