当通过RestTemplate调用服务发生异常时,往往会返回400 Bad Request或500 internal error等错误信息。如果想捕捉服务本身抛出的异常信息,需要通过自行实现RestTemplate的ErrorHandler。
RestTemplate实例可以通过调用setErrorHandler方法设置ErrorHandler,实现对请求响应异常的判别和处理。自定义的ErrorHandler需实现ResponseErrorHandler接口,同时Spring boot也提供了默认实现DefaultResponseErrorHandler,因此也可以通过继承该类来实现自己的ErrorHandler。
getForObject和postForObject方法调用底层doExecute方法来执行HTTP请求,通过Spring boot中doExecute方法可以看到RestTemplate在进行HTTP请求时分成了三个步骤:
1)创建请求,获取响应;
2)判断响应是否异常,处理异常
3)将响应消息体封装为java对象
Object varl4;
// 1 创建请求,获取响应
ClientHttpRequest request = this.createRequest(url, method);
if (requestCallback != null) {
requestCallback.doWithRequest(request);
}
response = request.execute();
// 2 判断响应是否存在异常,处理异常
this.handleResponse(url, method, response);
// 3 将响应消息体封装为java对象
if (responseExtractor == null) {
resource = null;
return resource;
}
var14 = responseExtractor.extractData(response);
在handleResponse方法中对调用ErrorHandler来判断响应是否异常,并处理异常。这里需要注意的是,如果自定义ErrorHandler中的handlerError方法中获取了response中body内容就需要抛出异常,防止doExecute方法继续执行responseExtractor.extractData(response)语句导致response.body(类型为inputstream)被重复读取。
protected void handleResponse(URI url, HttpMethod method, ClientHttpResponse response) throws IOException {
ResponseErrorHandler errorHandler = this.getErrorHandler();
boolean hasError = errorHandler.hasError(response);
if (this.logger.isDebugEnabled()) {
try {
this.logger.debug(method.name() + " request for \"" + url + "\" resulted in " + response.getRawStatusCode() + " (" + response.getStatusText() + ")" + (hasError ? "; invoking error handler" : ""));
} catch (IOException var7) {
;
}
}
if (hasError) {
errorHandler.handleError(url, method, response);
}
}
学习了ErrorHandler在RestTemplate中的调用,开始实现自定义的ErrorHandler。首先创建自定义异常(由于ResponseErrorHandler中定义了handlerError方法抛出IOException,因此自定义的异常只能为RuntimeException)
public class MyException extends RuntimeException {
public MyException(String message){
super(message);
}
public MyException(String message, Throwable e){
super(message + e.getLocalizedMessage());
}
}
实现自定义ErrorHandler,一种思路是根据响应消息体进行相应的异常处理策略,对于其他异常情况由父类DefaultResponseErrorHandler来进行处理。
public class MyErrorHandler extends DefaultResponseErrorHandler {
@Override
public boolean hasError(ClientHttpResponse response) throws IOException{
return super.hasError(response);
}
@Override
public void handleError(ClientHttpResponse response) throws IOException{
Scanner scanner = new Scanner(response.getBody()).useDelimiter("\\A");
String stringResponse = scanner.hasNext() ? scanner.next() : "";
if(stringResponse.matches(".*XXX.*")){
throw new MyException(stringResponse);
}
else{
super.handleError(response);
}
}
}