httpclient模拟访问第三方接口(登录后携带cookie进行接口访问)

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.NameValuePair;
import org.apache.http.client.CookieStore;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.cookie.Cookie;
import org.apache.http.impl.client.BasicCookieStore;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.util.EntityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * http请求工具类
 * 
 * @author aaa
 * @date 2019/07/23
 */
public class HttpUtil {

    private final static Logger logger = LoggerFactory.getLogger(HttpUtil.class);

    public static String sendPost(String loginUrl, String formUrl, Map loginParam,
        Map formParam) throws Exception {
        StringBuffer cookie = new StringBuffer();
        CookieStore cookieStore = new BasicCookieStore();
        CloseableHttpClient httpClient = HttpClients.custom().setDefaultCookieStore(cookieStore).build();
        // POST请求
        HttpPost httpPost = new HttpPost(loginUrl);
        // 构造请求参数
        List list = new ArrayList();
        for (Map.Entry entry : loginParam.entrySet()) {
            list.add(new BasicNameValuePair(entry.getKey(), entry.getValue()));
        }
        if (list.size() > 0) {
            UrlEncodedFormEntity entity = new UrlEncodedFormEntity(list, "utf-8");
            httpPost.setEntity(entity);
        }
        // 设置请求头
        httpPost.addHeader("Accept", "*/*");
        httpPost.addHeader("Accept-Language", "zh-CN,zh;q=0.8");
        httpPost.addHeader("Connection", "keep-alive");
        httpPost.addHeader("User-Agent",
            "Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36");

        Map headParamsMap = new HashMap();
        headParamsMap.put("Host", "");
        headParamsMap.put("Referer", "");
        Set> entrySet = headParamsMap.entrySet();
        for (Entry entry : entrySet) {
            httpPost.addHeader(entry.getKey(), entry.getValue());
        }
        HttpResponse response = httpClient.execute(httpPost);
        String result = "";
        if (response != null) {
            int statusCode = response.getStatusLine().getStatusCode();
            // result = EntityUtils.toString(response.getEntity(), "utf-8");
            if (statusCode == HttpStatus.SC_OK) {
                // 获得Cookies
                List cookies = cookieStore.getCookies();
                for (Cookie c : cookies) {
                    cookie.append(c.getName()).append("=").append(c.getValue()).append(";");
                }
                result = sendPostByCookie(formUrl, cookieStore, formParam);
            }
        }
        return result;
    }

    public static String sendPostByCookie(String url, CookieStore cookieStore, Map param)
        throws Exception {
        CloseableHttpClient httpClient = HttpClients.custom().setDefaultCookieStore(cookieStore).build();
        // POST请求
        HttpPost httpPost = new HttpPost(url);
        // 构造请求参数
        List list = new ArrayList();
        for (Map.Entry entry : param.entrySet()) {
            list.add(new BasicNameValuePair(entry.getKey(), entry.getValue()));
        }
        if (list.size() > 0) {
            UrlEncodedFormEntity entity = new UrlEncodedFormEntity(list, "utf-8");
            httpPost.setEntity(entity);
        }
        // 设置请求头
        httpPost.addHeader("Accept", "*/*");
        httpPost.addHeader("Accept-Language", "zh-CN,zh;q=0.8");
        httpPost.addHeader("Connection", "keep-alive");
        httpPost.addHeader("User-Agent",
            "Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36");

        Map headParamsMap = new HashMap();
        headParamsMap.put("Host", "");
        headParamsMap.put("Referer", "");

        Set> entrySet = headParamsMap.entrySet();
        for (Entry entry : entrySet) {
            httpPost.addHeader(entry.getKey(), entry.getValue());
        }
        HttpResponse response = httpClient.execute(httpPost);
        String result = "";
        if (response != null) {
            int statusCode = response.getStatusLine().getStatusCode();
            result = EntityUtils.toString(response.getEntity(), "utf-8");
            logger.info("\n返回码:" + statusCode + "\n返内容:" + result);
        }
        return result;
    }
}
 

你可能感兴趣的:(Java)