使用Volley获取多个cookie

问题描述

最近接手了一个论坛性质的应用，因为流量小，网络请求频繁，所以选择使用Volley进行重构。因为默认Volley只是读取一个cookie，而这个项目使用了4个cookie，所以用户登录部分先使用原来的HttpClient来保存用户的登录cookie信息。

重构获取和发送内容部分重构基本能够顺利完成，只剩下cookie这个难题，网上找了很多资料，发现很多都是片面一讲，其实说的不对。

网上找到的方法（不能解决问题）

有时服务器会返回多个 Set-Cookie 值，而Volley默认只取第一个，如果有需求，就要自己修改Volley的代码啦，HurlStack（sdk_int>9会走这里)里面的performRequest，默认解析header方式如下，多个值的话只需要拿到header.getValue().get(1)等等，具体需求可以自行修改。

Header h = new BasicHeader(header.getKey(), header.getValue().get(0));
response.addHeader(h);

上面是网上搜到的方法，其实只是说了第一步，后面还有需要注意的地方。但是解决问题的思路是对的：

1. 重写或者是继承一个

HurlStack

，在方法

performRequest（）

中把cookie添加到response。

2. 定义一个model包含请求返回的内容和cookie list。

3. 重写一个返回对象是上面我们model的request，在request的

parseNetworkResponse

方法中，把获取到的cookie放到model里面。

这里需要额外注意一点，也是上面方法没有说明的一点：

在我们的

HurlStack

中

performRequest

返回的是一个

HttpResponse

(header是一个list)，我们的cookie也是添加到了它的headers里面。到目前还没有问题。但是此处返回的response还要经过一个转化，因为

Request

类的

protected abstract Response<T> parseNetworkResponse(NetworkResponse paramNetworkResponse);

参数为一个

NetworkResponse

，这个的headers是一个

Map<String, String>

，所以这就要求我们的多个cookie的key不能都为

Set-Cookie

。因为上面的代码中cookie的

header.getKey()

一般是

Set-Cookie

。

下面是部分源码：

HurlStack.class

public HttpResponse performRequest(Request<?> request, Map<String, String> additionalHeaders)
throws IOException, AuthFailureError
{

...
BasicHttpResponse response = new BasicHttpResponse(responseStatus);
response.setEntity(entityFromConnection(connection));
for (Map.Entry<String, List<String>> header : connection.getHeaderFields().entrySet()) {
if (header.getKey() != null)
{
Header h = new BasicHeader((String)header.getKey(), (String)((List)header.getValue()).get(0));
response.addHeader(h);
}
}
return response;
}

BasicNetwork.class

private static Map<String, String> convertHeaders(Header[] headers)
{
Map<String, String> result = new HashMap();
for (int i = 0; i < headers.length; i++) {
result.put(headers[i].getName(), headers[i].getValue());
}
return result;
}

我的解决办法

关键代码：

MyHurlStack.class

public HttpResponse performRequest(Request<?> request, Map<String, String> additionalHeaders)
throws IOException, AuthFailureError {
String url = request.getUrl();
HashMap map = new HashMap();
map.putAll(request.getHeaders());
map.putAll(additionalHeaders);
if (this.mUrlRewriter != null) {
String parsedUrl = this.mUrlRewriter.rewriteUrl(url);
if (parsedUrl == null) {
throw new IOException("URL blocked by rewriter: " + url);
}

url = parsedUrl;
}

URL parsedUrl1 = new URL(url);
HttpURLConnection connection = this.openConnection(parsedUrl1, request);
Iterator responseCode = map.keySet().iterator();

while (responseCode.hasNext()) {
String protocolVersion = (String) responseCode.next();
connection.addRequestProperty(protocolVersion, (String) map.get(protocolVersion));
}

setConnectionParametersForRequest(connection, request);
ProtocolVersion protocolVersion1 = new ProtocolVersion("HTTP", 1, 1);
int responseCode1 = connection.getResponseCode();
if (responseCode1 == -1) {
throw new IOException("Could not retrieve response code from HttpUrlConnection.");
} else {
BasicStatusLine responseStatus =
new BasicStatusLine(protocolVersion1, connection.getResponseCode(),
connection.getResponseMessage());
BasicHttpResponse response = new BasicHttpResponse(responseStatus);
response.setEntity(entityFromConnection(connection));

for (Object o : connection.getHeaderFields().entrySet()) {
Entry header = (Entry) o;
if (header.getKey() != null) {
// 如果是cookie则保存为name = Set-Cookie + 序号，保证key不同
if (header.getKey().equals("Set-Cookie")) {
List headerList = (List) header.getValue();
for (int i = 0; i < headerList.size(); i++) {
BasicHeader h =
new BasicHeader((String) header.getKey() + i, (String) headerList.get(i));
response.addHeader(h);
}
} else {
List headerList = (List) header.getValue();
for (Object aHeaderList : headerList) {
BasicHeader h = new BasicHeader((String) header.getKey(), (String) aHeaderList);
response.addHeader(h);
}
}
}
}

return response;
}
}

MyRequest.class

//使用正则表达式从reponse的头中提取cookie内容的子串
private static final Pattern pattern = Pattern.compile("Set-Cookie\\d*");

@Override
protected Response<MyResponse> parseNetworkResponse(NetworkResponse networkResponse) {
try {
MyResponse response = new MyResponse();
String jsonString =
new String(networkResponse.data, HttpHeaderParser.parseCharset(networkResponse.headers));

response.setJsonString(jsonString);
// get all cookie
List<Cookie> cookies = new ArrayList<>();
for (String key : networkResponse.headers.keySet()) {
Matcher matcher = pattern.matcher(key);
if (matcher.find()) {
cookieFromResponse = networkResponse.headers.get(key);
cookieFromResponse = cookieFromResponse.substring(0, cookieFromResponse.indexOf(";"));
String keyValue[] = cookieFromResponse.split("=", -1);
Cookie cookie = new BasicClientCookie(keyValue[0], keyValue[1]);
cookies.add(cookie);
}
}

response.setCookies(cookies);

return Response.success(response, HttpHeaderParser.parseCacheHeaders(networkResponse));
} catch (UnsupportedEncodingException e) {
return Response.error(new ParseError(e));
}
}

MyResponse.class

public class MyResponse{

private static String TAG = "MyResponse";

private int status = 0;

private String jsonString = "";

private List<Cookie> cookies = new ArrayList<>();

public int getStatus() {
return status;
}

public void setStatus(int status) {
this.status = status;
}

public String getJsonString() {
return jsonString;
}

public void setJsonString(String jsonString) {
this.jsonString = jsonString;
}

public List<Cookie> getCookies() {
return cookies;
}

public void setCookies(List<Cookie> cookies) {
this.cookies = cookies;
}
}

VolleyHelper.class

public static void init(Context context) {
queue = Volley.newRequestQueue(context, new MyHurlStack());
}

后记

当初按照网上的方法只是改了HurlStack里面的部分内容，最后只能得到一个cookie，这里一般都会想到是不是Map的原因。最后查找源码发现真的是这个原因，然后使用了一个笨方法来解决了问题。所以根据源码来分析解决问题还是很需要的。

这个方法只是根据我们目前的情况解决了问题，应该是不能解决所有的情况，这里只是提供一个解决的思路，如果你有更好的解决方法，请留言，一起讨论一下。