Files
xiaodoudou-ComfyUI-RigShare/classes/WebSocketManager.py
T
2024-08-24 19:30:32 +05:30

14 lines
283 B
Python

class WebSocketManager:
def __init__(self):
self.sockets = {}
def get(self, id):
return self.sockets.get(id)
def set(self, id, ws):
self.sockets[id] = ws
def delete(self, id):
if id in self.sockets:
del self.sockets[id]