java - 在不暴露执行线程的情况下返回一个 CompletableFuture

Question

我在库中公开了一个方法，该方法返回一个 CompletableFuture。该方法的计算发生在单线程执行器上，这是我的瓶颈，因此我不希望任何后续工作发生在同一线程上。

如果我使用返回“supplyAsync”结果的简单方法，我会将我宝贵的线程暴露给调用者，调用者可能正在添加同步操作(例如通过 thenAccept)，这可能会在该线程上占用一些 CPU 时间。

重现如下:

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class CfPlayground {
    private ExecutorService preciousExecService = Executors.newFixedThreadPool(1);

    CfPlayground() {}

    private static void log(String msg) {
        System.out.println("[" + Thread.currentThread().getName() + "] " + msg);
    }

    CompletableFuture<String> asyncOp(String param) {
        return CompletableFuture.supplyAsync(() -> {
            log("In asyncOp");
            return "Hello " + param;
        }, preciousExecService);
    }

    void syncOp(String salutation) {
        log("In syncOp: " + salutation);
    }

    void run() {
        log("run");
        asyncOp("world").thenAccept(this::syncOp);
    }

    public static void main(String[] args) throws InterruptedException {
        CfPlayground compFuture = new CfPlayground();
        compFuture.run();
        Thread.sleep(500);
        compFuture.preciousExecService.shutdown();
    }
}

这确实打印了:

[main] run
[pool-1-thread-1] In asyncOp
[pool-1-thread-1] In syncOp: Hello world

我找到的一个解决方案是引入另一个执行器，并在返回 CompletableFuture 之前为该执行器添加一个空操作 thenApplyAsync

    CompletableFuture<String> asyncOp(String param) {
        return CompletableFuture.supplyAsync(() -> {
            log("In asyncOp");
            return "Hello " + param;
        }, preciousExecService).thenApplyAsync(s -> s, secondExecService);
    }

这行得通，但感觉不是很优雅 - 有没有更好的方法来做到这一点？

Answer 1

没有将您的完成与相关操作的执行分离的功能。当链接依赖 Action 的线程已经完成注册并且你的执行者的线程完成 future 时，如果没有其他执行者被指定，哪个其他线程应该执行依赖 Action ？

您将另一个 Action 与不同的执行者链接起来的方法似乎是您可以获得的最佳方法。然而，重要的是要注意，在异常完成的情况下，异常会在不评估传递给 thenApply 的函数的情况下传播。如果调用者链接了 whenComplete、handle 或 exceptionally 等操作，此异常传播可能再次导致线程暴露。

另一方面，您不需要指定辅助执行器，因为您可以使用不带执行器参数的async 方法来获取默认(常见的 Fork/Join)池。

因此，到目前为止，链接 .whenCompleteAsync((x,y) -> {}) 是解决您的问题的最佳方案。