From f4afd409611e2577094a5919d2c047f201b3ae6d Mon Sep 17 00:00:00 2001 From: JIAKUNHAO Date: Thu, 24 Nov 2022 15:31:40 +0800 Subject: [PATCH] ws --- app/utils/websocket_tool.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/app/utils/websocket_tool.py b/app/utils/websocket_tool.py index cd313af..18c5f02 100644 --- a/app/utils/websocket_tool.py +++ b/app/utils/websocket_tool.py @@ -24,11 +24,13 @@ class WebsocketUtil: active_connections_dist = {} self.queue = Queue(maxsize=0) self.queue.put(active_connections_dist) + print("--------------------队列长度：", self.queue.qsize()) def connect(self, ws, id: str): # 等待连接 msg = ws.receive() # 存储ws连接对象 + print("--------------------队列长度：", self.queue.qsize()) active_connections_dist = self.queue.get() if id in active_connections_dist: active_connections_dist[id].append(ws) @@ -36,16 +38,19 @@ class WebsocketUtil: ws_list = [ws, ] active_connections_dist[id] = ws_list self.queue.put(active_connections_dist) + print("--------------------队列长度：", self.queue.qsize()) def disconnect(self, ws, id): # ws关闭时 移除ws对象 + print("--------------------队列长度：", self.queue.qsize()) if ws.closed: active_connections_dist = self.queue.get() if active_connections_dist is not None: if ws in active_connections_dist.values(): active_connections_dist[id].pop(ws) self.queue.put(active_connections_dist) + print("--------------------队列长度：", self.queue.qsize()) @staticmethod async def send_personal_message(message: str, ws): @@ -74,6 +79,7 @@ class WebsocketUtil: except Exception as e: logger.error("websocket异常断开，{}", e) self.disconnect(ws=connection, id=id) + print("--------------------队列长度：", self.queue.qsize()) manager = WebsocketUtil() \ No newline at end of file