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源码解析（目录）

---