java/meitianmeiya.git

package com.sinata.rest.core.util;
 
import cn.hutool.core.collection.CollUtil;
import com.alibaba.fastjson.JSONObject;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpPost;
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.DefaultHostnameVerifier;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.conn.BasicHttpClientConnectionManager;
import org.apache.http.util.EntityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.*;
import org.springframework.stereotype.Component;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestTemplate;
 
import javax.net.ssl.KeyManagerFactory;
import javax.net.ssl.SSLContext;
import java.io.FileInputStream;
import java.nio.charset.StandardCharsets;
import java.security.KeyStore;
import java.security.SecureRandom;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
 
@Component
public class RestTemplateUtil {
 
    @Autowired
    private RestTemplate restTemplate;
 
    public static final String WXPAYSDK_VERSION = "WXPaySDK/3.0.9";
    public static final String USER_AGENT = WXPAYSDK_VERSION +
            " (" + System.getProperty("os.arch") + " " + System.getProperty("os.name") + " " + System.getProperty("os.version") +
            ") Java/" + System.getProperty("java.version") + " HttpClient/" + HttpClient.class.getPackage().getImplementationVersion();
 
    /**
     * 发送GET请求
     *
     * @param url
     * @param param
     * @return
     */
 
    public String getData(String url, Map<String, String> param) {
        // 请勿轻易改变此提交方式，大部分的情况下，提交方式都是表单提交
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
        return restTemplate.getForEntity(url, String.class, param).getBody();
    }
 
    /**
     * 发送GET请求
     *
     * @param url
     * @return
     */
 
    public String getLink(String url) {
        return restTemplate.getForEntity(url, String.class).getBody();
    }
 
 
    /**
     * 发送GET请求
     *
     * @param url
     * @return
     */
 
    public JSONObject getData(String url) {
        HttpHeaders headers = new HttpHeaders();
        headers.add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36");
        headers.setContentType(MediaType.APPLICATION_JSON);
 
        HttpEntity<Map<String, Object>> requestEntity =
                new HttpEntity<>(headers);
 
 
        ResponseEntity<String> responseEntity = restTemplate.exchange(
                url,
                HttpMethod.POST,
                requestEntity,
                String.class);
 
        String body = responseEntity.getBody();
        return JSONObject.parseObject(body);
    }
 
    /**
     * 发送GET请求 支持header
     *
     * @param url
     * @return
     */
 
    public JSONObject getData(String url, HashMap<String, String> hashMap) {
 
        HttpHeaders headers = new HttpHeaders();
        headers.add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36");
        headers.setContentType(MediaType.APPLICATION_JSON_UTF8);
 
        for (Map.Entry<String, String> entry : hashMap.entrySet()) {
            headers.add(entry.getKey(), entry.getValue());
        }
 
        HttpEntity<Map<String, Object>> requestEntity =
                new HttpEntity<>(headers);
 
        return restTemplate.exchange(url, HttpMethod.GET, requestEntity, JSONObject.class).getBody();
    }
 
 
    /**
     * 发送POST-JSON请求
     *
     * @param url
     * @param param
     * @return
     */
 
    public String postJsonData(String url, JSONObject param) {
 
        HttpHeaders headers = new HttpHeaders();
 
        headers.setContentType(MediaType.APPLICATION_JSON_UTF8);
 
        headers.add("Accept", MediaType.APPLICATION_JSON.toString());
 
        HttpEntity<JSONObject> requestEntity = new HttpEntity<JSONObject>(param, headers);
 
        return restTemplate.postForEntity(url, param, String.class).getBody();
 
    }
 
    /**
     * 发送POST 表单请求
     *
     * @param url
     * @return
     */
 
    public String postFormData(String url, MultiValueMap<String, String> map) {
        HttpHeaders headers = new HttpHeaders();
//        headers.setContentType(MediaType.MULTIPART_FORM_DATA);
        HttpEntity<MultiValueMap<String, String>> requests = new HttpEntity<MultiValueMap<String, String>>(map, headers);
        String body = restTemplate.postForEntity(url, requests, String.class).getBody();
        return body;
 
 
    }
 
 
    /**
     * 发送POST-JSON请求
     *
     * @param url
     * @param param
     * @return
     */
 
    public JSONObject postJsonDataAndReturnJson(String url, JSONObject param) {
        HttpHeaders headers = new HttpHeaders();
 
        headers.setContentType(MediaType.APPLICATION_JSON_UTF8);
        headers.add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36");
 
        headers.add("Accept", MediaType.APPLICATION_JSON.toString());
 
        HttpEntity<JSONObject> requestEntity = new HttpEntity<JSONObject>(param, headers);
 
        JSONObject body = restTemplate.postForEntity(url, requestEntity, JSONObject.class).getBody();
 
        return body;
 
    }
 
    /**
     * 发送POST-JSON请求
     *
     * @param url
     * @return
     */
 
    public JSONObject post(String url) {
        HttpHeaders headers = new HttpHeaders();
 
        headers.setContentType(MediaType.APPLICATION_JSON_UTF8);
        headers.add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36");
 
        headers.add("Accept", MediaType.APPLICATION_JSON.toString());
 
        HttpEntity<String> requestEntity = new HttpEntity<>(headers);
 
        JSONObject body = restTemplate.postForEntity(url, requestEntity, JSONObject.class).getBody();
 
        return body;
 
    }
 
    /**
     * 发送POST-JSON请求
     *
     * @param url
     * @return
     */
 
    public String postXml(String url, String xml) {
        HttpHeaders headers = new HttpHeaders();
 
        MediaType type = MediaType.parseMediaType("application/xml; charset=UTF-8");
        headers.setContentType(type);
        HttpEntity<String> requestEntity = new HttpEntity<>(xml, headers);
 
        ResponseEntity<String> responseEntity = restTemplate.postForEntity(url, requestEntity, String.class);
        try {
            System.out.println("responseEntity" + responseEntity);
            return new String(Objects.requireNonNull(responseEntity.getBody()).getBytes("UTF-8"), StandardCharsets.UTF_8);
        } catch (Exception e) {
            return "";
        }
    }
 
    /**
     * 发送POST-JSON请求(微信退款专用)
     *
     * @param url
     * @return
     */
 
    public String postWXRefundXml(String url, String xml, String mchId, String path) throws Exception {
        KeyStore clientStore = KeyStore.getInstance("PKCS12");
        // 读取本机存放的PKCS12证书文件
        FileInputStream instream = new FileInputStream(path);
        try {
            // 指定PKCS12的密码(商户ID)
            clientStore.load(instream, mchId.toCharArray());
        } finally {
            instream.close();
        }
 
        // 实例化密钥库 & 初始化密钥工厂
        KeyManagerFactory kmf = KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm());
        kmf.init(clientStore, mchId.toCharArray());
 
        // 创建 SSLContext
        SSLContext sslContext = SSLContext.getInstance("TLS");
        sslContext.init(kmf.getKeyManagers(), null, new SecureRandom());
 
        SSLConnectionSocketFactory sslConnectionSocketFactory = new SSLConnectionSocketFactory(
                sslContext,
                new String[]{"TLSv1"},
                null,
                new DefaultHostnameVerifier());
 
        BasicHttpClientConnectionManager connManager = new BasicHttpClientConnectionManager(
                RegistryBuilder.<ConnectionSocketFactory>create()
                        .register("http", PlainConnectionSocketFactory.getSocketFactory())
                        .register("https", sslConnectionSocketFactory)
                        .build(),
                null,
                null,
                null
        );
 
        HttpClient httpClient = HttpClientBuilder.create()
                .setConnectionManager(connManager)
                .build();
 
        HttpPost httpPost = new HttpPost(url);
 
        RequestConfig requestConfig = RequestConfig.custom().setSocketTimeout(8 * 1000).setConnectTimeout(6 * 1000).build();
        httpPost.setConfig(requestConfig);
 
        StringEntity postEntity = new StringEntity(xml, "UTF-8");
        httpPost.addHeader("Content-Type", "text/xml");
        httpPost.addHeader("User-Agent", USER_AGENT + " " + mchId);
        httpPost.setEntity(postEntity);
 
        HttpResponse httpResponse = httpClient.execute(httpPost);
        org.apache.http.HttpEntity httpEntity = httpResponse.getEntity();
        return EntityUtils.toString(httpEntity, "UTF-8");
    }
 
    /**
     * 发送POST 表单请求
     *
     * @param url
     * @param params
     * @return 表单提交，返回数据分为两部分，一个代表成功，一个是返回的数据内容
     */
 
    public String postFormData(String url, LinkedMultiValueMap<String, Object> params) {
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.MULTIPART_FORM_DATA);
 
        HttpEntity<LinkedMultiValueMap<String, Object>> requestEntity =
                new HttpEntity<>(params, headers);
 
        ResponseEntity<String> responseEntity = restTemplate.exchange(
                url,
                HttpMethod.POST,
                requestEntity,
                String.class);
 
        String body = responseEntity.getBody();
 
        return body;
 
    }
 
 
    public String postMapData(String url, Map<String, Object> params) {
 
        //cs
        HttpHeaders headers = new HttpHeaders();
        headers.add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36");
        headers.setContentType(MediaType.APPLICATION_JSON);
        headers.add("Accept", MediaType.APPLICATION_JSON.toString());
 
        HttpEntity<Map<String, Object>> requestEntity =
                new HttpEntity<>(params, headers);
 
        String body1 = restTemplate.postForEntity(url, requestEntity, String.class).getBody();
 
 
        System.out.println(body1);
 
        return body1;
 
    }
 
    public String postMapData(String url, Map<String, Object> params, Map<String, String> header) {
 
        //cs
        HttpHeaders headers = new HttpHeaders();
        headers.add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36");
        headers.setContentType(MediaType.APPLICATION_JSON);
        headers.add("Accept", MediaType.APPLICATION_JSON.toString());
 
        for (Map.Entry<String, String> entry : header.entrySet()) {
            headers.add(entry.getKey(), entry.getValue());
        }
 
        HttpEntity<Map<String, Object>> requestEntity =
                new HttpEntity<>(params, headers);
 
        String body1 = restTemplate.postForEntity(url, requestEntity, String.class).getBody();
 
 
        System.out.println(body1);
 
        return body1;
 
    }
 
 
    public String postStringData(String url, String data) {
 
        //cs
        HttpHeaders headers = new HttpHeaders();
        headers.add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36");
        headers.setContentType(MediaType.APPLICATION_JSON);
        headers.add("Accept", MediaType.APPLICATION_JSON.toString());
 
        HttpEntity<String> requestEntity =
                new HttpEntity<>(data, headers);
 
        String body1 = restTemplate.postForEntity(url, requestEntity, String.class).getBody();
 
 
        System.out.println(body1);
 
        return body1;
 
    }
 
    public byte[] postJsonDataAndReturnBuffer(String url, JSONObject param) {
        HttpHeaders headers = new HttpHeaders();
 
        headers.setContentType(MediaType.APPLICATION_JSON_UTF8);
 
        headers.add("Accept", MediaType.APPLICATION_JSON.toString());
 
        HttpEntity<JSONObject> requestEntity = new HttpEntity<JSONObject>(param, headers);
 
        return restTemplate.postForEntity(url, param, byte[].class).getBody();
    }
 
 
    public byte[] getBuffer(String url) {
        return restTemplate.getForEntity(url, byte[].class).getBody();
    }
 
    /**
     * post——from-urlencoded格式请求
     */
    public String postFromUrlencoded(String url, MultiValueMap<String, Object> params, Map<String, String> header) {
        HttpHeaders headers = new HttpHeaders();
        headers.add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.99 Safari/537.36");
        headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
 
        if (CollUtil.isNotEmpty(header)) {
            for (Map.Entry<String, String> entry : header.entrySet()) {
                headers.add(entry.getKey(), entry.getValue());
            }
        }
 
        HttpEntity<MultiValueMap<String, Object>> requestEntity =
                new HttpEntity<>(params, headers);
 
//        SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
//        //30s
//        requestFactory.setConnectTimeout(30*1000);
//        requestFactory.setReadTimeout(30*1000);
//        restTemplate = new RestTemplate(requestFactory);
        return restTemplate.postForEntity(url, requestEntity, String.class).getBody();
    }
}