mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-18 10:00:40 +08:00
dcaaf6286a
- Add 100+ new test files covering provider sources, platform adapters, agent runners, star/plugin system, knowledge base, core utils, pipeline, computer tools, builtin commands, and dashboard routes - Fix Python type errors in sqlite.py (col() wrappers for SQLModel), astr_agent_tool_exec, core_lifecycle, star context/manager - Fix TypeScript strict mode errors across 30+ dashboard Vue files - Add import smoke tests, auth roundtrip, startup tests - Add compile-all check and CLI entry test for AUR compatibility - Restore BotMessageAccumulator and helpers lost in merge
70 lines
2.8 KiB
Python
70 lines
2.8 KiB
Python
"""事件总线, 用于处理事件的分发和处理
|
|
事件总线是一个异步队列, 用于接收各种消息事件, 并将其发送到Scheduler调度器进行处理
|
|
其中包含了一个无限循环的调度函数, 用于从事件队列中获取新的事件, 并创建一个新的异步任务来执行管道调度器的处理逻辑
|
|
|
|
class:
|
|
EventBus: 事件总线, 用于处理事件的分发和处理
|
|
|
|
工作流程:
|
|
1. 维护一个异步队列, 来接受各种消息事件
|
|
2. 无限循环的调度函数, 从事件队列中获取新的事件, 打印日志并创建一个新的异步任务来执行管道调度器的处理逻辑
|
|
"""
|
|
|
|
import asyncio
|
|
from asyncio import Queue
|
|
from typing import Any
|
|
|
|
from astrbot.core import logger
|
|
from astrbot.core.astrbot_config_mgr import AstrBotConfigManager
|
|
from astrbot.core.pipeline.scheduler import PipelineScheduler
|
|
|
|
from .platform import AstrMessageEvent
|
|
|
|
|
|
class EventBus:
|
|
"""用于处理事件的分发和处理"""
|
|
|
|
def __init__(
|
|
self,
|
|
event_queue: Queue[Any],
|
|
pipeline_scheduler_mapping: dict[str, PipelineScheduler],
|
|
astrbot_config_mgr: AstrBotConfigManager,
|
|
) -> None:
|
|
self.event_queue = event_queue # 事件队列
|
|
# abconf uuid -> scheduler
|
|
self.pipeline_scheduler_mapping = pipeline_scheduler_mapping
|
|
self.astrbot_config_mgr = astrbot_config_mgr
|
|
|
|
async def dispatch(self) -> None:
|
|
while True:
|
|
event: AstrMessageEvent = await self.event_queue.get()
|
|
conf_info = self.astrbot_config_mgr.get_conf_info(event.unified_msg_origin)
|
|
conf_id = conf_info.get("id", "")
|
|
conf_name = conf_info.get("name", conf_id)
|
|
self._print_event(event, conf_name)
|
|
scheduler = self.pipeline_scheduler_mapping.get(conf_id)
|
|
if not scheduler:
|
|
logger.error(
|
|
f"PipelineScheduler not found for id: {conf_id}, event ignored.",
|
|
)
|
|
continue
|
|
asyncio.create_task(scheduler.execute(event)) # noqa: RUF006
|
|
|
|
def _print_event(self, event: AstrMessageEvent, conf_name: str) -> None:
|
|
"""用于记录事件信息
|
|
|
|
Args:
|
|
event (AstrMessageEvent): 事件对象
|
|
|
|
"""
|
|
# 如果有发送者名称: [平台名] 发送者名称/发送者ID: 消息概要
|
|
if event.get_sender_name():
|
|
logger.info(
|
|
f"[{conf_name}] [{event.get_platform_id()}({event.get_platform_name()})] {event.get_sender_name()}/{event.get_sender_id()}: {event.get_message_outline()}",
|
|
)
|
|
# 没有发送者名称: [平台名] 发送者ID: 消息概要
|
|
else:
|
|
logger.info(
|
|
f"[{conf_name}] [{event.get_platform_id()}({event.get_platform_name()})] {event.get_sender_id()}: {event.get_message_outline()}",
|
|
)
|