牧安云哨-后端

bbit_ai/app/routers/Iot.py

@@ -0,0 +1,123 @@
+from uuid import UUID
+
+from fastapi import APIRouter
+from fastapi import Depends
+
+from config.redis import RedisClient
+from db.postgres.iot import *
+from models.BaseResponse import BaseResponse
+from models.EMQXWebhook import EMQXWebhook
+from routers.WS import ws_manager
+
+iot_router = APIRouter()
+redis_client = RedisClient()
+
+from config.security import get_user_id_from_token
+
+# -------------------- 设备接口 --------------------
+
+
+@iot_router.post("/common/webhook")
+async def emqx_webhook(data: EMQXWebhook):
+    device_id = data.clientid
+    event = data.event
+
+    if event == "client.connected":
+        redis_client.set_online(device_id)
+
+        await ws_manager.broadcast({"deviceId": device_id, "online": True})
+
+        print(f"[ONLINE] {device_id}")
+
+    elif event == "client.disconnected":
+        redis_client.set_offline(device_id)
+
+        await ws_manager.broadcast({"deviceId": device_id, "online": False})
+
+        print(f"[OFFLINE] {device_id}")
+
+    else:
+        # 其他事件直接忽略
+        print(f"[IGNORE] {event}")
+
+    return {"ok": True}
+
+
+@iot_router.get("/common/device/list")
+async def get_device_list(
+    page: int = 1,
+    pageSize: int = 10,
+    id: str | None = None,
+    name: str | None = None,
+    status: int | None = None,
+    is_superuser: int | None = None,
+    dept_id: str | None = None,
+    startTime: str | None = None,
+    endTime: str | None = None,
+    user_id: UUID = Depends(get_user_id_from_token),
+):
+    if not user_id:
+        return {"error": "userId is required"}
+
+    devices, total = get_device_list_db_page(
+        page, pageSize, id, name, status, is_superuser, dept_id, startTime, endTime
+    )
+
+    # ===== 👇 核心：补在线状态 =====
+    for d in devices:
+        device_id = d["name"]  # 账号
+        d["online"] = redis_client.is_device_online(device_id) == 1
+
+    return BaseResponse(data={"list": devices, "total": total})
+
+
+@iot_router.post("/common/device")
+async def create_device(data: dict, user_id: UUID = Depends(get_user_id_from_token)):
+    if not user_id:
+        return {"error": "userId is required"}
+    new_id = insert_device(data)
+    return BaseResponse(data={"id": new_id})
+
+
+@iot_router.put("/common/device/{id}")
+async def update_device(
+    id: str,
+    data: dict,
+    user_id: UUID = Depends(get_user_id_from_token),
+):
+    if not user_id:
+        return {"error": "userId is required"}
+
+    count = update_device_db(id, data)
+    if count == 0:
+        return BaseResponse(status=False, message="设备不存在", data=None)
+    return BaseResponse(data=True)
+
+
+@iot_router.patch("/common/device/{id}")
+async def patch_device(
+    id: str,
+    data: dict,
+    user_id: UUID = Depends(get_user_id_from_token),
+):
+    if not user_id:
+        return {"error": "userId is required"}
+
+    count = patch_device_db(id, data)
+    if count == 0:
+        return BaseResponse(status=False, message="设备不存在", data=None)
+    return BaseResponse(data=True)
+
+
+@iot_router.delete("/common/device/{id}")
+async def delete_device(
+    id: str,
+    user_id: UUID = Depends(get_user_id_from_token),
+):
+    if not user_id:
+        return {"error": "userId is required"}
+
+    deleted = delete_device_db(id)
+    if deleted == 0:
+        return BaseResponse(status=False, message="设备不存在", data=None)
+    return BaseResponse(data=True)