xkrs_manan/RODY
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

全局变量

Browse Source
This commit is contained in:
JIAKUNHAO 2022-11-24 11:18:07 +08:00
parent b26bf52eda
commit b31867f12a
1 changed files with 22 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
app/utils/websocket_tool.py
View File

@ -14,39 +14,39 @@ from app.core.common_utils import logger
from app.utils.JSONEncodeTools import MyEncoder from app.utils.JSONEncodeTools import MyEncoder
from configs.global_var import * from configs.global_var import *
# global active_connections global active_connections
# active_connections = multiprocessing.Manager().list() active_connections = multiprocessing.Manager().list()
# global active_connections_dist global active_connections_dist
# active_connections_dist = multiprocessing.Manager().dict() active_connections_dist = multiprocessing.Manager().dict()
class WebsocketUtil: class WebsocketUtil:
def __init__(self): # def __init__(self):
self.active_connections = multiprocessing.Manager().list() # self.active_connections = multiprocessing.Manager().list()
self.active_connections_dist = multiprocessing.Manager().dict() # self.active_connections_dist = multiprocessing.Manager().dict()
def connect(self, ws, id: str): def connect(self, ws, id: str):
# 等待连接 # 等待连接
msg = ws.receive() msg = ws.receive()
# 存储ws连接对象 # 存储ws连接对象
# global active_connections global active_connections
self.active_connections.append(ws) active_connections.append(ws)
# global active_connections_dist global active_connections_dist
if id in self.active_connections_dist: if id in active_connections_dist:
self.active_connections_dist[id].append(ws) active_connections_dist[id].append(ws)
else: else:
ws_list = [ws, ] ws_list = [ws, ]
self.active_connections_dist[id] = ws_list active_connections_dist[id] = ws_list
def disconnect(self, ws, id): def disconnect(self, ws, id):
# ws关闭时 移除ws对象 # ws关闭时 移除ws对象
if ws.closed: if ws.closed:
if ws in self.active_connections_dist.values(): if ws in active_connections_dist.values():
# global active_connections global active_connections
self.active_connections.remove(ws) active_connections.remove(ws)
# global active_connections_dist global active_connections_dist
self.active_connections_dist[id].pop(ws) active_connections_dist[id].pop(ws)
@staticmethod @staticmethod
async def send_personal_message(message: str, ws): async def send_personal_message(message: str, ws):
@ -55,14 +55,14 @@ class WebsocketUtil:
def broadcast(self, message: str): def broadcast(self, message: str):
# 广播消息 # 广播消息
# global active_connections global active_connections
for connection in self.active_connections: for connection in active_connections:
connection.send(message) connection.send(message)
def send_message_proj_json(self, message: Union[str, int, List, Dict], id: str): def send_message_proj_json(self, message: Union[str, int, List, Dict], id: str):
# 广播该项目的消息 # 广播该项目的消息
# global active_connections_dist global active_connections_dist
for connection in self.active_connections_dist[id]: for connection in active_connections_dist[id]:
try: try:
connection.send(json.dumps(message, cls=MyEncoder, indent=4), ) connection.send(json.dumps(message, cls=MyEncoder, indent=4), )
except Exception as e: except Exception as e: