From 95e18d831b6e1e3509e24e1fe3eed9f1d0b70f6d Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期四, 31 七月 2025 14:13:27 +0800
Subject: [PATCH] 添加推送日志

---
 socket_manager.py |   20 ++++++++++++++------
 1 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/socket_manager.py b/socket_manager.py
index 9c64bd9..ae6847a 100644
--- a/socket_manager.py
+++ b/socket_manager.py
@@ -6,7 +6,12 @@
 
 class ClientSocketManager:
     # 瀹㈡埛绔被鍨�
+    CLIENT_TYPE_COMMON = "common"
     CLIENT_TYPE_TRADE = "trade"
+    CLIENT_TYPE_TRADE_SELL = "trade_sell"
+    CLIENT_TYPE_TRADE_LOW_SUCTION = "trade_low_suction"
+    # 鍙浆鍊哄鎴风
+    CLIENT_TYPE_TRADE_CB = "trade_cb"
 
     socket_client_dict = {}
     socket_client_lock_dict = {}
@@ -14,7 +19,8 @@
 
     @classmethod
     def add_client(cls, _type, rid, sk):
-        if _type == cls.CLIENT_TYPE_TRADE:
+        if _type in {cls.CLIENT_TYPE_COMMON, cls.CLIENT_TYPE_TRADE, cls.CLIENT_TYPE_TRADE_SELL,
+                     cls.CLIENT_TYPE_TRADE_CB, cls.CLIENT_TYPE_TRADE_LOW_SUCTION}:
             # 浜ゆ槗鍒楄〃
             if _type not in cls.socket_client_dict:
                 cls.socket_client_dict[_type] = []
@@ -26,7 +32,8 @@
 
     @classmethod
     def acquire_client(cls, _type):
-        if _type == cls.CLIENT_TYPE_TRADE:
+        if _type in {cls.CLIENT_TYPE_COMMON, cls.CLIENT_TYPE_TRADE, cls.CLIENT_TYPE_TRADE_SELL,
+                     cls.CLIENT_TYPE_TRADE_CB, cls.CLIENT_TYPE_TRADE_LOW_SUCTION}:
             if _type in cls.socket_client_dict:
                 # 鏍规嵁鎺掑簭娲昏穬鏃堕棿鎺掑簭
                 client_list = sorted(cls.socket_client_dict[_type], key=lambda x: cls.active_client_dict.get(x[0]) if x[
@@ -96,8 +103,10 @@
                 cls.del_client(k)
 
     @classmethod
-    def list_client(cls):
-        _type = cls.CLIENT_TYPE_TRADE
+    def list_client(cls, type_=None):
+        _type = type_
+        if not _type:
+            _type = cls.CLIENT_TYPE_TRADE
         client_list = sorted(cls.socket_client_dict[_type],
                              key=lambda x: cls.active_client_dict.get(x[0]) if x[0] in cls.active_client_dict else 0,
                              reverse=True)
@@ -108,6 +117,5 @@
                 active_time = 0
             active_time = tool.to_time_str(int(active_time))
             fdata.append(
-                (client[0], cls.socket_client_lock_dict[client[0]].locked(),active_time))
+                (client[0], cls.socket_client_lock_dict[client[0]].locked(), active_time))
         return fdata
-

--
Gitblit v1.8.0