Webclient 错误处理 - Spring Webflux

Question

我想在以下条件下抛出我的自定义异常：

1. 如果我得到 json 格式的正确错误响应，我想反序列化它并在 onStatus()

   中抛出名为 CommonException 的异常

2. 如果我收到 HTML 内容作为响应的一部分，或者反序列化没有成功发生，那么我想抛出我在 onErrorMap()

   中创建的 GenericException >

3. 在抛出 GenericException 时，我想将相同的 HttpStatus 代码传递给从下游响应中获得的上游。

    IdVerificationResponse idVerificationResponse = client.get()
    .uri(idVerificationUrl)
    .headers(headers -> headers.addAll(httpEntity.getHeaders()))
    .retrieve()
    .onStatus(HttpStatus::isError, response ->
        //Throw this one only if deserialization of error response to IdVerificationErrorResponse class happens successfully
   
        response.bodyToMono(IdVerificationErrorResponse.class)
            .flatMap(error -> Mono.error(CommonException.builder().message(error.getCustomMessage()).build()))
    )
    .bodyToMono(IdVerificationResponse.class)
    .onErrorMap(error -> {
        //Come over here only if there is an error in deserialization in onStatus()
   
        //How to get the HttpStatus we are getting as part of error response from the downstream
        HttpStatus httpStatus = HttpStatus.BAD_REQUEST;
        ApiErrorDetails errorDetailsObj = ApiErrorDetails.builder().errorCode(httpStatus.name()).errorDescription("Error related to HTML")
              .errorDetails("Error related to HTML").build();
        ErrorDetails errorDetails = ErrorDetails.builder().errors(errorDetailsObj).build();
        return GenericException.builder().errorDetails(errorDetails).httpStatus(httpStatus).build();
    }).block();
   

目前 onErrorMap() 每次都会被调用并覆盖我在 onStatus() 中抛出的异常

Answer 1

找到解决办法

IdVerificationResponse idVerificationResponse = client.get()
    .uri(processCheckUrl)
    .headers(headers -> headers.addAll(httpEntity.getHeaders()))
    .retrieve()
    .onStatus(HttpStatus::isError, response -> {
            HttpStatus errorCode = response.statusCode();
            return response.bodyToMono(IdVerificationErrorResponse.class)
                .onErrorMap(error -> new Exception("Throw your generic exception over here if there is any error in deserialization"))
                .flatMap(error -> Mono.error(new Exception("Throw your custom exception over here after successful deserialization")));
        })
    .bodyToMono(IdVerificationResponse.class).block();