Spring中RestTemplate源码解析

在实现客户端的负载均衡中，有一种简单的方式就是通过调用被@LoadBalanced注解过的RestTemplate来实现面向服务的接口调用。

例如：

@Bean @LoadBalanced RestTemplate restTemplate() { //return new RestTemplate(); return new SimpleRestTemplateBuilder().build(); }

这里我没有使用默认的new RestTemplate()，我使用了自定义的配置

public class SimpleRestTemplateBuilder { public SimpleRestTemplateBuilder() { } public RestTemplate build() { HttpComponentsClientHttpRequestFactory factory = new HttpComponentsClientHttpRequestFactory(); factory.setReadTimeout(15000); factory.setConnectTimeout(3000); RestTemplate restTemplate = new RestTemplate(factory); List<ClientHttpRequestInterceptor> interceptors = new ArrayList(); //添加自定义拦截器RestTemplateAutoHeaderInterceptor interceptors.add(new RestTemplateAutoHeaderInterceptor()); restTemplate.setInterceptors(interceptors); //添加自定义处理器CustomResponseErrorHandler restTemplate.setErrorHandler(new CustomResponseErrorHandler()); return restTemplate; } }

使用时直接

String resp = restTemplate.postForEntity("http://WXGATEWAY-SERVICE/internal/push_channel_configs", weixinConfigDTOList, String.class).getBody();

 

这是常见的RestTemplate使用方式。接下来看一下RestTemplate的源码。

public <T> ResponseEntity<T> postForEntity(URI url, Object request, Class<T> responseType) throws RestClientException { //这是个构造方法，设置了request为HttpEntity，设置了返回值类型为responseType RequestCallback requestCallback = this.httpEntityCallback(request, responseType); //把返回类型包装成数据提取器ResponseExtractor，当返回结果后转换成对应的类型对象。 ResponseExtractor<ResponseEntity<T>> responseExtractor = this.responseEntityExtractor(responseType); return (ResponseEntity)this.execute(url, HttpMethod.POST, requestCallback, responseExtractor); }

核心方法在于org.springframework.web.client.RestTemplate#doExecute。

通过ClientHttpRequestFactory创建了一个ClientHttpRequest。

ClientHttpRequest request = this.createRequest(url, method);

然后调用execute()方法返回了一个ClientHttpResponse

org.springframework.http.client.AbstractClientHttpRequest#execute

ClientHttpResponse response = request.execute();

然后用之前创建RestTemplate时添加的自定义的处理器处理返回结果

this.handleResponse(url, method, response);

最后使用数据提起器提取信息转换成返回类型数据。

//这个Object的类型是ResponseEntity，其中的body就是调用的时候填写的返回值类型 Object var14 = responseExtractor.extractData(response);

继续跟进执行方法，找到

SimpleBufferingClientHttpRequest类上，最终调用的是HttpURLConnection。

org.springframework.http.client.SimpleBufferingClientHttpRequest#executeInternal。

 

其余的已经有前辈写好了。我们可以参照一下别人的文章

参考：RestTemplate源码解析（目录）

---