java 实现类似于python requests包的Session类，自动管理cookie。

﹏ヽ暗。殇╰゛Y 2023-06-02 09:25 1阅读 0赞

以前想用java来写接口自动化测试，总感觉没有python requests模块方便，网上找到一篇可以实现类似于requests的Session类的方法，如下：

原文地址：[https://blog.csdn.net/weixin\_34117211/article/details/85979529][https_blog.csdn.net_weixin_34117211_article_details_85979529]

1、在py中requests.post()和get()函数都是在那个函数内部里面自动生成了一个Session类的实例，所以requests,post和get函数要想干登陆后才能干的事情，需要添加cookie或者把cookie写在headers里面，要想自动管理cookie，就不能每次请求都实例化一个新的Session类的对象了，需要直接实例化Session类，然后使用该实例，而不是使用那两个函数。

py的Session类的使用方法是：

ss = requests.Session()

ss.post(login\_url,data = \{"username":"xiaomin", "password":"123456"\})

ss.get(some\_url)

2、java的OkHttp3默认是不自动管理cookie。

![1108990-20180329103334455-1217452614.png][]

默认是使用NIO\_COOKIES

![1108990-20180329111305249-1060925059.png][]

3、实现cookie自动管理，需要在OkhttpClient类的Builder类中的cookieJar方法传入CookieJar实例。

![1108990-20180329103500968-448722641.png][]

实现CookieJar接口中的saveFromResponse 和loadForRequest方法，把cookie保存到haspmap中，读取也是从hashmap中，这样就实现自动管理cookie。如果要保持代码重启还能持久化cookie管理可以使用redis  sqllite 或者实现Serializable，序列化cookie到文件。

okhttp3.internal.http.BridgeInterceptor;
    okhttp3.internal.http.HttpHeaders;
    为什么实现了CookieJar这两个文件中会调用传进去的实例的这两个方法，根据了请求的url的域名，把cookie添加到request的header了，返回时候从header里面保存cookie，如果传的实例如果没有这两个方法那会运行出错。

![1108990-20180329140621048-1974431953.png][]

![1108990-20180329140632143-142434584.png][]

package com.touna.httprequest;
    
    import java.text.MessageFormat;
    import java.util.ArrayList;
    import java.util.HashMap;
    import java.util.List;
    import java.util.Set;
    import java.util.regex.*;
    
    import okhttp3.Cookie;
    import okhttp3.CookieJar;
    import okhttp3.FormBody;
    import okhttp3.HttpUrl;
    import okhttp3.OkHttpClient;
    import okhttp3.Request;
    import okhttp3.Response;
    
    import com.touna.view.LogUtil;
    
    
    
    public class Session {
    
        private final OkHttpClient mOkHttpClient = new OkHttpClient.Builder().cookieJar(new CookieJarManager()).build();  
    　　 //如果直接写个类，里面写saveFromResponse和loadForRequest，而不实现CookieJar接口，运行可以正常通过，但ide会显红，因为Bulider类的cookieJsr方法需要接受CookieJar类型的实例。鸭子类除了能节约代码行数，在可理解性 可读性 多人合作性上面都不如接口规范，谁知道鸭子类里面要写什么方法，除了写代码得人自己。
        private class CookieJarManager implements CookieJar{
            private final HashMap<String, List<Cookie>> cookieStore = new HashMap<>();
    
            @Override
            public void saveFromResponse(HttpUrl url, List<Cookie> cookies) {
                cookieStore.put(url.host(), cookies);
            }
    
            @Override
            public List<Cookie> loadForRequest(HttpUrl url) {
                List<Cookie> cookies = cookieStore.get(url.host());
                return cookies != null ? cookies : new ArrayList<Cookie>(){};
            }
        }
    
        /**
         * @param url  要请求的url
         * @param  paramsMap post的请求参数
         * @return  post的返回结果
         */
        public String post(String url, HashMap<String, String > paramsMap){
            LogUtil.printLog("请求的url是：" + url);
            FormBody.Builder formBodyBuilder = new FormBody.Builder();
            Set<String> keySet = paramsMap.keySet();
            for(String key:keySet) {
                String value = paramsMap.get(key);
                formBodyBuilder.add(key,value);
            }
            FormBody formBody = formBodyBuilder.build();
            Request request = new Request
                    .Builder()
                    .post(formBody)
                    .url(url)
                    .build();
            try (Response response = mOkHttpClient.newCall(request).execute()) {
                String  respStr = response.body().string();
                LogUtil.printLog("返回是： "   + respStr);
                return respStr;
            }catch (Exception e){
                LogUtil.printLog("post失败");
                e.printStackTrace();
                return "";
            }
        }
    
        public String get(String url)  {
            final Request.Builder builder = new Request.Builder();
            builder.url(url);
            final Request request = builder.build();
            try (Response response = mOkHttpClient.newCall(request).execute()) {
                return response.body().string();
            }catch (Exception e){
                e.printStackTrace();
                return "";
            }
        }
    
        /**
         * @param mail163Account 163邮箱账号;
         * @param passwd 邮箱密码
         * */
        private static void test163(String mail163Account, String passwd){
            String loginUrl = MessageFormat.format("https://mail.163.com/entry/cgi/ntesdoor?funcid=loginone&language=-1&passtype=1" +
                    "&iframe=1&product=mail163&from=web&df=email163&race=-2_262_-2_hz&module=&uid={0}&style=-1&net=t&skinid=null",mail163Account);
            Session ss = new Session();
            HashMap<String,String> paramsMap = new HashMap<>() ;
            paramsMap.put("username",mail163Account);
            paramsMap.put("url2","http://email.163.com/errorpage/error163.htm");
            paramsMap.put("savalogin","0");
            paramsMap.put("password",passwd);
            String respStr = ss.post(loginUrl,paramsMap);  //登录
    
            Pattern sidPattern = Pattern.compile("sid=(.*)&");
            Matcher m = sidPattern.matcher(respStr);
            if (!m.find()){
                LogUtil.printLog("登录失败");
            }else{
                String sid  = m.group(1);
                LogUtil.printLog(sid);
                String mailListUrl = MessageFormat.format("https://mail.163.com/js6/s?sid={0}&func=mbox:listMessages",sid);
                HashMap<String,String> dataMap = new HashMap<>() ;
                dataMap.put("var","<?xml version=\"1.0\"?><object><int name=\"fid\">1</int><string name=\"order\">date</string><boolean name=\"desc\">true" +
                        "</boolean><int name=\"limit\">20</int><int name=\"start\">0</int><boolean name=\"skipLockedFolders\">false</boolean><string name" +
                        "=\"topFlag\">top</string><boolean name=\"returnTag\">true</boolean><boolean name=\"returnTotal\">true</boolean></object>");
                LogUtil.printLog("读取邮件列表:");
                ss.post(mailListUrl,dataMap);
            }
        }
    
        public static void main(String[] args) {
            test163("13148804506@163.com", "123456");
        }
    
    }

4、使用163邮箱登录，然后测试下获取邮件列表能否获取到。

测试结果，这样只要Session类的实例请求了登录接口后，就能干其他的事情了。

![1108990-20180329105617585-1636437239.png][]

转载于:https://www.cnblogs.com/linwenbin/p/11362708.html

[https_blog.csdn.net_weixin_34117211_article_details_85979529]: https://blog.csdn.net/weixin_34117211/article/details/85979529
[1108990-20180329103334455-1217452614.png]: /images/20230601/3a96b5e3fb294c9f869cfac4595a085a.png
[1108990-20180329111305249-1060925059.png]: /images/20230601/f38122a7c6974129a6acc31a4525e3df.png
[1108990-20180329103500968-448722641.png]: /images/20230601/ba36c03a761f43078334cd71e23c0d8b.png
[1108990-20180329140621048-1974431953.png]: /images/20230601/7b87a049f43b4c85aa3b880a9b44003e.png
[1108990-20180329140632143-142434584.png]: /images/20230601/4353441749174ac980c11b19b4b9d362.png
[1108990-20180329105617585-1636437239.png]: /images/20230601/0cd9ddd446724d68add05741eb9dccaf.png