在使用 httpclient 做接口相关测试的过程中，遇到过一个障碍：如何处理多用户同时登陆。之前用户身份凭证一般都是做公参里面处理或者在 header 中单独定义一个或者几个字段，cookie 都是使用 httpclient 自带的管理器自动管理的。
后来用户凭证存到了 cookie 里，这里就有了一些障碍，经过翻看资料得到了解决。
设置如下：
/**
* 获取请求超时控制器
* <p>
* cookieSpec:即cookie策略。参数为cookiespecs的一些字段。作用：
* 1、如果网站header中有set-cookie字段时，采用默认方式可能会被cookie reject，无法写入cookie。将此属性设置成CookieSpecs.STANDARD_STRICT可避免此情况。
* 2、如果要想忽略cookie访问，则将此属性设置成CookieSpecs.IGNORE_COOKIES。
* </p>
*
* @return
*/
private static RequestConfig getRequestConfig() {
return RequestConfig.custom().setConnectionRequestTimeout(HttpClientConstant.CONNECT_REQUEST_TIMEOUT).setConnectTimeout(HttpClientConstant.CONNECT_TIMEOUT).setSocketTimeout(HttpClientConstant.SOCKET_TIMEOUT).setCookieSpec(CookieSpecs.IGNORE_COOKIES).setRedirectsEnabled(false).build();
}
这里说明一点，这个 requestconfig 既可以在对 HTTPrequestbase 进行设置，也可以对 CloseableHttpClient 进行设置，由于在各个项目中都采取了单独处理 cookie 的设置，所以我是直接放在了连接池 CloseableHttpClient 对象的设置里面，如下：
/**
* 通过连接池获取https协议请求对象
* <p>
* 此处会默认添加一天defaultcookiesstore，会处理响应头中的set-cookie字段
* 增加默认的请求控制器
* </p>
*
* @return
*/
private static CloseableHttpClient getCloseableHttpsClients() {
// 创建自定义的httpsclient对象
CloseableHttpClient client = HttpClients.custom().setConnectionManager(connManager).setRetryHandler(httpRequestRetryHandler).setDefaultRequestConfig(requestConfig).build();
// CloseableHttpClient client = HttpClients.createDefault();//非连接池创建
return client;
}
我的方案是在处理响应的时候，只用 closeablehttpresponse 对象接收响应的，然后在单独在 header 里面遍历 set-cookie 字段的值，在处理 json 对象作为返回体的时候添加进去，如下：
/**
* 响应结束之后，处理响应头信息，如set-cookien内容
*
* @param response 响应内容
* @return
*/
private static JSONObject afterResponse(CloseableHttpResponse response) {
JSONObject cookies = new JSONObject();
List<Header> headers = Arrays.asList(response.getHeaders("Set-Cookie"));
if (headers.size() == 0) return cookies;
headers.forEach(x -> {
String[] split = x.getValue().split(";")[0].split("=", 2);
cookies.put(split[0], split[1]);
});
return cookies;
}
/**
* 根据解析好的content，转化json对象
*
* @param content
* @return
*/
private static JSONObject getJsonResponse(String content, JSONObject cookies) {
JSONObject jsonObject = new JSONObject();
try {
jsonObject = JSONObject.fromObject(content);
} catch (Exception e) {
jsonObject.put("content", content);
jsonObject.put("code", TEST_ERROR_CODE);
logger.warn("响应体非json格式，已经自动转换成json格式！");
} finally {
if (!cookies.isEmpty()) jsonObject.put(HttpClientConstant.COOKIE, cookies);
return jsonObject;
}
}
在每个项目的 base 对象接收到响应之后存储 cookie 以便子类继承，在每次发送请求的时候带上当前对象的 cookie，以对象形式存储每一个用户，达到多用户同时登录的目的，如下：
@Override
public JSONObject getResponse(HttpRequestBase httpRequestBase) {
setHeaders(httpRequestBase);
JSONObject response = FanLibrary.getHttpResponse(httpRequestBase);
handleResponseHeader(response);
return response;
}
@Override
public void setHeaders(HttpRequestBase httpRequestBase) {
httpRequestBase.addHeader(Common.REQUEST_ID);
httpRequestBase.addHeader(FanLibrary.getHeader("token", token));
if (!cookies.isEmpty()) httpRequestBase.addHeader(FanLibrary.getCookies(cookies));
}
@Override
public void handleResponseHeader(JSONObject response) {
if (!response.containsKey(HttpClientConstant.COOKIE)) return;
cookies.putAll(response.getJSONObject(HttpClientConstant.COOKIE));
response.remove(HttpClientConstant.COOKIE);
}
