Http资源池结合springboot搭建教程

当http请求特别频繁的时候建议使用http资源池,好处这里不在赘述。这里采用okhttp2结合springboot搭建完整流程,大概分为4个步骤:

1.maven引入依赖

       <dependency>
            <groupId>com.squareup.okhttp3</groupId>
            <artifactId>okhttp</artifactId>
            <version>4.9.3</version>
        </dependency>

2.配置文件参数设置,根据自己的实际情况配置参数

ok.http.connect-timeout=60
ok.http.read-timeout=60
ok.http.write-timeout=60
# 连接池中整体的空闲连接的最大数量
ok.http.max-idle-connections=20
# 连接空闲时间最多为 300 秒
ok.http.keep-alive-duration=300

3.配置资源池单列模式

package com.midea.datauserincr.config;

import okhttp3.ConnectionPool;
import okhttp3.OkHttpClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import javax.net.ssl.TrustManager;
import javax.net.ssl.X509TrustManager;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
import java.util.concurrent.TimeUnit;

/**
 * @Author:ex_weiwh6
 * @Package:com.midea.datauserincr.config
 * @Project:data-user-incr
 * @name:OkHttpConfiguration
 * @Date:2025/2/28 09:34
 * @Filename:OkHttpConfiguration
 */

@Configuration
public class OkHttpConfiguration {

    @Value("${ok.http.connect-timeout}")
    private Integer connectTimeout;
    @Value("${ok.http.read-timeout}")
    private Integer readTimeout;
    @Value("${ok.http.write-timeout}")
    private Integer writeTimeout;
    @Value("${ok.http.max-idle-connections}")
    private Integer maxIdleConnections;
    @Value("${ok.http.keep-alive-duration}")
    private Long keepAliveDuration;

    @Bean
    public OkHttpClient okHttpClient() {
        return new OkHttpClient.Builder()
                .sslSocketFactory(sslSocketFactory(), x509TrustManager())
                // 是否开启缓存
                .retryOnConnectionFailure(false)
                .connectionPool(pool())
                .connectTimeout(connectTimeout, TimeUnit.SECONDS)
                .readTimeout(readTimeout, TimeUnit.SECONDS)
                .writeTimeout(writeTimeout, TimeUnit.SECONDS)
                .hostnameVerifier((hostname, session) -> true)
                // 设置代理
                // .proxy(new Proxy(Proxy.Type.HTTP, new InetSocketAddress("127.0.0.1", 8888)))
                // 拦截器
                // .addInterceptor()
                .build();
    }

    @Bean
    public X509TrustManager x509TrustManager() {
        return new X509TrustManager() {
            @Override
            public void checkClientTrusted(X509Certificate[] chain, String authType)
                    throws CertificateException {
            }

            @Override
            public void checkServerTrusted(X509Certificate[] chain, String authType)
                    throws CertificateException {
            }

            @Override
            public X509Certificate[] getAcceptedIssuers() {
                return new X509Certificate[0];
            }
        };
    }

    @Bean
    public SSLSocketFactory sslSocketFactory() {
        try {
            // 信任任何链接
            SSLContext sslContext = SSLContext.getInstance("TLS");
            sslContext.init(null, new TrustManager[]{x509TrustManager()}, new SecureRandom());
            return sslContext.getSocketFactory();
        } catch (NoSuchAlgorithmException | KeyManagementException e) {
            e.printStackTrace();
        }
        return null;
    }

    @Bean
    public ConnectionPool pool() {
        return new ConnectionPool(maxIdleConnections, keepAliveDuration, TimeUnit.SECONDS);
    }
}

4.使用spring IOC容器管理,并自动注入,创建一个工具类,这里写的多可以根据自己的业务定制或者删除方法

package com.midea.datauserincr.util;

import lombok.extern.slf4j.Slf4j;
import okhttp3.*;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.jetbrains.annotations.Nullable;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.io.IOException;
import java.util.Map;

/**
 * @Author:ex_weiwh6
 * @Package:com.midea.datauserincr.util
 * @Project:data-user-incr
 * @name:OkHttpClientUtil
 * @Date:2025/2/28 09:39
 * @Filename:OkHttpClientUtil
 */
@Slf4j
@Component
public class OkHttpClientUtil {
    private static final MediaType JSON = MediaType.parse("application/json; charset=utf-8");
    private static final MediaType XML = MediaType.parse("application/xml; charset=utf-8");

    private static OkHttpClient okHttpClient;

    // 为使用静态调用异步注入
    @Autowired
    private OkHttpClient getOkHttpClient;
    @PostConstruct
    public void init() {
        okHttpClient = getOkHttpClient;
    }

    /**
     * get请求
     * @param httpUrl 请求地址
     * @param auth 认证
     * @param okCode 返回码
     * @return 返回结果
     */
    public static String  doGet(String httpUrl, String auth, int okCode) {
        Request request = new Request.Builder()
                .url(httpUrl)
                .header("Content-Type", "application/json")
                .addHeader("Authorization", "Bearer " + auth)
                .get()
                .build();

        return executeResponse(okCode, request);
    }

    /**
     * post请求
     * @param httpUrl 请求地址
     * @param auth 认证
     * @param okCode 返回码
     * @param param 参数
     * @return 返回结果
     */
    public static String doPost(String httpUrl, String auth, int okCode, String param) {
        Request request = new Request.Builder()
                .url(httpUrl)
                .post(RequestBody.create(MediaType.parse("application/json"), param.getBytes()))
                .addHeader("Authorization", "Bearer " + auth)
                .addHeader("Content-Type", "application/json")
                .build();

        return executeResponse(okCode, request);

    }




    /**
     * PUT请求
     * @param httpUrl 请求地址
     * @param auth 认证
     * @param okCode 返回码
     * @param param 参数
     * @return 返回结果
     */
    public static String doPut(String httpUrl, String auth, int okCode, String param) {
        Request request = new Request.Builder()
                .url(httpUrl)
                .put(RequestBody.create(MediaType.parse("application/json"), param.getBytes()))
                .addHeader("Authorization", "Bearer " + auth)
                .build();
        return executeResponse(okCode, request);

    }

    /**
     * DELETE请求
     * @param httpUrl 请求地址
     * @param auth 认证
     * @param okCode 返回码
     * @return 返回结果
     */
    public static boolean doDelete(String httpUrl, String auth, int okCode) {
        Request request = new Request.Builder()
                .url(httpUrl)
                .delete()
                .addHeader("Authorization", "Bearer " + auth)
                .build();
        try(Response response = okHttpClient.newCall(request).execute()) {
            if (response.code() == okCode) {
                log.info("Request successful, response.code:{} response content: {}", response.code(), response.body().string());
                return true;
            } else {
                log.error("Request failed, error code: {}", response.code());
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return false;

    }

    /**
     * PATCH请求
     * @param httpUrl 请求地址
     * @param auth 认证
     * @param okCode 返回码
     * @param param 参数
     * @return 返回结果
     */
    public static String doPatch(String httpUrl, String auth, int okCode, String param) {
        //发送的json
        RequestBody requestBody = RequestBody.create(JSON, param);
        Request request = new Request.Builder()
                //请求的url
                .url(httpUrl)
                .patch(requestBody)
                .header("Authorization", "Bearer " + auth)
                .addHeader("Content-Type", "application/json")
                .build();
        return executeResponse(okCode, request);

    }


    @Nullable
    private static String executeResponse(int okCode, Request request) {
        try (Response response = okHttpClient.newCall(request).execute()) {
            if (response.code() == okCode) {
                String responseBody = response.body().string();
                log.debug("Request successful, response content: {}", responseBody);
                return responseBody;
            } else {
                log.error("Request failed, error code: {}", response.code());
            }
        } catch (IOException e) {
            log.error("Request failed, error message: {}", e.getMessage());
            e.printStackTrace();
        }

        return null;
    }





    /**
     * get 请求
     * @param url  请求url地址
     * @return string
     * */
    public static String doGet(String url) {
        return doGet(url, null, null);
    }

    public static byte[] doGetByte(String url) {
        return doGetByte(url, null, null);
    }

    public static String doPost(String url) {
        return doPost(url, null, null);
    }
    /**
     * get 请求
     * @param url  请求url地址
     * @param params 请求参数 map
     * @return string
     * */
    public static String doGetToParams(String url, Map<String, String> params) {
        return doGet(url, params, null);
    }
    /**
     * get 请求
     * @param url  请求url地址
     * @param headers 请求头字段 {k1, v1 k2, v2, ...}
     * @return string
     * */
    public static String doGetToHeaders(String url, Map<String, String> headers) {
        return doGet(url, null, headers);
    }
    /**
     * get 请求
     * @param url  请求url地址
     * @param params 请求参数 map
     * @param headers 请求头字段 {k1, v1 k2, v2, ...}
     * @return string
     * */
    public static String doGet(String url, Map<String, String> params, Map<String, String> headers) {
        StringBuilder sb = new StringBuilder(url);
        if (params != null && params.keySet().size() > 0) {
            boolean firstFlag = true;
            for (String key : params.keySet()) {
                if (firstFlag) {
                    sb.append("?").append(key).append("=").append(params.get(key));
                    firstFlag = false;
                } else {
                    sb.append("&").append(key).append("=").append(params.get(key));
                }
            }
        }
        Request.Builder builder = new Request.Builder();
        if (headers != null && !headers.isEmpty()) {
            for (String header:headers.keySet()){
                builder.addHeader(header, headers.get(header));
            }
        }
        Request request = builder.url(sb.toString()).build();
        log.info("do get request and url[{}]", sb.toString());
        return executeBody(request);
    }

    public static byte[] doGetByte(String url, Map<String, String> params, Map<String, String> headers) {
        StringBuilder sb = new StringBuilder(url);
        if (params != null && params.keySet().size() > 0) {
            boolean firstFlag = true;
            for (String key : params.keySet()) {
                if (firstFlag) {
                    sb.append("?").append(key).append("=").append(params.get(key));
                    firstFlag = false;
                } else {
                    sb.append("&").append(key).append("=").append(params.get(key));
                }
            }
        }
        Request.Builder builder = new Request.Builder();
        if (headers != null && !headers.isEmpty()) {
            for (String header:headers.keySet()){
                builder.addHeader(header, headers.get(header));
            }
        }
        Request request = builder.url(sb.toString()).build();
        log.info("do get request and url[{}]", sb.toString());
        return executeByte(request);
    }
    /**
     * post 请求
     * @param url  请求url地址
     * @param params 请求参数 map
     * @return string
     */
    public static String doPostForm(String url, Map<String, String> params) {
        FormBody.Builder builder = new FormBody.Builder();
        if (params != null && params.keySet().size() > 0) {
            for (String key : params.keySet()) {
                builder.add(key, params.get(key));
            }
        }
        Request request = new Request.Builder().url(url).post(builder.build()).build();
        log.info("do post request and url[{}]", url);
        return execute(request);
    }

    /**
     * post 请求
     * @param url  请求url地址
     * @param params 请求参数 map
     * @param headers 请求头字段 {k1:v1, k2: v2, ...}
     * @return string
     */
    public static String doPost(String url, Map<String, String> params, Map<String, String> headers) {
        FormBody.Builder builder = new FormBody.Builder();
        if (params != null && params.keySet().size() > 0) {
            for (String key : params.keySet()) {
                builder.add(key, params.get(key));
            }
        }
        Request.Builder requestBuilder = new Request.Builder();
        if (headers != null && !headers.isEmpty()) {
            for (String header:headers.keySet()){
                requestBuilder.addHeader(header, headers.get(header));
            }
        }
        Request request = requestBuilder.url(url).post(builder.build()).build();
        log.info("do post request and url[{}]", url);
        return execute(request);
    }
    /**
     * post 请求, 请求数据为 json 的字符串
     * @param url  请求url地址
     * @param json  请求数据, json 字符串
     * @return string
     */
    public static String doPostJson(String url, String json) {
        log.info("do post request and url[{}]", url);
        return executePost(url, json, JSON);
    }
    /**
     * post 请求, 请求数据为 json 的字符串
     * @param url  请求url地址
     * @param json  请求数据, json 字符串
     * @param headers 请求头字段 {k1, v1 k2, v2, ...}
     * @return string
     */
    public static String doPostJson(String url, String json, Map<String, String> headers) {
        log.info("do post request and url[{}]", url);

        RequestBody requestBody = RequestBody.create(json, JSON);
        Request.Builder builder = new Request.Builder();
        if (headers != null && !headers.isEmpty()) {
            for (String header:headers.keySet()){
                builder.addHeader(header, headers.get(header));
            }
        }
        Request request = builder.url(url).post(requestBody).build();
        return execute(request);
    }
    /**
     * post 请求, 请求数据为 xml 的字符串
     * @param url  请求url地址
     * @param xml  请求数据, xml 字符串
     * @return string
     */
    public static String doPostXml(String url, String xml) {
        log.info("do post request and url[{}]", url);
        return executePost(url, xml, XML);
    }
    private static String executePost(String url, String data, MediaType contentType) {
        RequestBody requestBody = RequestBody.create(data ,contentType);
        Request request = new Request.Builder().url(url).post(requestBody).build();
        return execute(request);
    }
    private static String execute(Request request) {
        Response response = null;
        try {
            response = okHttpClient.newCall(request).execute();
            if (response.isSuccessful()) {
                return response.body().string();
            }
        } catch (Exception e) {
            log.error(ExceptionUtils.getStackTrace(e));
        } finally {
            if (response != null) {
                response.close();
            }
        }
        return "";
    }

    private static String executeBody(Request request) {
        Response response = null;
        try {
            response = okHttpClient.newCall(request).execute();
            if (response.body() != null) {
                return response.body().string();
            }
        } catch (Exception e) {
            log.error(ExceptionUtils.getStackTrace(e));
        } finally {
            if (response != null) {
                response.close();
            }
        }
        return "";
    }

    private static byte[] executeByte(Request request) {
        Response response = null;
        try {
            response = okHttpClient.newCall(request).execute();
            if (response.isSuccessful()) {
                return response.body().bytes();
            }
        } catch (Exception e) {
            log.error(ExceptionUtils.getStackTrace(e));
        } finally {
            if (response != null) {
                response.close();
            }
        }
        return null;
    }


}

5.测试一下

@RestController
@RequestMapping("httpapi")
public class testController {
    
    @GetMapping("test")
    public String test(){
        return OkHttpClientUtil.doGet("https://www.ai.com");
    }
}

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容