如何将 Spring WebClient 响应转换为 ResponseEntity？

Question

我可以使用 toEntity() 方法返回 ResponseEntity，如下所示：

@GetMapping("/uri")
public Mono<ResponseEntity<Data[]>> methodName() {
    return webClient
            .get()
            .uri("http://localhost:8088/externalService")
            .accept(MediaType.APPLICATION_JSON)
            .retrieve()
            .toEntity(Data[].class);
}

但我想在返回之前处理响应标头。 上面的代码将 WebClient 响应转换为 ResponseEntity 并立即返回，但我想将其存储在 ResponseEntity 变量中，对其进行处理，然后将 ResponseEntity 返回。

我提到了这个 -> Spring WebClient Documentation

当我尝试将其存储在变量中时，我收到此错误 ->“block()/blockFirst()/blockLast() 正在阻塞，线程 reactor-http-nio-3 不支持”

Answer 1

您可以简单地使用 Reactor 的 map 运算符来修改标头：

return webClient
        .get()
        .uri("http://localhost:8088/externalService")
        .accept(MediaType.APPLICATION_JSON)
        .retrieve()
        .toEntity(Data[].class)
        .map(responseEntity ->  responseEntity.getHeaders().add("header", "header-value");

或者，您可以使用.handle 运算符来提供响应处理：

.handle((responseEntity, sink) -> {
        if(!isValid(responseEntity)){
            sink.error(new InvalidResponseException());
        } else if (isOk(responseEntity))
            sink.next(responseEntity);
        }
        else {
            //just ignore element
        }
    })