改进项目结构

app/websocket/web_socket_server.py

@@ -15,6 +15,8 @@ class SocketManager:
             self.rooms[room].remove(websocket)
             if len(self.rooms[room]) == 0:
                 del self.rooms[room]
+                if room.startswith('detect_rtsp_'):
+                    print()
 
     async def broadcast_to_room(self, room: str, message: str, exclude_websocket: WebSocket = None):
         if room in self.rooms:
@@ -30,8 +32,16 @@ class SocketManager:
             for ws in self.rooms[room]:
                 try:
                     await ws.send_text(message)
-                except:
-                    await self.remove_from_room(room, ws)
+                except Exception as e:
+                    print(e)
+
+    async def send_stream_to_room(self, room: str, message):
+        if room in self.rooms:
+            for ws in self.rooms[room]:
+                try:
+                    await ws.send_bytes(message)
+                except Exception as e:
+                    print(e)
 
 
 room_manager = SocketManager()