Spring Boot 从异常处理程序返回 401 状态的自定义对象

Question

我有一个 Spring Boot rest API，它有一个 LoginController 用于执行简单的身份验证过程。 这是令牌验证操作。

@PostMapping("validatetoken")
public ResponseEntity<ValidateTokenResponse> validateToken(
        @RequestBody ValidateTokenRequest validateTokenRequest, HttpServletRequest request) throws Exception {

    if (StringUtils.isEmpty(validateTokenRequest.getToken())) throw new AuthenticationFailedException("token parameter cannot be null or empty");

    Boolean isValid = authenticationService.validateToken(request, validateTokenRequest.getToken());
    ValidateTokenResponse response = new ValidateTokenResponse();
    response.setIsValid(isValid);
    response.setToken(validateTokenRequest.getToken());
    return new ResponseEntity<>(response, isValid? HttpStatus.OK : HttpStatus.UNAUTHORIZED);
}

在我的 api 中，我在ResponseEntityExceptionHandler捕获所有错误并以这种方式转换自定义对象。

@ExceptionHandler(AuthenticationFailedException.class)
@ResponseBody
protected ResponseEntity<Object> handleAuthenticationFailed(AuthenticationFailedException ex) {
    LogManager.error(ex);
    ApiError apiError = new ApiError(HttpStatus.UNAUTHORIZED);
    apiError.setMessage(ex.getMessage());
    return buildResponseEntity(apiError);
}

但是，当我想使用如下所示的RestTemplate调用此 api 时，我收到了类似java.net.HttpRetryException: cannot retry due to server authentication, in streaming mode 的异常。

ResponseEntity<String> responseEntity =
    restTemplate.exchange(
        this.validateTokenUrl,
        HttpMethod.POST,
        requestHttpEntity,
        String.class);

但是如果我将HttpStatus从ExceptionHandler更改为 except HttpStatus.UNAUTHORIZED我可以从客户端获取真正的ApiError对象。 什么可能导致此问题，我该如何解决？

编辑：创建了一个github repo来模仿我项目中的问题部分。

Answer 1

默认情况下， RestTemplate使用DefaultResponseErrorHandler 。 每当从REST API发送4xx / 5xx响应时，此错误处理程序就会引发异常。

如果要自定义错误处理，只需通过restTemplate.setErrorHandler注册您的自定义错误处理程序，即可在其中使用ResponseErrorHandler接口的实现。

Answer 2

为了清楚起见，我建议将带有以下详细信息的resttemplate自动连接到单独的配置文件中

@Bean
  public RestTemplate restTemplate() {
    RestTemplate restTemplate = new RestTemplate();
    restTemplate.setErrorHandler(new ErrorHandler());

    SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
    requestFactory.setOutputStreaming(false);

    restTemplate.setRequestFactory(requestFactory);

    return restTemplate;
  }

Answer 3

这是 SimpleClientHttpRequestFactory 的默认行为（使用 JDK 的内部 http 客户端）

一个简单的解决方法是使用 apache http 组件库：

ClientHttpRequestFactory  requestFactory = new HttpComponentsClientHttpRequestFactory();
restTemplate.setRequestFactory(requestFactory);

有关更多详细信息，请参阅此链接： https : //github.com/spring-projects/spring-security-oauth/issues/441#issuecomment-92033542