select如何在springboot java的方法中使用特定的任务执行器？

Question

我有一个使用 CompletableFuture 处理来自 stream 异步数据的应用程序。 展示我的异步实现的演示如下：

    @Async
    @Transactional(readOnly = true)
    public void beginProcessing() {
        try(Stream<String> st = myJpa.getMyStream()) { // use spring jpa to get a stream of data from db 
            CompletableFuture.allOf(st.map(i -> CompletableFuture.supplyAsync(() ->
                            myMethod(i)))
                    .toArray(CompletableFuture[]::new)).join();
        }
    }

    @Async
    private CompletableFuture<Void> myMethod(String i) {
       // logic goes here
    }

它工作正常。 然而，目前 CompletableFuture 使用一些默认的线程池来完成它的工作。 我想改用自定义定义的 taskExecutor。 这可以通过提供 taskExecutor 的名称作为supplyAsync(...)方法的第二个参数来实现。 我以前用ForkJoin线程池做过，所以我确信它能工作。

现在，我想让它与 taskExecutor 一起工作，所以我在我的配置 class 中添加了一个新 bean，如下所示（以及我将在其他地方使用的另一个 bean）：

@Configuration
public class MyConfigClass {

    @Bean
    public TaskExecutor myNewExecutor() {
        RejectedExecutionHandler re = new ThreadPoolExecutor.CallerRunsPolicy();
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(10);
        executor.setMaxPoolSize(100);
        executor.setQueueCapacity(0);
        executor.setThreadNamePrefix("myNewExecutor-");
        executor.setRejectedExecutionHandler(re);
        return executor;
    }

    @Bean
    public TaskExecutor someOtherExecutor() { // would be used elsewhere
        RejectedExecutionHandler re = new ThreadPoolExecutor.CallerRunsPolicy();
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(10);
        executor.setMaxPoolSize(100); 
        executor.setQueueCapacity(0);
        executor.setThreadNamePrefix("someOtherExecutor-");
        executor.setRejectedExecutionHandler(re);
        return executor;
    }
  
}

然后我将 TaskExecutor 自动连接到我的 class 并添加了第二个参数。 所以代码如下所示：

    @Autowired
    private TaskExecutor myNewExecutor;

    @Async
    @Transactional(readOnly = true)
    public void beginProcessing() {
        try(Stream<String> st = myJpa.getMyStream()) { // use spring jpa to get a stream of data from db 
            CompletableFuture.allOf(st.map(i -> CompletableFuture.supplyAsync(() ->
                            myMethod(i),myNewExecutor))
                    .toArray(CompletableFuture[]::new)).join();
        }
    }

    @Async
    private CompletableFuture<Void> myMethod(String i) {
       // logic goes here
    }

但是，这不起作用，因为我们有多个 TaskExecutor 类型的 bean，所以我得到以下异常

Caused by: org.springframework.beans.factory.NoUniqueBeanDefinitionException: 
No qualifying bean of type 'org.springframework.core.task.TaskExecutor' available: expected single matching bean but found 3: myNewExecutor, someOtherExecutor, taskScheduler

好的，所以我想我可以通过使用@Qualifier注释来解决它。 但是，它对我不起作用 - 仍然抛出异常（我在配置 class 中的@Bean注释下使用了所述注释，然后在逻辑所在的 class 中的@Autowired注释下使用了注释。）。

我知道您可以通过为每个 bean 命名并将相应的名称作为 arg 传递给@Async注释来使方法使用自定义线程池。 所以，我想我可以将我的代码重新编写为一个简单for循环，它调用mytMethod(...)并将方法的异步注释替换为@Async("myNewExecutor") 。 这可能行得通，但如果可以的话，我想保留 CompletableFuture，所以想知道是否有人能确定我错过了什么导致上述错误？

我觉得我缺少一些微不足道的东西来让它工作，但我就是看不到它。

Answer 1

所以……我想通了。 这确实是我不知何故错过的一件微不足道的事情（令人尴尬）。

通过指定在我的逻辑中使用哪个线程池，我确保有效地告诉@Async上面的myMethod(...)使用 3 个可用 bean 中的哪一个。 但是我忘记了我在beginProcessing()方法之上还有一个@Async 。 所以很自然地，它对这些更改感到困惑，并抛出了我提到的异常。 通过指定它需要使用哪个线程池（使用我提到的我知道的 bean 名称技巧），代码就像一个魅力。