【Java】CookieStore 类使用示例

CookieStore 是 Java API 中用来处理 HTTP 客户端的 Cookie 存储策略的类。psd素材

1. [代码]WebClient.java     

01import java.io.BufferedReader;

02import java.io.InputStream;

03import java.io.InputStreamReader;

04import java.net.CookieHandler;

05import java.net.CookieManager;

06import java.net.CookiePolicy;

07import java.net.CookieStore;

08import java.net.HttpCookie;

09import java.net.URI;

10import java.net.URL;

11import java.net.URLConnection;

12import java.util.ArrayList;

13import java.util.Collection;

14import java.util.HashMap;

15import java.util.List;

16import java.util.Map;

17import java.util.Set;

18 

19 

20public class WebClient {

21  public static void main(String[] args) throws Exception {

22    CookieStore store = new MyCookieStore();

23    CookiePolicy policy = new MyCookiePolicy();

24    CookieManager handler = new CookieManager(store, policy);

25    CookieHandler.setDefault(handler);

26    URL url = new URL("http://localhost:8080/cookieTest.jsp");

27    URLConnection conn = url.openConnection();

28 

29    InputStream in = conn.getInputStream();

30    BufferedReader reader = new BufferedReader(new InputStreamReader(in));

31    String input;

32    while ((input = reader.readLine()) != null) {

33      System.out.println(input);

34    }

35    reader.close();

36 

37  }

38}

39 

40class MyCookiePolicy implements CookiePolicy {

41  public boolean shouldAccept(URI uri, HttpCookie cookie) {

42//    String host = uri.getHost();

43  //  return host.equals("localhost");

44    return true;

45  }

46}

47 

48class MyCookieStore implements CookieStore {

49  private Map<URI, List<HttpCookie>> map = new HashMap<URI, List<HttpCookie>>();

50 

51  public void add(URI uri, HttpCookie cookie) {

52    List<HttpCookie> cookies = map.get(uri);

53    if (cookies == null) {

54      cookies = new ArrayList<HttpCookie>();

55      map.put(uri, cookies);

56    }

57    cookies.add(cookie);

58  }

59 

60  public List<HttpCookie> get(URI uri) {

61    List<HttpCookie> cookies = map.get(uri);

62    if (cookies == null) {

63      cookies = new ArrayList<HttpCookie>();

64      map.put(uri, cookies);

65    }http://www.huiyi8.com/jiaoben/

66    return cookies;

67  }

68 

69  public List<HttpCookie> getCookies() {

70    Collection<List<HttpCookie>> values = map.values();

71    List<HttpCookie> result = new ArrayList<HttpCookie>();

72    for (List<HttpCookie> value : values) {

73      result.addAll(value);

74    }

75    return result;

76  }

77 

78  public List<URI> getURIs() {

79    Set<URI> keys = map.keySet();

80    return new ArrayList<URI>(keys);

81 

82  }

83 

84  public boolean remove(URI uri, HttpCookie cookie) {

85    List<HttpCookie> cookies = map.get(uri);

86    if (cookies == null) {

87      return false;

88    }

89    return cookies.remove(cookie);

90  }

91 

92  public boolean removeAll() {

93    map.clear();

94    return true;

95  }

96}