使用Spring WebClient在错误情况下获取响应主体

Question

我试图用一个网络客户端来替换现有的客户机代码。因此，大多数调用都需要阻塞，这样应用程序的主要部分就不需要更改。当谈到错误处理时，这会带来一些问题。必须涵盖以下几种情况：

• 在成功的情况下，响应包含A类型的JSON对象
• 在错误情况下（HTTP状态4xx或5xx）响应可能包含B类型的JSON对象
• 在某些具有响应状态404的请求上，我需要返回一个与成功响应类型匹配的空List

为了产生正确的错误（异常），需要考虑错误响应。到目前为止，我还无法掌握错误主体。

我正在使用这个RestController方法来产生错误响应：

@GetMapping("/error/404")
@ResponseStatus(HttpStatus.NOT_FOUND)
public ResponseEntity error404() {
    return ResponseEntity
            .status(HttpStatus.NOT_FOUND)
            .body(new ErrorResponse());
}

使用此响应对象：

public class ErrorResponse {    
    private String message = "Error message";    
    public String getMessage() {
        return message;
    }
}

网络客户端定义如下：

WebClient.builder()
    .baseUrl("http://localhost:8081")
    .clientConnector(connector)
    .defaultHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE)
    .build();

连接器类型为CloseableHttpAsyncClient（Apache Http客户端5）。

从我的测试应用程序中，我这样调用：

public String get(int httpStatus) {
    try {
        return webClient.get()
                .uri("/error/" + httpStatus)
                .retrieve()
                .onStatus(HttpStatus::isError, clientResponse -> {
                    clientResponse.bodyToMono(String.class).flatMap(responseBody -> {
                        log.error("Body from within flatMap within onStatus: {}", responseBody);
                        return Mono.just(responseBody);
                    });
                    return Mono.error(new RuntimeException("Resolved!"));
                })
                .bodyToMono(String.class)
                .flatMap(clientResponse -> {
                    log.warn("Body from within flatMap: {}", clientResponse);
                    return Mono.just(clientResponse);
                })
                .block();
    } catch (Exception ex) {
        log.error("Caught Error: ", ex);
        return ex.getMessage();
    }
}

我得到的是来自onstatus返回的RuntimeException，当然还有最终捕获的异常。我缺少来自onstatus内的body ToMono的处理。我的怀疑是，由于阻塞性质，这没有执行，因为响应体是在onstatus之后处理body ToMono的。在注释掉onstatus时，我希望我们将警告记录在平图中，这也不会发生。

最后，我想将错误处理定义为一个过滤器，这样代码就不需要在每次调用时重复，但我需要获得错误响应主体，这样可以用正确的数据填充异常。

如何在同步WebClient调用中检索错误响应？

这个问题类似于Spring Webflux:Webclient:Get body on error，它没有公认的答案，并且一些建议的方法使用的方法没有被弃用。




Answer 1

以下是一种处理错误响应的方法：

• 使用onStatus捕获错误http状态

webClient.get()
        .uri("/error/" + httpStatus)
        .retrieve()
        .onStatus(HttpStatus::isError, clientResponse ->
                clientResponse.bodyToMono(ErrorResponse.class)
                        .flatMap(error ->
                                Mono.error(new RuntimeException(error.getMessage()))
                        )
        )
        .bodyToMono(Response.class)

你真的不需要试试catch。如果block，如果是非错误响应，上述代码将返回Response，如果是错误响应，则抛出带有自定义消息的异常。

更新

下面是使用WireMock的完整测试

class WebClientErrorHandlingTest {
    private WireMockServer wireMockServer;

    @BeforeEach
    void init() {
        wireMockServer = new WireMockServer(wireMockConfig().dynamicPort());
        wireMockServer.start();
        WireMock.configureFor(wireMockServer.port());
    }

    @Test
    void test() {
        stubFor(post("/test")
                .willReturn(aResponse()
                        .withHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE)
                        .withStatus(400)
                        .withBody("{\"message\":\"Request error\",\"errorCode\":\"10000\"}")
                )
        );

        WebClient webClient = WebClient.create("http://localhost:" + wireMockServer.port());

        Mono<Response> request = webClient.post()
                .uri("/test")
                .retrieve()
                .onStatus(HttpStatus::isError, clientResponse ->
                        clientResponse.bodyToMono(ErrorResponse.class)
                                .flatMap(error ->
                                        Mono.error(new RuntimeException(error.getMessage() + ": " + error.getErrorCode()))
                                )
                )
                .bodyToMono(Response.class);


        RuntimeException ex = assertThrows(RuntimeException.class, () -> request.block());
        assertEquals("Request error: 10000", ex.getMessage());
    }

    @Data
    private static class ErrorResponse {
        private String message;
        private int errorCode;
    }

    @Data
    private static class Response {
        private String result;
    }
}