SpringWeb 系列教程 RestTemplate 4xx/5xx 異常信息捕獲

時間 2020-01-13

標籤 springweb 系列教程 resttemplate 4xx 5xx 異常信息捕獲简体版

原文原文鏈接

200104-SpringWeb 系列教程 RestTemplate 4xx/5xx 異常信息捕獲java

近期使用 RestTemplate 訪問外部資源時，發現一個有意思的問題。由於權限校驗失敗，對方返回的 401 的 http code，此外返回數據中也會包含一些異常提示信息；然而在使用 RestTemplate 訪問時，倒是直接拋了以下提示 401 的異常，並不能拿到提示信息git

那麼 RestTemplate 若是但願能夠獲取到非 200 狀態碼返回數據時，能夠怎麼操做呢？github

web

I. 異常捕獲

1. 問題分析

RestTemplate 的異常處理，是藉助org.springframework.web.client.ResponseErrorHandler來作的，先看一下兩個核心方法spring

下面代碼來自 spring-web.5.0.7.RELEASE 版本

public interface ResponseErrorHandler {
  // 判斷是否有異常
	boolean hasError(ClientHttpResponse response) throws IOException;
  // 若是有問題，進入這個方法，處理問題
	void handleError(ClientHttpResponse response) throws IOException;
}

簡單來說，當 RestTemplate 發出請求，獲取到對方相應以後，會交給ResponseErrorHandler來判斷一下，返回結果是否 okapp

所以接下來將目標瞄準到 RestTemplate 默認的異常處理器: org.springframework.web.client.DefaultResponseErrorHandleride

a. 斷定返回結果是否 ok

從源碼上看，主要是根據返回的 http code 來判斷是否 okspring-boot

// 根據返回的http code判斷有沒有問題
@Override
public boolean hasError(ClientHttpResponse response) throws IOException {
	HttpStatus statusCode = HttpStatus.resolve(response.getRawStatusCode());
	return (statusCode != null && hasError(statusCode));
}

// 具體的斷定邏輯，簡單來說，就是返回的http code是標準的4xx, 5xx，那麼就認爲有問題了
protected boolean hasError(HttpStatus statusCode) {
	return (statusCode.series() == HttpStatus.Series.CLIENT_ERROR ||
			statusCode.series() == HttpStatus.Series.SERVER_ERROR);
}

請注意上面的實現，自定義的某些 http code 是不會被認爲是異常的，由於沒法轉換爲對應的HttpStatus （後面實例進行說明）學習

b. 異常處理

當上面的 hasError 返回 ture 的時候，就會進入異常處理邏輯ui

@Override
public void handleError(ClientHttpResponse response) throws IOException {
	HttpStatus statusCode = HttpStatus.resolve(response.getRawStatusCode());
	if (statusCode == null) {
		throw new UnknownHttpStatusCodeException(response.getRawStatusCode(), response.getStatusText(),
				response.getHeaders(), getResponseBody(response), getCharset(response));
	}
	handleError(response, statusCode);
}

protected void handleError(ClientHttpResponse response, HttpStatus statusCode) throws IOException {
	switch (statusCode.series()) {
		case CLIENT_ERROR:
			throw new HttpClientErrorException(statusCode, response.getStatusText(),
					response.getHeaders(), getResponseBody(response), getCharset(response));
		case SERVER_ERROR:
			throw new HttpServerErrorException(statusCode, response.getStatusText(),
					response.getHeaders(), getResponseBody(response), getCharset(response));
		default:
			throw new UnknownHttpStatusCodeException(statusCode.value(), response.getStatusText(),
					response.getHeaders(), getResponseBody(response), getCharset(response));
	}
}

從上面也能夠看到，異常處理邏輯很簡單，直接拋異常

2. 異常捕獲

定位到生面的問題以後，再想解決問題就相對簡單了，自定義一個異常處理類，無論狀態碼返回是啥，全都認爲正常便可

RestTemplate restTemplate = new RestTemplate();
restTemplate.setErrorHandler(new DefaultResponseErrorHandler(){
    @Override
    protected boolean hasError(HttpStatus statusCode) {
        return super.hasError(statusCode);
    }

    @Override
    public void handleError(ClientHttpResponse response) throws IOException {
    }
});

3. 實測

首先寫兩個結果，返回的 http 狀態碼非 200；針對返回非 200 狀態碼的 case，有多種寫法，下面演示兩種常見的

@RestController
public class HelloRest {
  @GetMapping("401")
  public ResponseEntity<String> _401(HttpServletResponse response) {
      ResponseEntity<String> ans =
              new ResponseEntity<>("{\"code\": 401, \"msg\": \"some error!\"}", HttpStatus.UNAUTHORIZED);
      return ans;
  }

  @GetMapping("525")
  public String _525(HttpServletResponse response) {
      response.setStatus(525);
      return "{\"code\": 525, \"msg\": \"自定義錯誤碼!\"}";
  }
}

首先來看一下自定義的 525 和標準的 401 http code，直接經過RestTemplate訪問的 case

@Test
public void testCode() {
    RestTemplate restTemplate = new RestTemplate();
    HttpEntity<String> ans = restTemplate.getForEntity("http://127.0.0.1:8080/525", String.class);
    System.out.println(ans);

    ans = restTemplate.getForEntity("http://127.0.0.1:8080/401", String.class);
    System.out.println(ans);
}

從上面的輸出結果也能夠看出來，非標準 http code 不會拋異常（緣由上面有分析），接下來看一下即使是標準的 http code 也不但願拋異常的 case

@Test
public void testSend() {
    String url = "http://127.0.0.1:8080/401";
    RestTemplate restTemplate = new RestTemplate();
    restTemplate.setErrorHandler(new DefaultResponseErrorHandler(){
        @Override
        protected boolean hasError(HttpStatus statusCode) {
            return super.hasError(statusCode);
        }

        @Override
        public void handleError(ClientHttpResponse response) throws IOException {
        }
    });
    HttpEntity<String> ans = restTemplate.getForEntity(url, String.class);
    System.out.println(ans);
}