如何在Webflux WebClient中自定义异常？

Question

在我的webflux应用程序中，我想通过WebClient发送一些请求。我想处理所有条件(200、401、403和...response ),然后向客户端响应一个json。对于错误状态代码，我想使用@RestControllerAdvice，所以我必须抛出一个自定义异常，然后在控制器建议中处理自定义json。参见示例代码:

WebClient.create().post
   .uri("URI")
   .retrieve()
   .onStatus(HttpStatus::is4xxClientError, {
       // create Mono<CustomException> 
   }
   .bodyToMono(ResponseDto.class)

现在异常处理程序如下：

@ResponseBody
@ResponseStatus(...)
@ExceptionHandler(CustomException1.class)
public void customException1(CustomException1 exception) {
   //do Something width response body
}

@ExceptionHandler(CustomException2.class)
public void customException2(CustomException2 exception) {
   //do Something width response body
}

Webclient获得一个包含json主体的401响应，如下所示：

{
   "message": "Password is incorrect"
}

我可以创建＜code＞Mono。error（newCustomException（）），但问题是WebClient响应体。如果消息是“密码不正确”，我想向客户端发送：

用户名或密码不正确

我该怎么办？

Answer 1

我从未使用过WebFlux，但我发现了Spring Web客户端的添加异常处理程序。看起来你可以通过编写返回Mono.error来引发异常（new CUomException1（））；在onState处理程序函数中。根据描述，异常将从身体ToMono（类）方法引发。

最后，您可以像这样修改您的 ControllerAdvice：

@ExceptionHandler(CustomException.class)
public ResponseEntity<CustomErrorModel> customException(CustomException exception) {
   return new ResponseEntity<>( new CustomErrorModel(), HttpStatus.BAD_REQUEST );
}

正如我所说 - 我没有测试它，但它可能对你来说是一个提示。

Answer 2

请检查以下 API 实现，该实现处理调用的 rest api 调用的错误响应。我假设我们知道 API 的响应，因此我创建了 ResponseDTO 类。

@GetMapping("/webclient/test")
public ResponseDTO testWebclient(@RequestBody RequestDTO requestPayload) {
    String url = "http://localhost:8080/api/write/simple";
    return WebClient.create()
            .post()
            .uri(url)
            .accept(MediaType.APPLICATION_JSON)
            .body(BodyInserters.fromObject(requestPayload))
            .retrieve()
            .onStatus(HttpStatus::is4xxClientError, response -> {
                return response.bodyToMono(ResponseDTO.class).flatMap(error -> {
                    return Mono.error(new CustomRuntimeException(error.getMessage(), response.statusCode(), error.getErrorCode()));
                });
            })
            .onStatus(HttpStatus::is5xxServerError, response -> {
                return response.bodyToMono(ResponseDTO.class).flatMap(error -> {
                    return Mono.error(new CustomRuntimeException(error.getMessage(), response.statusCode()));
                });
            })
            .bodyToMono(ResponseDTO.class);
}

public class ResponseDTO {

    private String message;
    private HttpStatus status;
    private ErrorCode errorCode;
    
    public ResponseDTO(String message, HttpStatus status) {
        this.message = message;
        this.status = status;
    }

    public ResponseDTO(String message, HttpStatus status, ErrorCode errorCode) {
        this.message = message;
        this.status = status;
        this.errorCode = errorCode;
    }
    /** getters **/
}

    public class ErrorCode {
        private String numCode;
        private String txtCode;
        
        /** getters **/
    }

要捕获异常，请在< code > @ RestControllerAdvice 的类中添加以下代码片段(希望您已经知道了)。

@ExceptionHandler(value = CustomRuntimeException.class)
public final ResponseEntity<ApiResponse> handleCustomRuntimeException(CustomRuntimeException exception) {
    List<ApiSubError> subErrors = Arrays.asList(new ApiSubError(exception.getMessage()));
    ApiResponse response = new ApiErrorResponse(exception.getMessage(), exception.getErrorCode());
    LOG.error("result: {}", response.getResult());
    return new ResponseEntity<>(response, exception.getHttpStatus());
}