0b7fc29ac4
Co-authored-by: Dt8333 <25431943+Dt8333@users.noreply.github.com> Co-authored-by: Soulter <905617992@qq.com>
62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
"""事件总线, 用于处理事件的分发和处理
|
|
事件总线是一个异步队列, 用于接收各种消息事件, 并将其发送到Scheduler调度器进行处理
|
|
其中包含了一个无限循环的调度函数, 用于从事件队列中获取新的事件, 并创建一个新的异步任务来执行管道调度器的处理逻辑
|
|
|
|
class:
|
|
EventBus: 事件总线, 用于处理事件的分发和处理
|
|
|
|
工作流程:
|
|
1. 维护一个异步队列, 来接受各种消息事件
|
|
2. 无限循环的调度函数, 从事件队列中获取新的事件, 打印日志并创建一个新的异步任务来执行管道调度器的处理逻辑
|
|
"""
|
|
|
|
import asyncio
|
|
from asyncio import Queue
|
|
|
|
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,
|
|
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):
|
|
while True:
|
|
event: AstrMessageEvent = await self.event_queue.get()
|
|
conf_info = self.astrbot_config_mgr.get_conf_info(event.unified_msg_origin)
|
|
self._print_event(event, conf_info["name"])
|
|
scheduler = self.pipeline_scheduler_mapping.get(conf_info["id"])
|
|
asyncio.create_task(scheduler.execute(event))
|
|
|
|
def _print_event(self, event: AstrMessageEvent, conf_name: str):
|
|
"""用于记录事件信息
|
|
|
|
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()}",
|
|
)
|