增强鲁棒性

Author: gaopu

CrawlDouyuDanmu/src/main/java/Crawl.java

@@ -16,14 +16,16 @@ public class Crawl extends Thread {
     String gid;
     //与弹幕服务器交互的控制器
     MessageHandler messageHandler;
+    //登录名
+    String username;
 
     public Crawl() throws IOException {
         rid = Utils.getRoomId();
     }
 
     /**
      *
-     * 初始化弹幕服务器地址数据和弹幕分组信息
+     * 初始化"弹幕服务器地址数据"和"弹幕分组信息"和"登录用户名"
      */
     public void init() throws IOException {
         String ip = Utils.getServerIP();
@@ -56,6 +58,8 @@ public void init() throws IOException {
                 }
             } else if (msg.startsWith("type@=setmsggroup")) {
                 gid = msg.split("gid@=")[1].split("/")[0];
+            } else if (msg.startsWith("type@=loginres")) {
+                username = msg.split("username@=")[1].split("/")[0];
             }
         }
         socket.close();
@@ -66,43 +70,59 @@ public void login() throws IOException {
         System.out.println("连接弹幕服务器(danmu.douyutv.com:" + ports.get(0) + ")");
         messageHandler = new MessageHandler(socket);
 
-        String loginreq = "type@=loginreq/username@=visitor503535/password@=1234567890123456/roomid@=" + rid + "/";
+        String loginreq = "type@=loginreq/username@=" + username + "/password@=1234567890123456/roomid@=" + rid + "/";
         messageHandler.send(loginreq);
-        String joinGroup = "type@=joingroup/rid@=" + rid + "/gid@=" + gid + "/";
+        System.out.println("登录名:" + username);
+
+        String joinGroup;
+        if (Utils.isSeaMode()) {
+            joinGroup = "type@=joingroup/rid@=" + rid + "/gid@=-9999/";
+            System.out.println("海量弹幕模式:开启");
+        } else {
+            joinGroup = "type@=joingroup/rid@=" + rid + "/gid@=" + gid + "/";
+            System.out.println("海量弹幕模式:关闭");
+            System.out.println("进入" + gid + "号弹幕分组");
+        }
+
         messageHandler.send(joinGroup);
-        System.out.println("进入" + gid + "号弹幕分组");
     }
 
     @Override
     public void run() {
         try {
             System.out.println("房间名:" + Utils.getRoomName());
             System.out.println("主播:" + Utils.getOwnerName());
-            if (!Utils.roomIsAlive()) {
+            /*if (!Utils.roomIsAlive()) {
                 System.out.println("房间未开播，程序结束.");
                 return;
             } else {
                 System.out.println("状态:正在直播");
-            }
+            }*/
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
 
+        try {
             init();
             login();
+            login();
+            System.out.println("--------------------------");
 
             long start = System.currentTimeMillis();
             while (true) {
                 byte[] bytes = messageHandler.read();
                 String msg = new String(Arrays.copyOfRange(bytes,8,bytes.length));
 
                 if (msg.startsWith("type@=chatmessage")) {
-                    String nickname = msg.split("@Snick@A=")[1].split("@Srg@A")[0];
-                    String content = msg.split("content@=")[1].split("/snick@=")[0];
+                    String nickname = msg.split("Snick@A=",2)[1].split("@",2)[0];
+                    String content = msg.split("content@=",2)[1].split("/",2)[0];
                     System.out.println("[" + nickname + "]:" + content);
-//                    System.out.println(msg);
                 }
 
                 long end = System.currentTimeMillis();
                 if (end - start > 30000) {
                     messageHandler.send("type=mrkl/");
+                    start = System.currentTimeMillis();
                 }
 
                 Thread.sleep(1);

CrawlDouyuDanmu/src/main/java/MessageHandler.java

@@ -36,7 +36,7 @@ public byte[] read() throws IOException {
         byte[] bytes = new byte[contentLen];
         ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
         while ((len = inputStream.read(bytes,0,contentLen - readLen)) != -1) {
-            byteArray.write(bytes);
+            byteArray.write(bytes,0,len);
             readLen += len;
             if (readLen == contentLen) {
                 break;

CrawlDouyuDanmu/src/main/java/Utils.java

@@ -69,6 +69,14 @@ public static String getRoomUrl() throws IOException {
         }
     }
 
+    /**
+     *
+     * @return 返回值代表是否开启海量弹幕模式
+     */
+    public static boolean isSeaMode() {
+        return config.getProperty("seaMode").equals("true");
+    }
+
     public static String getServerIP() throws IOException {
         return config.getProperty("serverIP");
     }

CrawlDouyuDanmu/src/main/resources/config.properties

@@ -1,6 +1,10 @@
 #要抓取的斗鱼房间网址
-url = http://www.douyutv.com/caomei
+url = http://www.douyutv.com/wt55kai
 
 #一开始通讯的服务器，不是弹幕服务器，弹幕服务器地址是与这个服务器通讯后返回的(抓包得到这个地址)
 serverIP = 119.90.49.93
-serverPort = 8062
+serverPort = 8062
+
+#海量弹幕模式 (true || false)
+#在人气少的主播房间好像是不管用的，看人气少的主播弹幕设置为false
+seaMode = true