我正在尝试通过线程建立多个连接。
但是每个连接似乎都会覆盖其他连接,导致使用错误的cookie连接。
在线程类的构造函数中:
manager = new CookieManager();
manager.setCookiePolicy(CookiePolicy.ACCEPT_ALL);
CookieHandler.setDefault(manager);
以任何方式管理每个线程或每个类的cookie?
新尝试失败:
现在每个线程都使用它自己的索引,但它们似乎仍然以cookie方式覆盖彼此。有什么想法吗?
public class threadedCookieStore implements CookieStore, Runnable {
CookieStore[] store = new CookieStore[1000];
int index;
public threadedCookieStore(int new_index) {
index = new_index;
// get the default in memory cookie store
store[index] = new CookieManager().getCookieStore();
// todo: read in cookies from persistant storage
// and add them store
// add a shutdown hook to write out the in memory cookies
Runtime.getRuntime().addShutdownHook(new Thread(this));
}
public void run() {
// todo: write cookies in store to persistent storage
}
public void add(URI uri, HttpCookie cookie) {
store[index].add(uri, cookie);
}
public List<HttpCookie> get(URI uri) {
return store[index].get(uri);
}
public List<HttpCookie> getCookies() {
return store[index].getCookies();
}
public List<URI> getURIs() {
return store[index].getURIs();
}
public boolean remove(URI uri, HttpCookie cookie) {
return store[index].remove(uri, cookie);
}
public boolean removeAll() {
return store[index].removeAll();
}
}
在课堂上:
threadedCookieStore cookiestore = new threadedCookieStore(index);
manager = new CookieManager(cookiestore,CookiePolicy.ACCEPT_ALL);
manager.setCookiePolicy(CookiePolicy.ACCEPT_ALL);
CookieHandler.setDefault(manager);
答案 0 :(得分:10)
谢谢大家。
我赞成所有答案,但没有人有完整的解决方案。
由于google'ing这个问题导致这个页面出现在这里,我将发布完整的解决方案并接受我自己的答案:
方法文档:
CookieHandler
扩展为SessionCookieManager
这基于How to use different cookies for each connection using HttpURLConnection and the CookieManager in Java,nivs正确描述,不提供完整的解决方案。所以大多数/所有的功劳归于他,我只是制作完整的HowTo。 SessionCookieManager基于Java
的源代码http://docs.oracle.com/javase/7/docs/api/java/net/CookieManager.html
import java.io.IOException;
import java.net.CookieHandler;
import java.net.CookiePolicy;
import java.net.CookieStore;
import java.net.HttpCookie;
import java.net.URI;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
public class SessionCookieManager extends CookieHandler
{
private CookiePolicy policyCallback;
public SessionCookieManager() {
this(null, null);
}
private final static SessionCookieManager ms_instance = new SessionCookieManager();
public static SessionCookieManager getInstance()
{
return ms_instance;
}
private final static ThreadLocal<CookieStore> ms_cookieJars = new ThreadLocal<CookieStore>() {
@Override
protected synchronized CookieStore initialValue() { return new InMemoryCookieStore(); }
};
public void clear()
{
getCookieStore().removeAll();
}
public SessionCookieManager(CookieStore store,
CookiePolicy cookiePolicy)
{
// use default cookie policy if not specify one
policyCallback = (cookiePolicy == null) ? CookiePolicy.ACCEPT_ALL //note that I changed it to ACCEPT_ALL
: cookiePolicy;
// if not specify CookieStore to use, use default one
}
public void setCookiePolicy(CookiePolicy cookiePolicy) {
if (cookiePolicy != null) policyCallback = cookiePolicy;
}
public CookieStore getCookieStore() {
return ms_cookieJars.get();
}
public Map<String, List<String>>
get(URI uri, Map<String, List<String>> requestHeaders)
throws IOException
{
// pre-condition check
if (uri == null || requestHeaders == null) {
throw new IllegalArgumentException("Argument is null");
}
Map<String, List<String>> cookieMap =
new java.util.HashMap<String, List<String>>();
// if there's no default CookieStore, no way for us to get any cookie
if (getCookieStore() == null)
return Collections.unmodifiableMap(cookieMap);
List<HttpCookie> cookies = new java.util.ArrayList<HttpCookie>();
for (HttpCookie cookie : getCookieStore().get(uri)) {
// apply path-matches rule (RFC 2965 sec. 3.3.4)
if (pathMatches(uri.getPath(), cookie.getPath())) {
cookies.add(cookie);
}
}
// apply sort rule (RFC 2965 sec. 3.3.4)
List<String> cookieHeader = sortByPath(cookies);
cookieMap.put("Cookie", cookieHeader);
return Collections.unmodifiableMap(cookieMap);
}
public void
put(URI uri, Map<String, List<String>> responseHeaders)
throws IOException
{
// pre-condition check
if (uri == null || responseHeaders == null) {
throw new IllegalArgumentException("Argument is null");
}
// if there's no default CookieStore, no need to remember any cookie
if (getCookieStore() == null)
return;
for (String headerKey : responseHeaders.keySet()) {
// RFC 2965 3.2.2, key must be 'Set-Cookie2'
// we also accept 'Set-Cookie' here for backward compatibility
if (headerKey == null
|| !(headerKey.equalsIgnoreCase("Set-Cookie2")
|| headerKey.equalsIgnoreCase("Set-Cookie")
)
)
{
continue;
}
for (String headerValue : responseHeaders.get(headerKey)) {
try {
List<HttpCookie> cookies = HttpCookie.parse(headerValue);
for (HttpCookie cookie : cookies) {
if (shouldAcceptInternal(uri, cookie)) {
getCookieStore().add(uri, cookie);
}
}
} catch (IllegalArgumentException e) {
// invalid set-cookie header string
// no-op
}
}
}
}
/* ---------------- Private operations -------------- */
// to determine whether or not accept this cookie
private boolean shouldAcceptInternal(URI uri, HttpCookie cookie) {
try {
return policyCallback.shouldAccept(uri, cookie);
} catch (Exception ignored) { // pretect against malicious callback
return false;
}
}
/*
* path-matches algorithm, as defined by RFC 2965
*/
private boolean pathMatches(String path, String pathToMatchWith) {
if (path == pathToMatchWith)
return true;
if (path == null || pathToMatchWith == null)
return false;
if (path.startsWith(pathToMatchWith))
return true;
return false;
}
/*
* sort cookies with respect to their path: those with more specific Path attributes
* precede those with less specific, as defined in RFC 2965 sec. 3.3.4
*/
private List<String> sortByPath(List<HttpCookie> cookies) {
Collections.sort(cookies, new CookiePathComparator());
List<String> cookieHeader = new java.util.ArrayList<String>();
for (HttpCookie cookie : cookies) {
// Netscape cookie spec and RFC 2965 have different format of Cookie
// header; RFC 2965 requires a leading $Version="1" string while Netscape
// does not.
// The workaround here is to add a $Version="1" string in advance
if (cookies.indexOf(cookie) == 0 && cookie.getVersion() > 0) {
cookieHeader.add("$Version=\"1\"");
}
cookieHeader.add(cookie.toString());
}
return cookieHeader;
}
static class CookiePathComparator implements Comparator<HttpCookie> {
public int compare(HttpCookie c1, HttpCookie c2) {
if (c1 == c2) return 0;
if (c1 == null) return -1;
if (c2 == null) return 1;
// path rule only applies to the cookies with same name
if (!c1.getName().equals(c2.getName())) return 0;
// those with more specific Path attributes precede those with less specific
if (c1.getPath().startsWith(c2.getPath()))
return -1;
else if (c2.getPath().startsWith(c1.getPath()))
return 1;
else
return 0;
}
}
}
请注意,在我的情况下,我将默认值CookiePolicy
更改为ACCEPT_ALL
CookieHandler.setDefault(SessionCookieManager.getInstance());
SessionCookieManager.getInstance().clear();
再次:不是我的想法,只是把它放在一起。所有赠送金额均归Java
和https://stackoverflow.com/users/1442259/nivs
答案 1 :(得分:6)
谢谢,我尝试使用你的答案,但它是基于旧版本的CookieManager(可能为什么你必须使用ACCEPT_ALL)并引用了包私有的InMemoryCookieStore,所以它激发了我最终的解决方案。以前应该对我们所有人都很明显:一个ThreadLocal CookieStore代理类。
CookieHandler.setDefault(new CookieManager(new ThreadLocalCookieStore(), null));
带
import java.net.CookieManager;
import java.net.CookieStore;
import java.net.HttpCookie;
import java.net.URI;
import java.util.List;
public class ThreadLocalCookieStore implements CookieStore {
private final static ThreadLocal<CookieStore> ms_cookieJars = new ThreadLocal<CookieStore>() {
@Override
protected synchronized CookieStore initialValue() {
return (new CookieManager()).getCookieStore(); /*InMemoryCookieStore*/
}
};
@Override
public void add(URI uri, HttpCookie cookie) {
ms_cookieJars.get().add(uri, cookie);
}
@Override
public List<HttpCookie> get(URI uri) {
return ms_cookieJars.get().get(uri);
}
@Override
public List<HttpCookie> getCookies() {
return ms_cookieJars.get().getCookies();
}
@Override
public List<URI> getURIs() {
return ms_cookieJars.get().getURIs();
}
@Override
public boolean remove(URI uri, HttpCookie cookie) {
return ms_cookieJars.get().remove(uri, cookie);
}
@Override
public boolean removeAll() {
return ms_cookieJars.get().removeAll();
}
}
似乎对我来说就像一个魅力
答案 2 :(得分:1)
您可以安装一个管理ThreadLocal CookieManager实例的CookieHandler。
答案 3 :(得分:1)
DavidBlackledge的ThreadLocal
CookieStore
是最好的方式。为了提高内存效率,我在这里提供了一个简单的常规CookieStore
实现,因此您不必为每个线程实例化整个CookieManager
(假设您不仅仅有一些)
/**
* @author lidor
* A simple implementation of CookieStore
*/
public class CookieJar implements CookieStore {
private Map<URI, List<HttpCookie>> jar;
private List<HttpCookie> freeCookies;
public CookieJar() {
jar = new HashMap<URI, List<HttpCookie>>();
freeCookies = new ArrayList<HttpCookie>();
}
@Override
public void add(URI uri, HttpCookie cookie) {
if (uri != null) {
if (!jar.containsKey(uri))
jar.put(uri, new ArrayList<HttpCookie>());
List<HttpCookie> cookies = jar.get(uri);
cookies.add(cookie);
} else {
freeCookies.add(cookie);
}
}
@Override
public List<HttpCookie> get(URI uri) {
Log.trace("CookieJar.get (" + this + ") called with URI " + uri + " (host=" + uri.getHost() + ")");
List<HttpCookie> liveCookies = new ArrayList<HttpCookie>();
if (jar.containsKey(uri)) {
for (HttpCookie cookie : jar.get(uri)) {
if (!cookie.hasExpired())
liveCookies.add(cookie);
}
}
for (HttpCookie cookie : getCookies()) {
if (cookie.getDomain().equals(uri.getHost()))
if (!liveCookies.contains(cookie))
liveCookies.add(cookie);
}
return Collections.unmodifiableList(liveCookies);
}
@Override
public List<HttpCookie> getCookies() {
List<HttpCookie> liveCookies = new ArrayList<HttpCookie>();
for (URI uri : jar.keySet())
for (HttpCookie cookie : jar.get(uri)) {
if (!cookie.hasExpired())
liveCookies.add(cookie);
}
for (HttpCookie cookie : freeCookies) {
if (!cookie.hasExpired())
liveCookies.add(cookie);
}
return Collections.unmodifiableList(liveCookies);
}
@Override
public List<URI> getURIs() {
return Collections.unmodifiableList(new ArrayList<URI>(jar.keySet()));
}
@Override
public boolean remove(URI uri, HttpCookie cookie) {
if (jar.containsKey(uri)) {
return jar.get(uri).remove(cookie);
} else {
return freeCookies.remove(cookie);
}
}
@Override
public boolean removeAll() {
boolean ret = (jar.size() > 0) || (freeCookies.size() > 0);
jar.clear();
freeCookies.clear();
return ret;
}
}
因此,如果你有这个CookieJar,那么你可以将ms_cookieJars
声明更改为:
private final static ThreadLocal<CookieStore> ms_cookieJars = new ThreadLocal<CookieStore>() {
@Override
protected synchronized CookieStore initialValue() {
return new CookieJar();
}
};
答案 4 :(得分:1)
根据此主题中的答案,我创建了另一个非常简单的ThreadLocalCookieStore
实现并将其推送到GitHub(同时将其作为Maven依赖项提供):
public class ThreadLocalCookieStore implements CookieStore {
private final static ThreadLocal<CookieStore> stores = new ThreadLocal<CookieStore>() {
@Override protected synchronized CookieStore initialValue() {
return (new CookieManager()).getCookieStore(); //InMemoryCookieStore
}
};
@Override public void add(URI uri, HttpCookie cookie) { getStore().add(uri,cookie); }
@Override public List<HttpCookie> get(URI uri) { return getStore().get(uri); }
@Override public List<HttpCookie> getCookies() { return getStore().getCookies(); }
@Override public List<URI> getURIs() { return getStore().getURIs(); }
@Override public boolean remove(URI uri, HttpCookie cookie) { return getStore().remove(uri,cookie); }
@Override public boolean removeAll() { return getStore().removeAll(); }
@Override public int hashCode() { return getStore().hashCode(); }
protected CookieStore getStore() { return stores.get(); }
public void purgeStore() { stores.remove(); }
}
如果没有太多代码,使用任何策略值设置cookie存储变得非常简单,例如:
CookieHandler.setDefault(new java.net.CookieManager(
new ThreadLocalCookieStore(), CookiePolicy.ACCEPT_ALL));
此外,依赖项还包含一个小型的@WebFilter
,用于在需要时在多个服务请求上分隔cookie存储。
答案 5 :(得分:0)
您可以为Cookie设置不同的路径。因此它不会被覆盖。
http://docs.oracle.com/javase/6/docs/api/java/net/HttpCookie.html#setPath%28java.lang.String%29
答案 6 :(得分:0)
ThreadLocal CookieManager怎么样?与其他一些答案相同,但似乎需要更少的代码:
public class ThreadLocalCookies extends CookieManager {
private static CookiePolicy s_policy = null;
private static ThreadLocal<CookieManager> s_impl =
new ThreadLocal<CookieManager>() {
@Override protected CookieManager initialValue() {
if (null == s_policy) {
throw new IllegalStateException("Call install() first");
}
return new CookieManager(null, s_policy);
}
};
public static void install() {
install(CookiePolicy.ACCEPT_ALL);
}
public static void install(CookiePolicy policy) {
s_policy = policy;
CookieHandler.setDefault(new ThreadLocalCookies());
}
public static void clear() {
s_impl.set(new CookieManager(null, s_policy));
}
@Override
public Map<String, List<String>>
get(URI uri, Map<String, List<String>> requestHeaders)
throws IOException {
return s_impl.get().get(uri, requestHeaders);
}
@Override
public void put(URI uri, Map<String,List<String>> responseHeaders)
throws IOException {
s_impl.get().put(uri, responseHeaders);
}
}