modify: internal_mqtt 测试

This commit is contained in:
2026-08-07 16:40:48 +08:00
parent 54461cad18
commit ba09b6768e
+12 -13
View File
@@ -115,7 +115,7 @@ async def _cmd_alg_media_fetch(payload: dict) -> dict | None:
请求: {"Event": "/alg_media_fetch", "BoardId": "..."} 请求: {"Event": "/alg_media_fetch", "BoardId": "..."}
返回: {"board_id": ..., "medias": [...], "tasks": [...]} 返回: {"board_id": ..., "medias": [...], "tasks": [...]}
""" """
from .global_var import db from .global_var import db, internal_mqtt
board_id = payload.get("BoardId") board_id = payload.get("BoardId")
if not board_id: if not board_id:
@@ -126,6 +126,8 @@ async def _cmd_alg_media_fetch(payload: dict) -> dict | None:
if db is None: if db is None:
raise RuntimeError("数据库未初始化") raise RuntimeError("数据库未初始化")
if internal_mqtt is None:
raise RuntimeError("internal_mqtt 未初始化")
async with db.get_session() as session: async with db.get_session() as session:
from .models.orm import BoardHeartbeat from .models.orm import BoardHeartbeat
@@ -139,19 +141,16 @@ async def _cmd_alg_media_fetch(payload: dict) -> dict | None:
print(f"[alg_media_fetch]Board {board_id} not found in database.") print(f"[alg_media_fetch]Board {board_id} not found in database.")
return None return None
import json resp = await internal_mqtt.request(
ping_data = {} request_topic="/chongming/aibox_sophon_adapter/edge_app_controller",
try: response_topic="/chongming/aibox_sophon_adapter/edge_app_controller_reply",
ping_data = json.loads(record.ping_data) if record.ping_data else {} payload=payload,
except json.JSONDecodeError: timeout=5
pass )
return { print(f"[alg_media_fetch]Response from internal_mqtt: {resp}")
"board_id": board_id,
"board_ip": record.board_ip, return resp
"medias": ping_data.get("Medias", []),
"tasks": ping_data.get("Tasks", []),
}
async def _cmd_alg_task_fetch(payload: dict) -> dict | None: async def _cmd_alg_task_fetch(payload: dict) -> dict | None:
""" """