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

ws

Browse Source
This commit is contained in:
JIAKUNHAO 2022-11-24 17:40:35 +08:00
parent 00c95cca9e
commit b178a93fdf
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/utils/websocket_tool.py
View File

@ -31,7 +31,8 @@ class WebsocketUtil:
msg = ws.receive() msg = ws.receive()
# 存储ws连接对象 # 存储ws连接对象
if os.path.exists(f"{id}.pkl"): if os.path.exists(f"{id}.pkl"):
ws_list = read(id=id) wsFile = read(id=id)
ws_list = wsFile.ws
ws_list.append(ws) ws_list.append(ws)
data = WsFile(id=id, ws=ws_list) data = WsFile(id=id, ws=ws_list)
write(id=id, ws=data) write(id=id, ws=data)
@ -65,8 +66,8 @@ class WebsocketUtil:
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):
# 广播该项目的消息 # 广播该项目的消息
active_connections = read(id=id) active_connections = read(id=id)
print("===================", type(active_connections), active_connections[0]) print("===================", type(active_connections.ws), active_connections.ws[0])
for connection in active_connections: for connection in active_connections.ws:
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: