如何使用@控件建议来处理来自反应式堆栈的Web客户端错误（Web通量->Spring）

Question

我使用weblux中的webClient向远程服务器发送请求。此时，我可以得到错误400。我需要拦截它并将其发送到客户端。

webClient
                        .post()
                        .uri(
                            
                        )
                        .contentType(MediaType.APPLICATION_FORM_URLENCODED)
                        .body(
                            BodyInserters
                                .fromFormData()
                                .with()
                                .with()
                
                        )
                        .retrieve()
                        .onStatus(
                            HttpStatus::isError, response -> response.bodyToMono(String.class) // error body as String or other class
                                .flatMap(error -> Mono.error(new WrongCredentialsException(error)))
                                )
                        .bodyToMono(TResponse.class)
                        .doOnNext(...);

• 错误

@ControllerAdvice
@Slf4j
public class ApplicationErrorHandler {

  @ExceptionHandler(WrongCredentialsException.class)
    public ResponseEntity<ErrorResponse> handleResponseException(WrongCredentialsException ex) {
      //  log.error("Error from WebClient - Status {}, Body {}", ex.getRawStatusCode(), ex.getResponseBodyAsString(), ex);

      ErrorResponse error = new ErrorResponse();
      return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
          .body(error);
    }

@Data
@NoArgsConstructor
@AllArgsConstructor
public class ErrorResponse {

    private String errorCode;

    private String message;
}

• rest api

@后映射

  public ResponseEntity<String> send(@RequestBody Dto dto) {

        log.debug("An notification has been send to user");

        return new ResponseEntity<>(HttpStatus.OK);
    }

我尝试了这里的选项，但没有成功。有人能解释它是如何工作的，以及如何为我的情况配置它吗？

Answer 1

• 第一个案例

return Objects.requireNonNull(oauthWebClient
                .post()
                .uri(uri)
                .bodyValue(dto)
                .attributes(oauth2AuthorizedClient(authorizedClient))
                .header(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE)
                .exchangeToMono(response -> {

                    HttpStatus httpStatus = response.statusCode();

                    if (httpStatus.is4xxClientError()) {
                        getErrFromClient(response, httpStatus);
                    }

                    if (httpStatus.is5xxServerError()) {
                        getErrFromServer(response, httpStatus);
                    }

                    return Mono.just(ResponseEntity.status(response.statusCode()));
                })
                .block())
                .build();
    }

    private void getErrFromServer(DtoResponse response, HttpStatus httpStatus) {

        String err = response.bodyToMono(String.class).toString();
        log.error("HttpStatus: {}, message: {}", httpStatus, err);

        HttpHeaders httpHeaders = response.headers().asHttpHeaders();

        List<String> errorBody = httpHeaders.get("errBody");

        assert errBody != null;
        throw new CustomException(
                "{ HttpStatus : " + httpStatus + " , message : " + errBody + " }");
    }

    private void getErrFromClient(DtoResponse response, HttpStatus httpStatus) {

        String err = response.bodyToMono(String.class).toString();

        log.error("HttpStatus: {}, err: {}", httpStatus, err);

        HttpHeaders httpHeaders = response.headers().asHttpHeaders();
        List<String> errorBody = httpHeaders.get("errBody");

        assert errBody != null;
        throw new CustomException(
                "{ HttpStatus : " + httpStatus + " , message : " + errBody + " }");
    }

• 然后

@ControllerAdvice
public class HandlerAdviceException {

    @ExceptionHandler(CustomException.class)
    public ResponseEntity<ErrorResponse> handleCustomException(CustomException e) {

//here your code
//for example:

        String errMessage = e.getLocalizedMessage();

        return ResponseEntity
                .internalServerError()
                .body(new ErrorResponse(ErrorCode.INTERNAL_ERROR, errMessage));
    }

}

• 第二种情况

          return webClient
            .post()
            .uri(
              properties......,
              Map.of("your-key", properties.get...())
            )
            .contentType(MediaType.APPLICATION_FORM_URLENCODED)
            .body(
              prepare....()
            )
            .retrieve()
            .bodyToMono(TokenResponse.class)
            .doOnSuccess(currentToken::set);
        }

在这里，如果成功，您将获得所需的结果，但如果发生错误，则只需要在建议控制器中为WebClientResponseException配置侦听器。

@ControllerAdvice
@Slf4j
public class CommonRestExceptionHandler extends ResponseEntityExceptionHandler {

  @ExceptionHandler(WebClientResponseException.class)
  protected ResponseEntity<ApiErrorResponse> handleWebClientResponseException(WebClientResponseException ex) {

    log.error(ex.getClass().getCanonicalName());

    String errMessageAdditional = .....

    final ApiErrorResponse apiError = ApiErrorResponse.builder()
      .message(ex.getLocalizedMessage())
      .status(HttpStatus.UNAUTHORIZED)
      .build();
//if it needs
    HttpHeaders httpHeaders = new HttpHeaders();
    httpHeaders.add(.......);

    return new ResponseEntity<>(apiError, httpHeaders,  apiError.getStatus());
  }
}