AI实验室后端

bbit_ai/app/routers/WS.py

@@ -49,3 +49,23 @@ async def websocket_sentinel_record(
             await websocket.receive_text()
     except WebSocketDisconnect:
         await ws_manager.disconnect(websocket)
+
+
+# Vue 牧安云哨 监控大屏 消息通知
+@iot_ws_router.websocket("/sentinel_record_notice")
+async def websocket_sentinel_record(
+    websocket: WebSocket,
+    token: str = Query(...),
+):
+    user_id = get_user_id_from_token_from_ws(token)
+    dept_id = get_dept_id_by_user_id(user_id)  # 查数据库或缓存
+    print("user_id:", user_id)
+    print("dept_id:", dept_id)
+    print("已接入")
+    await ws_manager.connect(websocket, user_id, dept_id, 2)
+
+    try:
+        while True:
+            await websocket.receive_text()
+    except WebSocketDisconnect:
+        await ws_manager.disconnect(websocket)