Soulter
28 lines
1010 B
Python
28 lines
1010 B
Python
import asyncio
|
|
from asyncio import Queue
|
|
from astrbot.core.pipeline.scheduler import PipelineScheduler
|
|
from astrbot.core import logger
|
|
from .platform import AstrMessageEvent
|
|
|
|
|
|
class EventBus:
|
|
def __init__(self, event_queue: Queue, pipeline_scheduler: PipelineScheduler):
|
|
self.event_queue = event_queue
|
|
self.pipeline_scheduler = pipeline_scheduler
|
|
|
|
async def dispatch(self):
|
|
while True:
|
|
event: AstrMessageEvent = await self.event_queue.get()
|
|
self._print_event(event)
|
|
asyncio.create_task(self.pipeline_scheduler.execute(event))
|
|
|
|
def _print_event(self, event: AstrMessageEvent):
|
|
if event.get_sender_name():
|
|
logger.info(
|
|
f"[{event.get_platform_name()}] {event.get_sender_name()}/{event.get_sender_id()}: {event.get_message_outline()}"
|
|
)
|
|
else:
|
|
logger.info(
|
|
f"[{event.get_platform_name()}] {event.get_sender_id()}: {event.get_message_outline()}"
|
|
)
|