restTemplate 配置连接池、使用map进行传参和设置headers

最新推荐文章于 2024-09-20 13:07:19 发布

weixin_44728369

最新推荐文章于 2024-09-20 13:07:19 发布

阅读量1k

点赞数

本文链接：https://blog.csdn.net/weixin_44728369/article/details/120348499

版权

SpringCloud 专栏收录该内容

27 篇文章 0 订阅

订阅专栏

1.通过HttpClient配置连接池

package com.kittlen.cloud.config;

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.beans.factory.annotation.Value;
import org.springframework.cloud.client.loadbalancer.LoadBalanced;
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;

/**
 * @author kittlen
 * @version 1.0
 * @date 2021/2/22 13:28
 */
@Configuration
public class RestTemplateConfig{

    @Value("${restTemplate.maxTotal:200}")
    int maxTotal;

    @Value("${restTemplate.defaultMaxPerRoute:100}")
    int defaultMaxPerRoute;

    @Value("${restTemplate.socketTimeout:10000}")
    int socketTimeout;

    @Value("${restTemplate.connectTimeout:5000}")
    int connectTimeout;

    @Value("${restTemplate.connectionRequestTimeout:1000}")
    int connectionRequestTimeout;

    @Bean
    @LoadBalanced
    public RestTemplate getRestTemplate() {
        RestTemplate restTemplate = new RestTemplate(clientHttpRequestFactory());
        restTemplate.getInterceptors().add(new HttpInterceptor());
        return restTemplate;
    }

    @Bean
    public ClientHttpRequestFactory clientHttpRequestFactory() {
        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(maxTotal);
        //路由是对maxTotal的细分,单个请求的最大连接数
        connectionManager.setDefaultMaxPerRoute(defaultMaxPerRoute);
        RequestConfig requestConfig = RequestConfig.custom()
                .setSocketTimeout(socketTimeout) //服务器返回数据(response)的时间，超过该时间抛出read timeout
                .setConnectTimeout(connectTimeout)//连接上服务器(握手成功)的时间，超出该时间抛出connect timeout
                .setConnectionRequestTimeout(connectionRequestTimeout)//从连接池中获取连接的超时时间，超过该时间未拿到可用连接，会抛出org.apache.http.conn.ConnectionPoolTimeoutException: Timeout waiting for connection from pool
                .build();
        return HttpClientBuilder.create()
                .setDefaultRequestConfig(requestConfig)
                .setConnectionManager(connectionManager)
                .build();
    }
}

2.使用map进行传参并添加headers

/**
     * 使用map进行传值
     *
     * @return
     */
    @PostMapping("/byMap")
    public Result<List<Test>> byMap() {
        //设置请求的参数
        MultiValueMap<String, String> map = new LinkedMultiValueMap<>();
        map.add("author", "kittlen");
        //设置请求头
        MultiValueMap<String, String> headers = new LinkedMultiValueMap<>();
        headers.add(HttpHeaders.ACCEPT, "application/json");//*设置期望的接收数据类型,在使用exchange进行制定数据类型,进行数据获取时至关重要(默认为xml方式)
        try {
            ResponseEntity<Result<List<Test>>> exchange = restTemplate.exchange(HTTP + "/test/getTestAll", HttpMethod.POST,
                    new HttpEntity<>(map, headers), new ParameterizedTypeReference<Result<List<Test>>>() {
                    });
            return exchange.getBody();
        } catch (Exception e) {
            return Result.error();
        }
    }