Kris
eb25707668
新增元宝（Yuanbao）渠道的完整适配器实现，包含以下核心模块： - 基础适配器与导出入口 - 协议编解码与WebSocket帧处理 - 会话管理与路由逻辑 - 事件队列与出站消息队列 - 消息格式转换与发送重试 - 安全审计与权限校验 - 配置映射与账户管理 - 视觉分析与工具函数 - 文档生成与设置向导
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from collections.abc import Awaitable, Callable
|
|
|
|
from yuxi.utils.logging_config import logger
|
|
|
|
|
|
class EventQueue:
|
|
def __init__(self, max_size: int = 10000):
|
|
self._queue: asyncio.Queue = asyncio.Queue(maxsize=max_size)
|
|
self._handler: Callable[[dict], Awaitable[None]] | None = None
|
|
self._task: asyncio.Task | None = None
|
|
self._running = False
|
|
|
|
async def start(self, handler: Callable[[dict], Awaitable[None]]) -> None:
|
|
self._handler = handler
|
|
self._running = True
|
|
self._task = asyncio.create_task(self._process_loop())
|
|
logger.info("[Yuanbao] Event queue started")
|
|
|
|
async def stop(self) -> None:
|
|
self._running = False
|
|
if self._task and not self._task.done():
|
|
self._task.cancel()
|
|
try:
|
|
await self._task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
self._task = None
|
|
logger.info("[Yuanbao] Event queue stopped")
|
|
|
|
async def enqueue(self, event: dict) -> None:
|
|
try:
|
|
self._queue.put_nowait(event)
|
|
except asyncio.QueueFull:
|
|
logger.warning("[Yuanbao] Event queue full, dropping event")
|
|
|
|
async def _process_loop(self) -> None:
|
|
while self._running:
|
|
try:
|
|
event = await asyncio.wait_for(self._queue.get(), timeout=1.0)
|
|
if self._handler:
|
|
await self._handler(event)
|
|
except TimeoutError:
|
|
continue
|
|
except asyncio.CancelledError:
|
|
break
|
|
except Exception as e:
|
|
logger.error(f"[Yuanbao] Event queue process error: {e}", exc_info=True)
|
|
|
|
@property
|
|
def size(self) -> int:
|
|
return self._queue.qsize()
|