HttpClient 4.5.2-(八)保持会话

2017-07-22  本文已影响0人  大漠知秋

紧接上一节,本节记录 【保持会话】
  平常网上冲浪是时,我们一般会先登陆自己的账号,比如淘宝,登陆账号以后加购物车,提交订单,付钱,等待发货,这中间都是依靠会话的保持才能够得以进行,平常我们使用浏览器帮助我们保持会话,那么用HttpClient怎么保持会话?


  • 环境:
  1. 自己搭建一个简单的web服务器
  2. 几个简单功能
  • /auth/*:Fliter,拦截没有登陆的会话,返回No Login!!!
  • /login:登陆Servlet,登陆完毕之后Session会保存登陆的凭证
  • /auth/isLogin:被拦截器拦截的Servlet,需要登录才能访问
  • /auth/logout:被拦截器拦截的Servlet,退出登录
首先测试没有进行保持回话的HttpClient操作:
访问
结果
把上面的注释打开,进行保持回话的HttpClient操作:
访问
结果

只要我们在每次的访问中都把客户端上下文给携带上,就可以在所有的访问中保持会话

完整代码:
package com.lynchj.utils.http;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.InterruptedIOException;
import java.io.UnsupportedEncodingException;
import java.net.SocketTimeoutException;
import java.net.URISyntaxException;
import java.net.UnknownHostException;
import java.security.KeyManagementException;
import java.security.KeyStore;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLException;
import javax.net.ssl.SSLHandshakeException;
import javax.net.ssl.TrustManager;
import javax.net.ssl.X509TrustManager;

import org.apache.http.HttpEntity;
import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.HttpRequest;
import org.apache.http.NameValuePair;
import org.apache.http.NoHttpResponseException;
import org.apache.http.ParseException;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.CookieStore;
import org.apache.http.client.HttpRequestRetryHandler;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
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.protocol.HttpClientContext;
import org.apache.http.client.utils.URIBuilder;
import org.apache.http.config.Registry;
import org.apache.http.config.RegistryBuilder;
import org.apache.http.conn.ConnectTimeoutException;
import org.apache.http.conn.ConnectionPoolTimeoutException;
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.conn.ssl.TrustSelfSignedStrategy;
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.impl.conn.PoolingHttpClientConnectionManager;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.protocol.HttpContext;
import org.apache.http.ssl.SSLContexts;
import org.apache.http.ssl.SSLInitializationException;
import org.apache.http.util.EntityUtils;

import com.alibaba.fastjson.JSONObject;

/**
 * 基于HttpClient实现的Http请求工具
 * 
 * @author 大漠知秋
 * @description 支持POST和GET请求,支持SSL
 * @description HttpClient 4.5.2
 * @description fastjson 1.2.31
 * 
 * <dependency>
 *  <groupId>org.apache.httpcomponents</groupId>
 *  <artifactId>httpclient</artifactId>
 *  <version>4.5.2</version>
 * </dependency>
 * 
 * <dependency>
 *  <groupId>com.alibaba</groupId>
 *  <artifactId>fastjson</artifactId>
 *  <version>1.2.31</version>
 * </dependency>
 */
public class HttpRequestUtils {

    /** 连接池 */
    private static PoolingHttpClientConnectionManager connManager;
    
    /** 编码 */
    private static final String ENCODING = "UTF-8";
    
    /** 出错返回结果 */
    private static final String RESULT = "-1";
    
    /**
     * 初始化连接池管理器,配置SSL
     */
    static {
        if (connManager == null) {
            
            try {
                // 创建ssl安全访问连接
                // 获取创建ssl上下文对象
                /**
                 * 使用带证书的定制SSL访问
                 */
                File authFile = new File("C:/Users/lynch/Desktop/my.keystore");
                SSLContext sslContext = getSSLContext(false, authFile, "mypassword");
                
                // 注册
                Registry<ConnectionSocketFactory> registry = RegistryBuilder.<ConnectionSocketFactory>create()
                    .register("http", PlainConnectionSocketFactory.INSTANCE)
                    .register("https", new SSLConnectionSocketFactory(sslContext))
                    .build();
                
                // ssl注册到连接池
                connManager = new PoolingHttpClientConnectionManager(registry);
                connManager.setMaxTotal(1000);  // 连接池最大连接数
                connManager.setDefaultMaxPerRoute(20);  // 每个路由最大连接数
                
            } catch (SSLInitializationException e) {
                e.printStackTrace();
            } catch (KeyManagementException e) {
                e.printStackTrace();
            } catch (NoSuchAlgorithmException e) {
                e.printStackTrace();
            } catch (KeyStoreException e) {
                e.printStackTrace();
            } catch (CertificateException e) {
                e.printStackTrace();
            } catch (FileNotFoundException e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
            }
            
        }
    }
    
    /**
     * 获取客户端连接对象
     * 
     * @param timeOut 超时时间
     * @return
     */
    private static CloseableHttpClient getHttpClient(Integer timeOut) {

        // 配置请求参数
        RequestConfig requestConfig = RequestConfig.custom()
                .setConnectionRequestTimeout(timeOut).
                setConnectTimeout(timeOut).
                setSocketTimeout(timeOut).
                build();
        // 配置超时回调机制
        HttpRequestRetryHandler retryHandler = new HttpRequestRetryHandler() {
            public boolean retryRequest(IOException exception, int executionCount, HttpContext context) {
                if (executionCount >= 3) {// 如果已经重试了3次,就放弃
                    return false;
                }
                if (exception instanceof NoHttpResponseException) {// 如果服务器丢掉了连接,那么就重试
                    return true;
                }
                if (exception instanceof SSLHandshakeException) {// 不要重试SSL握手异常
                    return false;
                }
                if (exception instanceof InterruptedIOException) {// 超时
                    return true;
                }
                if (exception instanceof UnknownHostException) {// 目标服务器不可达
                    return false;
                }
                if (exception instanceof ConnectTimeoutException) {// 连接被拒绝
                    return false;
                }
                if (exception instanceof SSLException) {// ssl握手异常
                    return false;
                }
                HttpClientContext clientContext = HttpClientContext.adapt(context);
                HttpRequest request = clientContext.getRequest();
                // 如果请求是幂等的,就再次尝试
                if (!(request instanceof HttpEntityEnclosingRequest)) {
                    return true;
                }
                return false;
            }
        };
        
        CloseableHttpClient httpClient = HttpClients.custom()
                .setConnectionManager(connManager)
                .setDefaultRequestConfig(requestConfig)
                .setRetryHandler(retryHandler)
                .build();
        
        return httpClient;
        
    }
    
    /**
     * 获取SSL上下文对象,用来构建SSL Socket连接
     * 
     * @param isDeceive 是否绕过SSL
     * @param creFile 整数文件,isDeceive为true 可传null
     * @param crePwd 整数密码,isDeceive为true 可传null, 空字符为没有密码
     * @return SSL上下文对象
     * @throws KeyManagementException 
     * @throws NoSuchAlgorithmException 
     * @throws KeyStoreException 
     * @throws IOException 
     * @throws FileNotFoundException 
     * @throws CertificateException 
     */
    private static SSLContext getSSLContext(boolean isDeceive, File creFile, String crePwd) throws KeyManagementException, NoSuchAlgorithmException, KeyStoreException, CertificateException, FileNotFoundException, IOException {

        SSLContext sslContext = null;
        
        if (isDeceive) {
            sslContext = SSLContext.getInstance("SSLv3");
            // 实现一个X509TrustManager接口,用于绕过验证,不用修改里面的方法
            X509TrustManager x509TrustManager = new X509TrustManager() {
                @Override
                public X509Certificate[] getAcceptedIssuers() {
                    return null;
                }
                @Override
                public void checkServerTrusted(X509Certificate[] chain, String authType) throws CertificateException {
                }
                @Override
                public void checkClientTrusted(X509Certificate[] chain, String authType) throws CertificateException {
                }
            };
            sslContext.init(null, new TrustManager[] {x509TrustManager}, null);
        } else {
            if (null != creFile && creFile.length() > 0) {
                if (null != crePwd) {
                    KeyStore keyStore = KeyStore.getInstance(KeyStore.getDefaultType());
                    keyStore.load(new FileInputStream(creFile), crePwd.toCharArray());
                    sslContext = SSLContexts.custom().loadTrustMaterial(keyStore, new TrustSelfSignedStrategy()).build();
                } else {
                    throw new SSLHandshakeException("整数密码为空");
                }
            }
        }
        
        return sslContext;
        
    }

    /**
     * post请求,支持SSL
     * 
     * @param url 请求地址
     * @param headers 请求头信息
     * @param params 请求参数
     * @param timeOut 超时时间(毫秒):从连接池获取连接的时间,请求时间,响应时间
     * @param isStream 是否以流的方式获取响应信息
     * @param clientContext Http请求客户端上下文对象,包含Cookie
     * @return 响应信息
     * @throws UnsupportedEncodingException 
     */
    public static String httpPost(String url, Map<String, Object> headers, Map<String, Object> params, Integer timeOut, boolean isStream, HttpClientContext clientContext) throws UnsupportedEncodingException {
        
        // 创建post请求
        HttpPost httpPost = new HttpPost(url);
        
        // 添加请求头信息
        if (null != headers) {
            for (Map.Entry<String, Object> entry : headers.entrySet()) {
                httpPost.addHeader(entry.getKey(), entry.getValue().toString());
            }
        }
        
        // 添加请求参数信息
        if (null != params) {
            httpPost.setEntity(new UrlEncodedFormEntity(covertParams2NVPS(params), ENCODING));
        }
        
        return getResult(httpPost, timeOut, isStream, clientContext);
        
    }
    
    /**
     * post请求,支持SSL
     * 
     * @param url 请求地址
     * @param params 请求参数
     * @param timeOut 超时时间(毫秒):从连接池获取连接的时间,请求时间,响应时间
     * @param clientContext Http请求客户端上下文对象,包含Cookie
     * @return 响应信息
     * @throws UnsupportedEncodingException 
     */
    public static String httpPost(String url, Map<String, Object> params, Integer timeOut, HttpClientContext clientContext) throws UnsupportedEncodingException {
        
        // 创建post请求
        HttpPost httpPost = new HttpPost(url);
        
        // 添加请求参数信息
        if (null != params) {
            httpPost.setEntity(new UrlEncodedFormEntity(covertParams2NVPS(params), ENCODING));
        }
        
        return getResult(httpPost, timeOut, true, clientContext);
        
    }
    
    /**
     * post请求,支持SSL
     * 
     * @param url 请求地址
     * @param headers 请求头信息
     * @param params 请求参数
     * @param timeOut 超时时间(毫秒):从连接池获取连接的时间,请求时间,响应时间
     * @param isStream 是否以流的方式获取响应信息
     * @param clientContext Http请求客户端上下文对象,包含Cookie
     * @return 响应信息
     * @throws UnsupportedEncodingException 
     */
    public static String httpPost(String url, JSONObject headers, JSONObject params, Integer timeOut, boolean isStream, HttpClientContext clientContext) throws UnsupportedEncodingException {
        
        // 创建post请求
        HttpPost httpPost = new HttpPost(url);
        
        // 添加请求头信息
        if (null != headers) {
            for (Map.Entry<String, Object> entry : headers.entrySet()) {
                httpPost.addHeader(entry.getKey(), entry.getValue().toString());
            }
        }
        
        // 添加请求参数信息
        if (null != params) {
            httpPost.setEntity(new UrlEncodedFormEntity(covertParams2NVPS(params), ENCODING));
        }
        
        return getResult(httpPost, timeOut, isStream, clientContext);
        
    }
    
    /**
     * post请求,支持SSL
     * 
     * @param url 请求地址
     * @param params 请求参数
     * @param timeOut 超时时间(毫秒):从连接池获取连接的时间,请求时间,响应时间
     * @param clientContext Http请求客户端上下文对象,包含Cookie
     * @return 响应信息
     * @throws UnsupportedEncodingException 
     */
    public static String httpPost(String url, JSONObject params, Integer timeOut, HttpClientContext clientContext) throws UnsupportedEncodingException {
        
        // 创建post请求
        HttpPost httpPost = new HttpPost(url);
        
        // 添加请求参数信息
        if (null != params) {
            httpPost.setEntity(new UrlEncodedFormEntity(covertParams2NVPS(params), ENCODING));
        }
        
        return getResult(httpPost, timeOut, true, clientContext);
        
    }
    
    /**
     * get请求,支持SSL
     * 
     * @param url 请求地址
     * @param headers 请求头信息
     * @param params 请求参数
     * @param timeOut 超时时间(毫秒):从连接池获取连接的时间,请求时间,响应时间
     * @param isStream 是否以流的方式获取响应信息
     * @param clientContext Http请求客户端上下文对象,包含Cookie
     * @return 响应信息
     * @throws URISyntaxException 
     */
    public static String httpGet(String url, Map<String, Object> headers, Map<String, Object> params, Integer timeOut, boolean isStream, HttpClientContext clientContext) throws URISyntaxException {
        
        // 构建url
        URIBuilder uriBuilder = new URIBuilder(url);
        // 添加请求参数信息
        if (null != params) {
            uriBuilder.setParameters(covertParams2NVPS(params));
        }
        
        // 创建post请求
        HttpGet httpGet = new HttpGet(url);
        
        // 添加请求头信息
        if (null != headers) {
            for (Map.Entry<String, Object> entry : headers.entrySet()) {
                httpGet.addHeader(entry.getKey(), entry.getValue().toString());
            }
        }
        
        return getResult(httpGet, timeOut, isStream, clientContext);
        
    }
    
    /**
     * get请求,支持SSL
     * 
     * @param url 请求地址
     * @param params 请求参数
     * @param timeOut 超时时间(毫秒):从连接池获取连接的时间,请求时间,响应时间
     * @param clientContext Http请求客户端上下文对象,包含Cookie
     * @return 响应信息
     * @throws URISyntaxException 
     */
    public static String httpGet(String url, Map<String, Object> params, Integer timeOut, HttpClientContext clientContext) throws URISyntaxException {
        
        // 构建url
        URIBuilder uriBuilder = new URIBuilder(url);
        // 添加请求参数信息
        if (null != params) {
            uriBuilder.setParameters(covertParams2NVPS(params));
        }
        
        // 创建post请求
        HttpGet httpGet = new HttpGet(url);
        
        return getResult(httpGet, timeOut, true, clientContext);
        
    }
    
    /**
     * get请求,支持SSL
     * 
     * @param url 请求地址
     * @param headers 请求头信息
     * @param params 请求参数
     * @param timeOut 超时时间(毫秒):从连接池获取连接的时间,请求时间,响应时间
     * @param isStream 是否以流的方式获取响应信息
     * @param clientContext Http请求客户端上下文对象,包含Cookie
     * @return 响应信息
     * @throws URISyntaxException 
     */
    public static String httpGet(String url, JSONObject headers, JSONObject params, Integer timeOut, boolean isStream, HttpClientContext clientContext) throws URISyntaxException {
        
        // 构建url
        URIBuilder uriBuilder = new URIBuilder(url);
        // 添加请求参数信息
        if (null != params) {
            uriBuilder.setParameters(covertParams2NVPS(params));
        }
        
        // 创建post请求
        HttpGet httpGet = new HttpGet(url);
        
        // 添加请求头信息
        if (null != headers) {
            for (Map.Entry<String, Object> entry : headers.entrySet()) {
                httpGet.addHeader(entry.getKey(), entry.getValue().toString());
            }
        }
        
        return getResult(httpGet, timeOut, isStream, clientContext);
        
    }
    
    /**
     * get请求,支持SSL
     * 
     * @param url 请求地址
     * @param params 请求参数
     * @param timeOut 超时时间(毫秒):从连接池获取连接的时间,请求时间,响应时间
     * @param clientContext Http请求客户端上下文对象,包含Cookie
     * @return 响应信息
     * @throws URISyntaxException 
     */
    public static String httpGet(String url, JSONObject params, Integer timeOut, HttpClientContext clientContext) throws URISyntaxException {
        
        // 构建url
        URIBuilder uriBuilder = new URIBuilder(url);
        // 添加请求参数信息
        if (null != params) {
            uriBuilder.setParameters(covertParams2NVPS(params));
        }
        
        // 创建post请求
        HttpGet httpGet = new HttpGet(url);
        
        return getResult(httpGet, timeOut, true, clientContext);
        
    }
    
    private static String getResult(HttpRequestBase httpRequest, Integer timeOut, boolean isStream, HttpClientContext clientContext) {
        
        // 响应结果
        StringBuilder sb = null;
        
        CloseableHttpResponse response = null;
        
        try {
            // 获取连接客户端
            CloseableHttpClient httpClient = getHttpClient(timeOut);
            // 发起请求
            if (null != clientContext) {
                response = httpClient.execute(httpRequest, clientContext);
            } else {
                response = httpClient.execute(httpRequest);
            }
            
            int respCode = response.getStatusLine().getStatusCode();
            // 如果是重定向
            if (302 == respCode) {
                String locationUrl = response.getLastHeader("Location").getValue();
                return getResult(new HttpPost(locationUrl), timeOut, isStream, clientContext);
            }
            // 正确响应
            if (200 == respCode) {
                // 获得响应实体
                HttpEntity entity = response.getEntity();
                sb = new StringBuilder();
                
                // 如果是以流的形式获取
                if (isStream) {
                    BufferedReader br = new BufferedReader(new InputStreamReader(entity.getContent(), ENCODING));
                    String len = "";
                    while ((len = br.readLine()) != null) {
                        sb.append(len);
                    }
                } else {
                    sb.append(EntityUtils.toString(entity, ENCODING));
                    if (sb.length() < 1) {
                        sb.append("-1");
                    }
                }
                
            }
        } catch (ConnectionPoolTimeoutException e) {
            System.err.println("从连接池获取连接超时!!!");
            e.printStackTrace();
        } catch (SocketTimeoutException e) {
            System.err.println("响应超时");
            e.printStackTrace();
        } catch (ConnectTimeoutException e) {
            System.err.println("请求超时");
            e.printStackTrace();
        } catch (ClientProtocolException e) {
            System.err.println("http协议错误");
            e.printStackTrace();
        } catch (UnsupportedEncodingException e) {
            System.err.println("不支持的字符编码");
            e.printStackTrace();
        } catch (UnsupportedOperationException e) {
            System.err.println("不支持的请求操作");
            e.printStackTrace();
        } catch (ParseException e) {
            System.err.println("解析错误");
            e.printStackTrace();
        } catch (IOException e) {
            System.err.println("IO错误");
            e.printStackTrace();
        } finally {
            if (null != response) {
                try {
                    response.close();
                } catch (IOException e) {
                    System.err.println("关闭响应连接出错");
                    e.printStackTrace();
                }
            }
            
        }
        
        return sb == null ? RESULT : ("".equals(sb.toString().trim()) ? "-1" : sb.toString());
        
    }

    /**
     * Map转换成NameValuePair List集合
     * 
     * @param params map
     * @return NameValuePair List集合
     */
    public static List<NameValuePair> covertParams2NVPS(Map<String, Object> params) {
        
        List<NameValuePair> paramList = new LinkedList<>();
        for (Map.Entry<String, Object> entry : params.entrySet()) {
            paramList.add(new BasicNameValuePair(entry.getKey(), entry.getValue().toString()));
        }
        
        return paramList;
        
    }
    
    public static void main(String[] args) throws Exception {
        
        HttpClientContext clientContext = HttpClientContext.create();
        CookieStore cookieStore = new BasicCookieStore();
        clientContext.setCookieStore(cookieStore);
        
        /** 登录 */
        System.out.println(httpPost("http://localhost/login", null, null, 6000, false, clientContext));
        /** 验证是否登录 */
        System.out.println(httpPost("http://localhost/auth/isLogin", null, null, 6000, false, clientContext));
        /** 退出登录 */
        System.out.println(httpPost("http://localhost/auth/logout", null, null, 6000, false, clientContext));
    
    }
    
}
上一篇下一篇

猜你喜欢

热点阅读