基于输入将WebClient并行调用的批处理结果转换为HashMap条目，而不阻止每个单独的调用

Question

我想将Mono从WebClient响应转换/添加到以输入为键的映射中

我正在使用WebClient并行执行一批REST调用，但是我不想返回用户列表，而是想返回一个ID的HashMap作为键，从REST调用返回的用户作为值。

我不想在添加到HashMap之前阻止每个单独的调用来获取值。

有没有一种方法可以将结果从WebClient转换为HashMap条目，而不会影响REST调用的并行执行？

我尝试了Mono的doOnSuccesscallback，但不确定这样做是否正确。

当前执行情况

public List<<User> fetchUsers(List<Integer> ids) {
        List<Mono<User>> userMonos = new ArrayList();
        
        for (int id : ids) {
            userMonos.add(webClient.get()
                .uri("/otheruser/{id}", id)
                .retrieve()
                .bodyToMono(User.class));
        }

       List<User> list = Flux.merge(userMonos).collectList().block();
       return list; 
    }

因此，预期产出为：

HashMap<Integer, User>()

如果我不能恰当地表达预期的结果，我道歉。请随时让我知道，如果我需要添加更多的细节或添加更多的清晰的问题。

我真的很感激你的帮助。与此同时，我也在努力寻找解决办法。

Answer 1

您正在混合命令式代码和反应式代码。你必须选择一种方式，并坚持下去。

如果你想要的实际值，而不是Mono或Flux你必须块。把它想象成一个Future，在我们等待值出现之前，那里没有“值”。所以阻塞是ONLY方式。

如果我正确理解您的代码，我将执行以下操作。

public HashMap<Integer, User> fetchUsers(List<Integer> ids) {
    final Map<Integer, User> userMap = new HashMap();
    return Flux.fromIterable(ids)
                .flatMap(id -> webClient.get()
                    .uri("/otheruser/{id}", id)
                    .retrieve()
                    .bodyToMono(User.class)
                    .doOnSuccess(user -> {
                        userMap.put(id, user);
                    })
                .thenReturn(userMap)
                .block()
}

那么这段代码做什么呢？

它获取一个id的列表，并将其放入一个Flux中。通量将async同时启动所有请求，因为我们使用的是flatMap。当所有的请求完成后，我们将通过将值添加到hashmap来执行副作用。因为我们不关心返回类型，我们使用然后来忽略返回。我们告诉它返回hashmap。最后，我们调用block使代码实际运行，并等待所有的请求等完成并产生最终的hashmap。

我已经在手机上写了这篇文章，所以我不能检查编译器，但类似的东西应该可以让你开始。如果有人看到任何错误，请随意编辑。

Answer 2

如果可能，最好避免修改副作用操作符（如doOnSuccess）的外部状态。例如，在这种特殊情况下，如果外部Map不是线程安全的，则可能会导致并发问题。

作为更好的选择，您可以使用Reactor操作符收集到地图：

public Map<Integer, User> fetchUsers(List<Integer> ids) {
    return Flux.fromIterable(ids)
               .flatMap(id -> webClient.get()
                                       .uri("/otheruser/{id}", id)
                                       .retrieve()
                                       .bodyToMono(User.class)
                                       .map(user -> Tuples.of(id, user)))
               .collectMap(Tuple2::getT1, Tuple2::getT2)
               .block();
}

您可以创建一个小类来提高可读性，而不是Tuple。或者更好的是，如果用户知道它的ID，那么您可以完全省略Tuple，您可以执行类似的操作。collectMap（用户：：getId，用户-