Kris
1f78c44b03
这是一个批量整理提交，包含以下主要改动： 1. 删除多处冗余的空行和未使用的导入 2. 修复文件末尾缺少换行符的问题 3. 调整部分模块的导入顺序与代码排版 4. 修复部分配置默认值与策略逻辑 5. 新增多个功能模块与辅助工具 6. 完善异常处理与日志记录 7. 修复速率限制、消息缓存、权限校验等逻辑bug 8. 废弃部分旧有API与配置项并添加警告提示
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from collections.abc import Awaitable, Callable
|
|
|
|
from yuxi.utils.logging_config import logger
|
|
|
|
|
|
SENTINEL = object()
|
|
|
|
|
|
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
|
|
self._dropped_count = 0
|
|
|
|
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
|
|
self._queue.put_nowait(SENTINEL)
|
|
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:
|
|
event_type = event.get("type", "unknown")
|
|
msg_id = event.get("msg_id", "N/A")
|
|
logger.warning(
|
|
f"[Yuanbao] Event queue full, dropping event: "
|
|
f"type={event_type} msg_id={msg_id}"
|
|
)
|
|
self._dropped_count += 1
|
|
|
|
async def _process_loop(self) -> None:
|
|
while self._running:
|
|
try:
|
|
event = await self._queue.get()
|
|
if event is SENTINEL:
|
|
break
|
|
if self._handler:
|
|
await self._handler(event)
|
|
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()
|