CookieHandler 使用Demo

[java]代码库

import java.io.IOException;
import java.net.CookieHandler;
import java.net.URI;
import java.net.URL;
import java.net.URLConnection;
import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.Map;

public class Fetch5 {
    public static void main (String args[]) throws Exception {
        String urlString = "java.sun.com";
        CookieHandler.setDefault (new ListCookieHandler() );
        URL url = new URL (urlString);
        URLConnection connection = url.openConnection();
        Object obj = connection.getContent();
        url = new URL (urlString);
        connection = url.openConnection();
        obj = connection.getContent();
    }
}

class ListCookieHandler extends CookieHandler {
    private List<Cookie> cookieJar = new LinkedList<Cookie>();

    public void put (URI uri, Map<String, List<String>> responseHeaders) throws IOException {
        List<String> setCookieList = responseHeaders.get ("Set-Cookie");
        if (setCookieList != null) {
for (String item : setCookieList) {
                Cookie cookie = new Cookie (uri, item);
for (Cookie existingCookie : cookieJar) {
                    if ( (cookie.getURI().equals (existingCookie.getURI() ) )
                    && (cookie.getName().equals (existingCookie.getName() ) ) ) {
                        cookieJar.remove (existingCookie);
                        break;
                    }
                }
                cookieJar.add (cookie);
            }
        }
    }

    public Map<String, List<String>> get (URI uri, Map<String, List<String>> requestHeaders)
    throws IOException {
        StringBuilder cookies = new StringBuilder();
for (Cookie cookie : cookieJar) {
            // Remove cookies that have expired
            if (cookie.hasExpired() ) {
                cookieJar.remove (cookie);
            } else if (cookie.matches (uri) ) {
                if (cookies.length() > 0) {
                    cookies.append (", ");
                }
                cookies.append (cookie.toString() );
            }
        }

        Map<String, List<String>> cookieMap = new HashMap<String, List<String>> (requestHeaders);

        if (cookies.length() > 0) {
            List<String> list = Collections.singletonList (cookies.toString() );
            cookieMap.put ("Cookie", list);
        }
        System.out.println ("CookieMap: " + cookieMap);
        return Collections.unmodifiableMap (cookieMap);
    }
}

class Cookie {

    String name;

    String value;

    URI uri;

    String domain;

    Date expires;

    String path;

    private static DateFormat expiresFormat1 = new SimpleDateFormat ("E, dd MMM yyyy k:m:s 'GMT'",
            Locale.US);

    private static DateFormat expiresFormat2 = new SimpleDateFormat ("E, dd-MMM-yyyy k:m:s 'GMT'",
            Locale.US);

    public Cookie (URI uri, String header) {
        String attributes[] = header.split (";");
        String nameValue = attributes[0].trim();
        this.uri = uri;
        this.name = nameValue.substring (0, nameValue.indexOf ('=') );
        this.value = nameValue.substring (nameValue.indexOf ('=') + 1);
        this.path = "/";
        this.domain = uri.getHost();

        for (int i = 1; i < attributes.length; i++) {
            nameValue = attributes[i].trim();
            int equals = nameValue.indexOf ('=');
            if (equals == -1) {
                continue;
            }
            String name = nameValue.substring (0, equals);
            String value = nameValue.substring (equals + 1);
            if (name.equalsIgnoreCase ("domain") ) {
                String uriDomain = uri.getHost();
                if (uriDomain.equals (value) ) {
                    this.domain = value;
                } else {
                    if (!value.startsWith (".") ) {
                        value = "." + value;
                    }
                    uriDomain = uriDomain.substring (uriDomain.indexOf ('.') );
                    if (!uriDomain.equals (value) ) {
                        throw new IllegalArgumentException ("Trying to set foreign cookie");
                    }
                    this.domain = value;
                }
            } else if (name.equalsIgnoreCase ("path") ) {
                this.path = value;
            } else if (name.equalsIgnoreCase ("expires") ) {
                try {
                    this.expires = expiresFormat1.parse (value);
                } catch (ParseException e) {
                    try {
                        this.expires = expiresFormat2.parse (value);
                    } catch (ParseException e2) {
                        throw new IllegalArgumentException ("Bad date format in header: " + value);
                    }
                }
            }
        }
    }

    public boolean hasExpired() {
        if (expires == null) {
            return false;
        }
        Date now = new Date();
        return now.after (expires);
    }

    public String getName() {
        return name;
    }

    public URI getURI() {
        return uri;
    }

    public boolean matches (URI uri) {

        if (hasExpired() ) {
            return false;
        }
        String path = uri.getPath();
        if (path == null) {
            path = "/";
        }

        return path.startsWith (this.path);
    }

    public String toString() {
        StringBuilder result = new StringBuilder (name);
        result.append ("=");
        result.append (value);
        return result.toString();
    }
}