Springboot RestTemplate 实现连接池代码实现

在 Spring Framework 中，RestTemplate 本身并不直接支持连接池（connection pooling）功能，因为它基于底层的 HTTP 客户端库（如 JDK 的 HttpURLConnection、Apache HttpClient、OkHttp 等）来发送 HTTP 请求。如果你需要连接池功能，你可以通过配置使用支持连接池的 HTTP 客户端库，比如 Apache HttpClient 或 OkHttp，来与 RestTemplate 集成。

以下是如何使用 Apache HttpClient 作为 RestTemplate 的底层 HTTP 客户端，并启用连接池的一个示例：

1. 添加依赖

首先，确保你的项目中加入了 Apache HttpClient 和 Spring Web 的依赖。以下是一个基于 Maven 的例子：

<dependencies>  
    <!-- Spring Web -->  
    <dependency>  
        <groupId>org.springframework.boot</groupId>  
        <artifactId>spring-boot-starter-web</artifactId>  
    </dependency>  
    <!-- Apache HttpClient -->  
    <dependency>  
        <groupId>org.apache.httpcomponents</groupId>  
        <artifactId>httpclient</artifactId>  
        <version>4.5.13</version> <!-- 请检查并使用最新版本 -->  
    </dependency>  
</dependencies>

2. 配置 HttpClient

接下来，配置 Apache HttpClient 以启用连接池：

import org.apache.http.impl.client.CloseableHttpClient;  
import org.apache.http.impl.client.HttpClients;  
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;  
import org.springframework.context.annotation.Bean;  
import org.springframework.context.annotation.Configuration;  
import org.springframework.web.client.RestTemplate;  
  
@Configuration  
public class RestClientConfig {  
  
    @Bean  
    public RestTemplate restTemplate(CloseableHttpClient httpClient) {  
        return new RestTemplateBuilder()  
                .requestFactory(() -> new HttpComponentsClientHttpRequestFactory(httpClient))  
                .build();  
    }  
  
    @Bean  
    public CloseableHttpClient httpClient() {  
        PoolingHttpClientConnectionManager cm = new PoolingHttpClientConnectionManager();  
        cm.setMaxTotal(100); // 最大连接数  
        cm.setDefaultMaxPerRoute(10); // 同一路由下最大连接数  
  
        return HttpClients.custom()  
                .setConnectionManager(cm)  
                .build();  
    }  
}

3. 使用 RestTemplate

现在你可以在你的应用中注入并使用 RestTemplate，它会自动使用配置好的 Apache HttpClient 发送 HTTP 请求，并享受连接池带来的好处。

import org.springframework.beans.factory.annotation.Autowired;  
import org.springframework.stereotype.Service;  
import org.springframework.web.client.RestTemplate;  
  
@Service  
public class MyService {  
  
    private final RestTemplate restTemplate;  
  
    @Autowired  
    public MyService(RestTemplate restTemplate) {  
        this.restTemplate = restTemplate;  
    }  
  
    public String callSomeApi() {  
        String url = "http://example.com/api/some-endpoint";  
        return restTemplate.getForObject(url, String.class);  
    }  
}

总结

通过配置 Apache HttpClient 作为 RestTemplate 的底层 HTTP 客户端，你可以很容易地实现连接池功能，从而优化 HTTP 请求的性能和效率。记得选择适合你的应用需求的连接池配置（如最大连接数、同一路由下的最大连接数等）。