斗鱼弹幕级b站消息队列添加

src/main/java/com/zhiwei/live/danmu/bilibili/BilibiliMessageHandler.java

@@ -65,8 +65,8 @@ public class BilibiliMessageHandler extends ChannelInboundHandlerAdapter{
             Matcher matcher = pattern.matcher(source);
             while(matcher.find()) {
                 JSONObject dataJson = JSONObject.parseObject(matcher.group());
-                BilibiliEntity bilibiliEntity = new BilibiliEntity(dataJson);
-                System.out.println(bilibiliEntity.toString());
+                BilibiliMessage bilibiliMessage = new BilibiliMessage(dataJson);
+                System.out.println(bilibiliMessage);
             }
         }
         ReferenceCountUtil.release(msg);

src/main/java/com/zhiwei/live/danmu/bilibili/BilibiliMessageListener.java

+package com.zhiwei.live.danmu.bilibili;
+
+import java.util.List;
+import java.util.Vector;
+
+public class BilibiliMessageListener {
+	public static List<BilibiliMessage> messages = new Vector<>();
+	
+	public static void addMessages(BilibiliMessage bilibiliMessage) {
+		messages.add(bilibiliMessage);
+	}
+
+}

src/main/java/com/zhiwei/live/danmu/douyu/Main.java

+package com.zhiwei.live.danmu.douyu;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.zhiwei.common.config.GroupType;
+import com.zhiwei.crawler.proxy.ProxyFactory;
+import com.zhiwei.live.danmu.douyu.entity.ChatMsg;
+
+/**
+ * 功能描述：
+ *
+ * @auther: coffee
+ * @date: 2018-07-04 19:37:41
+ * 修改日志:
+ */
+public class Main {
+    private final static Logger logger = LoggerFactory.getLogger(Main.class);
+    private static final String registry = "zookeeper://192.168.0.36:2181";
+	private static final String group = "local";
+	
+	static {
+		ProxyFactory.init(registry, group, GroupType.PROVIDER);
+	}
+   
+    
+    public static void main(String[] args) throws InterruptedException {
+    	DouYuClient client = new DouYuClient("3084150");
+        client.registerMessageListener(new MessageListener<ChatMsg>() {
+            @Override
+            public void read(ChatMsg message) {
+                //logger.info(message.toChatStr());
+//            	System.out.println(message.toChatStr());
+            }
+        });
+        client.registerMessageListener(new MessageListener<String>() {
+            @Override
+            public void read(String message) {
+            	System.out.println(message);
+//                logger.info(message);
+            }
+        });
+        client.login();
+        client.sync();
+
+        Thread.sleep(30000);
+        client.exit();
+    }
+    
+    
+    
+}

src/main/java/com/zhiwei/live/danmu/douyu/MessageHandler.java

-package com.zhiwei.live.danmu.douyu;
-
-import java.io.ByteArrayOutputStream;
-import java.io.IOException;
-import java.io.InputStream;
-import java.io.OutputStream;
-import java.net.Socket;
-
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
-import com.zhiwei.live.danmu.douyu.exceptions.DouYuSDKException;
-
-/**
- * 功能描述：消息处理助手
- *
- * @auther: coffee
- * @date: 2018-07-04 15:11:13
- * 修改日志:
- */
-public class MessageHandler {
-    private final static Logger logger = LoggerFactory.getLogger(MessageHandler.class);
-    private Socket socket;
-
-    public MessageHandler(Socket socket) {
-        this.socket = socket;
-    }
-
-    /**
-     * 发送消息
-     *
-     * @param content
-     */
-    public void send(String content) {
-        try {
-            Message message = new Message(content);
-            OutputStream out = socket.getOutputStream();
-            out.write(message.getBytes());
-            out.flush();
-        } catch (IOException e) {
-            throw new DouYuSDKException(e);
-        }
-    }
-
-    /**
-     * 读取消息
-     *
-     * @return
-     */
-    public byte[] read(){
-        try {
-            InputStream inputStream = socket.getInputStream();
-            //下条信息的长度
-            int contentLen = 0;
-
-            //读取前4个字节，得到数据长度
-            for (int i = 0; i < 4; i++) {
-                int tmp = inputStream.read();
-                contentLen += tmp * Math.pow(16, 2 * i);
-            }
-
-            int len = 0;
-            int readLen = 0;
-            byte[] bytes = new byte[contentLen];
-            ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
-            while ((len = inputStream.read(bytes, 0, contentLen - readLen)) != -1) {
-                byteArray.write(bytes, 0, len);
-                readLen += len;
-                if (readLen == contentLen) {
-                    break;
-                }
-            }
-
-            return byteArray.toByteArray();
-        } catch (IOException e) {
-            throw new DouYuSDKException(e);
-        }
-    }
-
-    /**
-     * 关闭socket通道
-     *
-     * @throws IOException
-     */
-    public void close(){
-        try {
-            socket.close();
-        } catch (IOException e) {
-            logger.warn("socket通道关闭异常",e);
-        }
-    }
-}
+package com.zhiwei.live.danmu.douyu;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.Socket;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.zhiwei.live.danmu.douyu.exceptions.DouYuSDKException;
+
+/**
+ * 功能描述：消息处理助手
+ *
+ * @auther: coffee
+ * @date: 2018-07-04 15:11:13
+ * 修改日志:
+ */
+public class MessageHandler {
+    private final static Logger logger = LoggerFactory.getLogger(MessageHandler.class);
+    private Socket socket;
+
+    public MessageHandler(Socket socket) {
+        this.socket = socket;
+    }
+
+    /**
+     * 发送消息
+     *
+     * @param content
+     */
+    public void send(String content) {
+        try {
+            Message message = new Message(content);
+            OutputStream out = socket.getOutputStream();
+            out.write(message.getBytes());
+            out.flush();
+        } catch (IOException e) {
+            throw new DouYuSDKException(e);
+        }
+    }
+
+    /**
+     * 读取消息
+     *
+     * @return
+     */
+    public byte[] read(){
+        try {
+            InputStream inputStream = socket.getInputStream();
+            //下条信息的长度
+            int contentLen = 0;
+
+            //读取前4个字节，得到数据长度
+            for (int i = 0; i < 4; i++) {
+                int tmp = inputStream.read();
+                contentLen += tmp * Math.pow(16, 2 * i);
+            }
+
+            int len = 0;
+            int readLen = 0;
+            byte[] bytes = new byte[contentLen];
+            ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
+            while ((len = inputStream.read(bytes, 0, contentLen - readLen)) != -1) {
+                byteArray.write(bytes, 0, len);
+                readLen += len;
+                if (readLen == contentLen) {
+                    break;
+                }
+            }
+
+            return byteArray.toByteArray();
+        } catch (IOException e) {
+            throw new DouYuSDKException(e);
+        }
+    }
+
+    /**
+     * 关闭socket通道
+     *
+     * @throws IOException
+     */
+    public void close(){
+        try {
+            socket.close();
+        } catch (IOException e) {
+            logger.warn("socket通道关闭异常",e);
+        }
+    }
+}