牧安云哨-后端

2025-12-23 09:55:06 +08:00
parent f018d96e60
commit b5a9714025
16 changed files with 1219 additions and 75 deletions
@@ -0,0 +1,23 @@
+from fastapi import APIRouter
+from starlette.websockets import WebSocket, WebSocketDisconnect
+
+from db.postgres.ws_manager import ConnectionManager
+
+ws_manager = ConnectionManager()
+
+
+iot_ws_router = APIRouter()
+
+
+@iot_ws_router.websocket("/device-status")
+async def websocket_device_status(websocket: WebSocket):
+    await ws_manager.connect(websocket)
+    print("[WS] client connected")
+
+    try:
+        while True:
+            # 这里不需要接收任何消息
+            await websocket.receive_text()
+    except WebSocketDisconnect:
+        await ws_manager.disconnect(websocket)
+        print("[WS] client disconnected")