초기 커밋

app/communication/mqtt/handlers.py

@@ -0,0 +1,86 @@
+from __future__ import annotations
+
+import json
+
+import structlog
+
+from app.communication.mqtt.client import mqtt
+from app.models.mongodb.device_log import DeviceLog
+from app.models.mongodb.telemetry import TelemetryData
+
+logger = structlog.get_logger("mqtt")
+
+
+def _extract_device_uid(topic: str) -> str:
+    parts = topic.split("/")
+    return parts[1] if len(parts) >= 3 else "unknown"
+
+
+@mqtt.on_message()
+async def on_message(client, topic: str, payload: bytes, qos: int, properties) -> None:  # type: ignore[no-untyped-def]
+    device_uid = _extract_device_uid(topic)
+
+    try:
+        data = json.loads(payload.decode())
+    except (json.JSONDecodeError, UnicodeDecodeError):
+        logger.warning("invalid_mqtt_payload", topic=topic)
+        return
+
+    if "/telemetry" in topic:
+        await _handle_telemetry(device_uid, data)
+    elif "/status" in topic:
+        await _handle_status(device_uid, data)
+    elif "/log" in topic:
+        await _handle_log(device_uid, data)
+    elif "/response" in topic:
+        await _handle_response(device_uid, data)
+
+
+async def _handle_telemetry(device_uid: str, data: dict) -> None:
+    telemetry = TelemetryData(device_id=device_uid, metrics=data)
+    await telemetry.insert()
+
+    # Broadcast via Socket.IO
+    from app.communication.socketio.server import sio
+
+    await sio.emit(
+        "telemetry",
+        {"device_uid": device_uid, "data": data},
+        namespace="/monitoring",
+    )
+    logger.debug("telemetry_saved", device_uid=device_uid)
+
+
+async def _handle_status(device_uid: str, data: dict) -> None:
+    log = DeviceLog(device_id=device_uid, event_type="status_change", payload=data)
+    await log.insert()
+
+    from app.communication.socketio.server import sio
+
+    await sio.emit(
+        "device_status",
+        {"device_uid": device_uid, "status": data},
+        namespace="/device",
+    )
+    logger.debug("status_update", device_uid=device_uid)
+
+
+async def _handle_log(device_uid: str, data: dict) -> None:
+    log = DeviceLog(
+        device_id=device_uid,
+        event_type=data.get("event_type", "log"),
+        payload=data,
+    )
+    await log.insert()
+    logger.debug("device_log_saved", device_uid=device_uid)
+
+
+async def _handle_response(device_uid: str, data: dict) -> None:
+    from app.communication.socketio.server import sio
+
+    await sio.emit(
+        "device_response",
+        {"device_uid": device_uid, "data": data},
+        namespace="/device",
+    )
+    logger.debug("device_response", device_uid=device_uid)