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": "..."}
返回: {"board_id": ..., "medias": [...], "tasks": [...]}
"""
from .global_var import db
from .global_var import db, internal_mqtt
board_id = payload.get("BoardId")
if not board_id:
@@ -126,6 +126,8 @@ async def _cmd_alg_media_fetch(payload: dict) -> dict | None:
if db is None:
raise RuntimeError("数据库未初始化")
if internal_mqtt is None:
raise RuntimeError("internal_mqtt 未初始化")
async with db.get_session() as session:
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.")
return None
import json
ping_data = {}
try:
ping_data = json.loads(record.ping_data) if record.ping_data else {}
except json.JSONDecodeError:
pass
resp = await internal_mqtt.request(
request_topic="/chongming/aibox_sophon_adapter/edge_app_controller",
response_topic="/chongming/aibox_sophon_adapter/edge_app_controller_reply",
payload=payload,
timeout=5
)
return {
"board_id": board_id,
"board_ip": record.board_ip,
"medias": ping_data.get("Medias", []),
"tasks": ping_data.get("Tasks", []),
}
print(f"[alg_media_fetch]Response from internal_mqtt: {resp}")
return resp
async def _cmd_alg_task_fetch(payload: dict) -> dict | None:
"""