""" 通道消费循环:inbox → RouteDecision → dispatch → handler;outbox → 企微发送。 本文件职责:在 HTTP 进程内独立线程跑快路径,不与 LibreOffice 串队。 禁止:在回调请求线程调用本循环的阻塞逻辑;禁止 Redis 全局单例锁。 """ from __future__ import annotations import logging import threading from typing import Optional from agent.channel.outbox.sender import PERMANENT_SEND_ERRORS, WeComAppClient from agent.channel.queue import get_message_store from agent.config import Settings from agent.routing import IntentRouter from agent.routing.dispatch import dispatch_inbound logger = logging.getLogger(__name__) # 避免 httpx 把 access_token / secret 打进 INFO logging.getLogger("httpx").setLevel(logging.WARNING) class ChannelRuntime: """ 企微通道运行时:inbox/outbox 各一条守护线程。 启动时机:FastAPI lifespan;停止:进程退出。 inbox:IntentRouter.route(默认 stub)→ dispatch 查表;无业务分支森林。 """ def __init__(self, settings: Settings) -> None: self._settings = settings self._store = get_message_store() self._stop = threading.Event() self._threads: list[threading.Thread] = [] self._router = IntentRouter(allow_network=False) agent_id = int(settings.wecom_agent_id or "0") or 0 self._client = WeComAppClient( corp_id=settings.wecom_corp_id, secret=settings.wecom_secret, agent_id=agent_id, ) @property def store(self): return self._store def start(self) -> None: """启动 inbox/outbox 消费线程。""" self._stop.clear() self._threads = [ threading.Thread(target=self._inbox_loop, name="wecom-inbox", daemon=True), threading.Thread(target=self._outbox_loop, name="wecom-outbox", daemon=True), ] for t in self._threads: t.start() logger.info("ChannelRuntime 已启动(inbox/outbox 独立线程;dispatch 分发)") def stop(self) -> None: self._stop.set() def _inbox_loop(self) -> None: while not self._stop.is_set(): item = self._store.claim_inbound() if item is None: self._store.wait_inbox(1.0) continue try: decision = self._router.route( text=item.message.content or "", sender_id=item.message.sender_id or "", ) dispatch_inbound(item.message, decision) self._store.complete_inbound(item.id) except Exception as exc: # noqa: BLE001 logger.exception("inbox 处理失败 id=%s", item.id) self._store.complete_inbound(item.id, error=str(exc)) def _outbox_loop(self) -> None: while not self._stop.is_set(): item = self._store.claim_outbound() if item is None: self._store.wait_outbox(1.0) continue # 过期丢弃:超过 30 分钟的出站不再发(对齐「过期不计业务成功」) import time if time.time() - item.created_at > 1800: logger.warning("outbox 过期丢弃 id=%s", item.id) self._store.complete_outbound(item.id) continue result = self._client.send_text(touser=item.touser, content=item.content) if result.ok: self._store.complete_outbound(item.id) elif result.errcode in PERMANENT_SEND_ERRORS: logger.error( "outbox 永久失败不再重试 id=%s err=%s:%s", item.id, result.errcode, result.errmsg, ) self._store.complete_outbound( item.id, error=f"permanent:{result.errcode}:{result.errmsg}", permanent=True, ) else: self._store.complete_outbound( item.id, error=f"{result.errcode}:{result.errmsg}", ) _RUNTIME: Optional[ChannelRuntime] = None def get_channel_runtime() -> Optional[ChannelRuntime]: return _RUNTIME def start_channel_runtime(settings: Settings) -> ChannelRuntime: """幂等启动通道运行时。""" global _RUNTIME if _RUNTIME is None: _RUNTIME = ChannelRuntime(settings) _RUNTIME.start() return _RUNTIME