如何等待订阅完成?

时间:2019-06-03 07:02:42

标签: java project-reactor spring-webclient

我想使用Spring webclient并返回Mono来进行异步rest调用。我也正在并行执行一些数据库调用,但是由于某种原因,它不能被动地完成。

    Map<String, Object> models = new HashMap<>();

    Mono<User> users = this.webClient...;
    users.map(resp -> new UserState(userRequest, resp))
            .subscribe(response -> {
                models.put("userState", response);
            });
    Iterable<Product> messages = this.productRepository.findAll();
    models.put("products", messages);
    //Wait for users.subscribe to finish <<<<<<<<<<<<<HERE
    return new ModelAndView("messages/list", models);

在返回ModelAndView之前,我如何等待订阅完成。如果我使用Future可以随时get()进行操作,那将很容易。

1 个答案:

答案 0 :(得分:1)

您可以将阻塞调用包装在单独的调度程序上执行的Mono中,并用包含Mono数据的UserState压缩,并将其组合转换为Mono<ModelAndView>(可以从Spring控制器方法返回)。调用将并行执行,两个调用完成后将合并结果。

您可以为每个应用程序定义一个单独的有界调度程序,专门用于阻塞调用,并将其作为构造函数的参数提供给进行阻塞调用的任何类。

代码如下:

@Configuration 
class SchedulersConfig {

  @Bean
  Scheduler parallelScheduler(@Value("${blocking-thread-pool-size}") int threadsCount) {
    return Schedulers.parallel(threadsCount);
  }
}

@RestController
class Controller {

  final Scheduler parallelScheduler;

  ...

  Mono<User> userResponse = // webClient...

  Mono<Iterable<Product>> productsResponse = Mono.fromSupplier(productRepository::findAll)
    .subscribeOn(parallelScheduler); 

  return Mono.zip(userResponse, productsResponse, (user, products) -> 
    new ModelAndView("messages/list", 
      ImmutableMap.of(
        "userState", new UserState(userRequest, user),
        "products", products
      ))
  );
}

根据评论进行更新:
如果您只需要异步执行HTTP调用,然后将其与数据库结果结合起来,则可以执行以下操作

Map<String, Object> models = new HashMap<>();
Mono<User> userMono = webClient...;
CompletableFuture<User> userFuture = userMono.toFuture();
Iterable<Product> messages = productRepository.findAll();
User user = userFuture.join();
models.put("products", messages);
models.put("userState", new UserState(userRequest, user));
return new ModelAndView("messages/list", models);