cc.kkon.request.GmSSLRequests Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tomcat-embed-gmssl Show documentation
支持国密 HTTPS 的嵌入式 Tomcat 容器。
The newest version!
package cc.kkon.request;

import org.apache.http.HttpResponse;
import org.apache.http.HttpVersion;
import org.apache.http.NameValuePair;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.client.utils.URIBuilder;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.message.BasicNameValuePair;

import java.io.IOException;
import java.net.URISyntaxException;
import java.util.*;

/**
 * 支持国密 https 的请求类
 */
public class GmSSLRequests {


    private static final String ENCODING = "UTF-8";


    private static final int timeout = 30 * 1000;

    public static Response0 post(String url, Map params, Map headers) throws IOException {
        HttpPost httpPost = new HttpPost(url);
        httpPost.setProtocolVersion(HttpVersion.HTTP_1_1);
        putHeaders(httpPost, headers);
        /*
         * 处理参数
         */
        List list = new ArrayList<>();
        if (params != null) {
            Set keySet = params.keySet();
            for (String key : keySet) {
                list.add(new BasicNameValuePair(key, params.get(key)));
            }
        }

        httpPost.setEntity(new UrlEncodedFormEntity(list, ENCODING));

        HttpResponse response = ClientBuilder.initGMSSL().execute(httpPost);
        Response0 res0 = new Response0(response);
        httpPost.abort();
        return res0;
    }

    public static Response0 post4json(String url, String json, Map headers) throws IOException {
        HttpPost httpPost = new HttpPost(url);
        httpPost.setProtocolVersion(HttpVersion.HTTP_1_1);
        putHeaders(httpPost, headers);

        StringEntity entity = new StringEntity(json, ContentType.APPLICATION_JSON);

        httpPost.setEntity(entity);

        HttpResponse response = ClientBuilder.initGMSSL().execute(httpPost);
        Response0 res0 = new Response0(response);
        httpPost.abort();
        return res0;
    }

    public static Response0 get(String url) throws IOException {
        return get(url, null, null);
    }

    public static Response0 get(String url, Map params, Map headers) throws IOException {
        HttpGet httpGet = null;
        try {
            URIBuilder uriBuilder = new URIBuilder(url);
            // 设置请求参数
            if (params != null && !params.isEmpty()) {
                List list = new LinkedList<>();
                for (Map.Entry entry : params.entrySet()) {
                    list.add(new BasicNameValuePair(entry.getKey(), entry.getValue()));
                }
                uriBuilder.setParameters(list);
            }

            httpGet = new HttpGet(uriBuilder.build());
        } catch (URISyntaxException e) {
            throw new RuntimeException(e);
        }
        // 设置超时时间
        if (timeout > 0) {
            RequestConfig requestConfig = RequestConfig.custom().setSocketTimeout(timeout).setConnectTimeout(timeout).build();
            httpGet.setConfig(requestConfig);
        }

        httpGet.setProtocolVersion(HttpVersion.HTTP_1_1);

        putHeaders(httpGet, headers);
        HttpResponse response = ClientBuilder.initGMSSL().execute(httpGet);

        Response0 res0 = new Response0(response);
        httpGet.abort();
        return res0;
    }

    /**
     * 设置请求头
     */
    private static void putHeaders(HttpRequestBase httpMethod, Map headers) {
        if (headers != null) {
            for (Map.Entry entry : headers.entrySet()) {
                httpMethod.setHeader(entry.getKey(), entry.getValue());
            }
        }
    }
}