序列化CookieManager/CookieStore

Question

我正在编写一个访问网站的Java程序，它依赖于分发cookie以便我可以登录。我发现使用CookieManager将允许我的URLConnection至少收集这些cookies，但是当java程序关闭时，我如何让这些cookies保持不变呢？

我已经有了一个管理一些变量的序列化的类，那么我如何在这个序列化过程中实现cookie呢？默认情况下，Java显然不允许我序列化CookieManagers和CookieStores。

Answer 1

您可以扩展CookieManager并使其实现可序列化。然后使用您刚刚创建的扩展CookieManager。

例如：

public class NewCookieManager extends CookieManager implements Serializable{
  //Code here...
}

Answer 2

您可以使用cookie管理器将所有cookie标头保存在一个可序列化的映射中。您可以调用put方法使其再次解析Cookie头。

public class KeepHeadersCookieManager extends CookieManager {

    private Map<URI, Map<String, List<String>>> cookieHeaders = new HashMap<>();

    public KeepHeadersCookieManager() {
    }

    public KeepHeadersCookieManager(Map<URI, Map<String, List<String>>> cookieHeaders) throws IOException {
        for (URI u : cookieHeaders.keySet()) {
            put(u, cookieHeaders.get(u));
        }
    }

    @Override
    public void put(URI uri, Map<String, List<String>> responseHeaders) throws IOException {
        super.put(uri, responseHeaders);

        Map<String, List<String>> uriCookiesHeaders = cookieHeaders.computeIfAbsent(uri, u -> new HashMap<>());
        List<String> uriSetCookieHeaders = uriCookiesHeaders.computeIfAbsent("Set-Cookie", h -> new ArrayList<>());
        List<String> uriSetCookie2Headers = uriCookiesHeaders.computeIfAbsent("Set-Cookie2", h -> new ArrayList<>());
        List<String> setCookieHeaders = responseHeaders.getOrDefault("Set-Cookie", new ArrayList<>());
        List<String> setCookie2Headers = responseHeaders.getOrDefault("Set-Cookie2", new ArrayList<>());
        uriSetCookieHeaders.addAll(setCookieHeaders);
        uriSetCookie2Headers.addAll(setCookie2Headers);
    }

    public Map<URI, Map<String, List<String>>> getCookieHeaders() {
        return cookieHeaders;
    }
}

这样，您就可以调用getCookieHeaders()来获取所有cookie标头的可序列化映射，并将其传递回构造函数以从先前的状态初始化cookie管理器。