spring-boot RestTemplate 鏈接池

 

之前咱們項目都是基於Apache HttpClient 鏈接池進行web 接口調用，後來用spring-boot， 發現 RestTemplate 挺好用。

簡單介紹下：

 什麼是RestTemplate？

RestTemplate是Spring提供的用於訪問Rest服務的客戶端，RestTemplate提供了多種便捷訪問遠程Http服務的方法，可以大大提升客戶端的編寫效率。
調用RestTemplate的默認構造函數，RestTemplate對象在底層經過使用java.net包下的實現建立HTTP 請求，能夠經過使用ClientHttpRequestFactory指定不一樣的HTTP請求方式。
ClientHttpRequestFactory接口主要提供了兩種實現方式

一種是SimpleClientHttpRequestFactory，使用J2SE提供的方式（既java.net包提供的方式）建立底層的Http請求鏈接。
第二種方式是使用HttpComponentsClientHttpRequestFactory方式，底層使用HttpClient訪問遠程的Http服務，使用HttpClient能夠配置鏈接池和證書等信息。
RestTemplate默認是使用SimpleClientHttpRequestFactory，內部是調用jdk的HttpConnection，默認超時爲-1

 

下面我介紹下spring-boot 如何配置RestTemplate鏈接池(github)：

import org.apache.http.client.HttpClient;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.config.Registry;
import org.apache.http.config.RegistryBuilder;
import org.apache.http.conn.socket.ConnectionSocketFactory;
import org.apache.http.conn.socket.PlainConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.client.ClientHttpRequestFactory;
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
import org.springframework.web.client.RestTemplate;

@Configuration
public class RestTemplateConfig {

    @Bean
    public RestTemplate restTemplate() {
        return new RestTemplate(httpRequestFactory());
    }

    @Bean
    public ClientHttpRequestFactory httpRequestFactory() {

        return new HttpComponentsClientHttpRequestFactory(httpClient());

    }

    @Bean
    public HttpClient httpClient() {
        Registry<ConnectionSocketFactory> registry = RegistryBuilder.<ConnectionSocketFactory>create()
                .register("http", PlainConnectionSocketFactory.getSocketFactory())
                .register("https", SSLConnectionSocketFactory.getSocketFactory())
                .build();
        PoolingHttpClientConnectionManager connectionManager = new PoolingHttpClientConnectionManager(registry);
        //設置整個鏈接池最大鏈接數 根據本身的場景決定
        connectionManager.setMaxTotal(200);
        //路由是對maxTotal的細分
        connectionManager.setDefaultMaxPerRoute(100);
        RequestConfig requestConfig = RequestConfig.custom()
                .setSocketTimeout(10000) //服務器返回數據(response)的時間，超過該時間拋出read timeout
                .setConnectTimeout(5000)//鏈接上服務器(握手成功)的時間，超出該時間拋出connect timeout
                .setConnectionRequestTimeout(1000)//從鏈接池中獲取鏈接的超時時間，超過該時間未拿到可用鏈接，會拋出org.apache.http.conn.ConnectionPoolTimeoutException: Timeout waiting for connection from pool
                .build();
        return HttpClientBuilder.create()
                .setDefaultRequestConfig(requestConfig)
                .setConnectionManager(connectionManager)
                .build();
    }
}