modify: IN2 DI 输入监听

2026-09-13 20:34:10 +08:00
parent d7636c4c3b
commit 293254c61a
9 changed files with 272 additions and 704 deletions
@@ -0,0 +1,175 @@
+"""
+AI Box Sophon Adapter — IN2 数字输入控制任务启停
+
+监听盒子 IN2 DI 输入（复用 diwatch 的 DIChannel + SysfsBackend，sysfs 轮询），
+电平变化时通过盒子本地 HTTP 接口控制所有算法任务：
+
+  - IN2 有电（电平 1）→ 启动全部任务（ControlCommand = start_command，默认 1）
+  - IN2 断电（电平 0）→ 停止全部任务（ControlCommand = stop_command，默认 0）
+
+盒子接口（POST，JSON）：
+  - /api/alg_task_fetch    查询全部任务：{} → {"Content": [{"AlgTaskSession": ...}], "Result": {"Code": 0}}
+  - /api/alg_task_control  控制单个任务：{"AlgTaskSession": ..., "ControlCommand": 0/1}
+                           → {"Result": {"Code": 0, "Desc": "已停止"}}
+
+说明：
+  - 只响应电平跳变，不按启动时的初始电平动作（避免插件重启时误触发全量启停）。
+  - DIChannel 阻塞轮询跑在独立线程，回调经 run_coroutine_threadsafe 调度回事件循环，
+    HTTP 控制在事件循环内串行执行（apply_lock 防止快速抖动导致并发穿插）。
+"""
+import asyncio
+import logging
+import threading
+
+import httpx
+
+from .diwatch.backends import SysfsBackend
+from .diwatch.config import IN2_INDEX, find_status_file
+from .diwatch.core import DIChannel
+
+logger = logging.getLogger(__name__)
+
+
+class In2TaskSwitch:
+    """IN2 数字输入 → 全部算法任务启停开关。"""
+
+    def __init__(
+        self,
+        api_base: str = "http://127.0.0.1",
+        start_command: int = 1,
+        stop_command: int = 0,
+        poll_interval: float = 0.05,
+        debounce: int = 3,
+        active_low: bool = False,
+        timeout: float = 10.0,
+    ):
+        self.api_base = api_base.rstrip("/")
+        self.start_command = start_command
+        self.stop_command = stop_command
+        self.poll_interval = poll_interval
+        self.debounce = debounce
+        self.active_low = active_low
+        self.timeout = timeout
+
+        self._channel: DIChannel | None = None
+        self._thread: threading.Thread | None = None
+        self._loop: asyncio.AbstractEventLoop | None = None
+        self._apply_lock = asyncio.Lock()
+
+    # ---- 生命周期 ----
+    def start(self, loop: asyncio.AbstractEventLoop) -> bool:
+        """启动监听线程；找不到 IN2 状态文件时返回 False（非盒子环境属正常）。"""
+        status = find_status_file(IN2_INDEX)
+        if not status:
+            logger.warning(
+                "[TaskSwitch] 找不到 IN2 状态文件，任务启停开关未启动"
+                "（非盒子环境属正常现象）"
+            )
+            return False
+
+        backend = SysfsBackend(status, debounce=self.debounce, interval=self.poll_interval)
+        self._channel = DIChannel(
+            "IN2", backend,
+            active_low=self.active_low,
+            on_change=self._on_level_change,
+            log=lambda m: logger.debug(f"[TaskSwitch] {m}"),
+        )
+        self._loop = loop
+        self._thread = threading.Thread(
+            target=self._run_channel, name="in2-task-switch", daemon=True,
+        )
+        self._thread.start()
+        logger.info(
+            f"[TaskSwitch] IN2 任务启停开关已启动: 状态文件={status}, "
+            f"api_base={self.api_base}, "
+            f"启动命令={self.start_command}, 停止命令={self.stop_command}"
+        )
+        return True
+
+    def stop(self) -> None:
+        if self._channel is not None:
+            self._channel.request_stop()
+        if self._thread is not None:
+            self._thread.join(timeout=2.0)
+            self._thread = None
+        logger.info("[TaskSwitch] IN2 任务启停开关已停止")
+
+    def _run_channel(self) -> None:
+        try:
+            self._channel.run()
+        except Exception as e:
+            logger.error(f"[TaskSwitch] IN2 监听线程异常退出: {type(e).__name__}: {e}")
+
+    # ---- DI 回调（轮询线程内执行）----
+    def _on_level_change(self, old: int, new: int, ts: float) -> None:
+        logger.info(f"[TaskSwitch] IN2 电平变化: {old} -> {new}")
+        if self._loop is not None:
+            asyncio.run_coroutine_threadsafe(self.apply(new), self._loop)
+
+    # ---- 任务控制（事件循环内执行）----
+    async def apply(self, level: int) -> None:
+        """按目标电平控制全部任务：1=启动，0=停止。"""
+        async with self._apply_lock:
+            command = self.start_command if level == 1 else self.stop_command
+            action = "启动" if level == 1 else "停止"
+
+            sessions = await self._fetch_task_sessions()
+            if not sessions:
+                logger.warning("[TaskSwitch] 未获取到任何任务，跳过控制")
+                return
+
+            logger.info(
+                f"[TaskSwitch] IN2 {'有电' if level else '断电'}，"
+                f"{action}全部 {len(sessions)} 个任务: {sessions}"
+            )
+            ok = 0
+            failed = 0
+            async with httpx.AsyncClient(timeout=self.timeout) as client:
+                for session in sessions:
+                    try:
+                        resp = await client.post(
+                            f"{self.api_base}/api/alg_task_control",
+                            json={"AlgTaskSession": session, "ControlCommand": command},
+                        )
+                        result = resp.json().get("Result", {})
+                        if resp.is_success and result.get("Code") == 0:
+                            ok += 1
+                            logger.info(
+                                f"[TaskSwitch] 任务 {session} {action}成功: "
+                                f"{result.get('Desc', '')}"
+                            )
+                        else:
+                            failed += 1
+                            logger.warning(
+                                f"[TaskSwitch] 任务 {session} {action}失败: "
+                                f"Status={resp.status_code}, Body={resp.text[:200]}"
+                            )
+                    except Exception as e:
+                        failed += 1
+                        logger.error(
+                            f"[TaskSwitch] 任务 {session} {action}异常: "
+                            f"{type(e).__name__}: {e}"
+                        )
+            logger.info(f"[TaskSwitch] {action}完成: 成功 {ok}, 失败 {failed}")
+
+    async def _fetch_task_sessions(self) -> list[str]:
+        """查询全部任务的 AlgTaskSession 列表；失败返回空列表。"""
+        try:
+            async with httpx.AsyncClient(timeout=self.timeout) as client:
+                resp = await client.post(f"{self.api_base}/api/alg_task_fetch", json={})
+            if not resp.is_success:
+                logger.warning(
+                    f"[TaskSwitch] 查询任务失败: Status={resp.status_code}, "
+                    f"Body={resp.text[:200]}"
+                )
+                return []
+            data = resp.json()
+            result = data.get("Result", {})
+            if result.get("Code") != 0:
+                logger.warning(f"[TaskSwitch] 查询任务返回错误: {result}")
+                return []
+            content = data.get("Content") or []
+            return [t.get("AlgTaskSession") for t in content if t.get("AlgTaskSession")]
+        except Exception as e:
+            logger.error(f"[TaskSwitch] 查询任务异常: {type(e).__name__}: {e}")
+            return []