217 lines
7.5 KiB
Python
217 lines
7.5 KiB
Python
"""
|
||
核心事件总线
|
||
|
||
面向最终架构的事件系统：
|
||
- 内部 handler 直接注册 async callable
|
||
- IPC 插件通过 plugin_runtime 桥接
|
||
- 不依赖任何插件基类
|
||
"""
|
||
|
||
import asyncio
|
||
from typing import Any, Awaitable, Callable, Dict, List, Optional, Tuple, TYPE_CHECKING
|
||
|
||
from src.common.logger import get_logger
|
||
from src.core.types import EventType, MaiMessages
|
||
|
||
if TYPE_CHECKING:
|
||
from src.common.data_models.llm_data_model import LLMGenerationDataModel
|
||
|
||
logger = get_logger("event_bus")
|
||
|
||
# Handler 签名：接收 MaiMessages，返回 (continue, modified_message)
|
||
EventHandler = Callable[[Optional[MaiMessages]], Awaitable[Tuple[bool, Optional[MaiMessages]]]]
|
||
|
||
|
||
class EventBus:
|
||
"""核心事件总线
|
||
|
||
支持两种 handler：
|
||
- 拦截型（intercept=True）：同步顺序执行，可修改消息、可中断流程
|
||
- 非拦截型（intercept=False）：异步并发执行，fire-and-forget
|
||
|
||
handler 是纯 async callable，不需要继承任何基类。
|
||
"""
|
||
|
||
def __init__(self):
|
||
# event_type -> [(handler, name, weight, intercept)]
|
||
self._handlers: Dict[EventType | str, List[_HandlerEntry]] = {}
|
||
self._running_tasks: Dict[str, List[asyncio.Task]] = {}
|
||
|
||
# 预注册所有内置事件类型
|
||
for event in EventType:
|
||
self._handlers[event] = []
|
||
|
||
def subscribe(
|
||
self,
|
||
event_type: EventType | str,
|
||
handler: EventHandler,
|
||
name: str,
|
||
weight: int = 0,
|
||
intercept: bool = False,
|
||
) -> None:
|
||
"""注册事件 handler
|
||
|
||
Args:
|
||
event_type: 事件类型
|
||
handler: async callable，签名 (Optional[MaiMessages]) -> (bool, Optional[MaiMessages])
|
||
name: handler 标识名
|
||
weight: 权重，越大越先执行
|
||
intercept: 是否为拦截型（同步执行，可中断流程）
|
||
"""
|
||
if event_type not in self._handlers:
|
||
self._handlers[event_type] = []
|
||
|
||
entry = _HandlerEntry(handler=handler, name=name, weight=weight, intercept=intercept)
|
||
self._handlers[event_type].append(entry)
|
||
self._handlers[event_type].sort(key=lambda e: e.weight, reverse=True)
|
||
logger.debug(f"注册事件 handler: {name} -> {event_type} (weight={weight}, intercept={intercept})")
|
||
|
||
def unsubscribe(self, event_type: EventType | str, name: str) -> bool:
|
||
"""取消注册事件 handler"""
|
||
handlers = self._handlers.get(event_type, [])
|
||
for i, entry in enumerate(handlers):
|
||
if entry.name == name:
|
||
del handlers[i]
|
||
logger.debug(f"取消注册事件 handler: {name} <- {event_type}")
|
||
return True
|
||
return False
|
||
|
||
async def emit(
|
||
self,
|
||
event_type: EventType | str,
|
||
message: Optional[MaiMessages] = None,
|
||
) -> Tuple[bool, Optional[MaiMessages]]:
|
||
"""触发事件
|
||
|
||
按权重顺序执行所有 handler：
|
||
- 拦截型 handler 同步执行，可修改消息和中断流程
|
||
- 非拦截型 handler 异步 fire-and-forget
|
||
|
||
Args:
|
||
event_type: 事件类型
|
||
message: 事件消息（可选）
|
||
|
||
Returns:
|
||
(continue_flag, modified_message)
|
||
- continue_flag: False 表示某个拦截型 handler 要求中断
|
||
- modified_message: 被拦截型 handler 修改后的消息
|
||
"""
|
||
handlers = self._handlers.get(event_type, [])
|
||
if not handlers:
|
||
return True, None
|
||
|
||
continue_flag = True
|
||
current_message = message.deepcopy() if message else None
|
||
|
||
for entry in handlers:
|
||
if entry.intercept:
|
||
try:
|
||
should_continue, modified = await entry.handler(current_message)
|
||
if modified is not None:
|
||
current_message = modified
|
||
if not should_continue:
|
||
continue_flag = False
|
||
break
|
||
except Exception as e:
|
||
logger.error(f"拦截型 handler {entry.name} 执行异常: {e}", exc_info=True)
|
||
else:
|
||
self._fire_and_forget(entry, event_type, current_message)
|
||
|
||
# 桥接到 IPC 插件运行时
|
||
continue_flag, current_message = await self._bridge_to_ipc_runtime(
|
||
event_type, continue_flag, current_message
|
||
)
|
||
|
||
return continue_flag, current_message
|
||
|
||
async def cancel_handler_tasks(self, handler_name: str) -> None:
|
||
"""取消某个 handler 的所有运行中任务"""
|
||
tasks = self._running_tasks.pop(handler_name, [])
|
||
remaining = [t for t in tasks if not t.done()]
|
||
if remaining:
|
||
for t in remaining:
|
||
t.cancel()
|
||
await asyncio.gather(*remaining, return_exceptions=True)
|
||
logger.info(f"已取消 handler {handler_name} 的 {len(remaining)} 个任务")
|
||
|
||
# --- 内部方法 ---
|
||
|
||
def _fire_and_forget(
|
||
self,
|
||
entry: "_HandlerEntry",
|
||
event_type: EventType | str,
|
||
message: Optional[MaiMessages],
|
||
) -> None:
|
||
"""创建异步任务执行非拦截型 handler"""
|
||
try:
|
||
task = asyncio.create_task(entry.handler(message))
|
||
task.set_name(entry.name)
|
||
task.add_done_callback(lambda t: self._task_done_callback(t, entry.name))
|
||
self._running_tasks.setdefault(entry.name, []).append(task)
|
||
except Exception as e:
|
||
logger.error(f"创建 handler 任务 {entry.name} 失败: {e}", exc_info=True)
|
||
|
||
def _task_done_callback(self, task: asyncio.Task, handler_name: str) -> None:
|
||
"""异步任务完成回调"""
|
||
try:
|
||
if task.cancelled():
|
||
return
|
||
exc = task.exception()
|
||
if exc:
|
||
logger.error(f"handler {handler_name} 异步任务异常: {exc}")
|
||
except Exception:
|
||
pass
|
||
finally:
|
||
task_list = self._running_tasks.get(handler_name, [])
|
||
try:
|
||
task_list.remove(task)
|
||
except ValueError:
|
||
pass
|
||
|
||
async def _bridge_to_ipc_runtime(
|
||
self,
|
||
event_type: EventType | str,
|
||
continue_flag: bool,
|
||
message: Optional[MaiMessages],
|
||
) -> Tuple[bool, Optional[MaiMessages]]:
|
||
"""将事件桥接到 IPC 插件运行时"""
|
||
if not continue_flag:
|
||
return continue_flag, message
|
||
|
||
try:
|
||
from src.plugin_runtime.integration import get_plugin_runtime_manager
|
||
|
||
prm = get_plugin_runtime_manager()
|
||
if not prm.is_running:
|
||
return continue_flag, message
|
||
|
||
event_value = event_type.value if isinstance(event_type, EventType) else str(event_type)
|
||
message_dict = message.to_dict() if message and hasattr(message, "to_dict") else None
|
||
|
||
new_continue, _ = await prm.bridge_event(
|
||
event_type_value=event_value,
|
||
message_dict=message_dict,
|
||
)
|
||
if not new_continue:
|
||
continue_flag = False
|
||
except Exception as e:
|
||
logger.warning(f"桥接事件到 IPC 运行时失败: {e}")
|
||
|
||
return continue_flag, message
|
||
|
||
|
||
class _HandlerEntry:
|
||
"""内部 handler 条目"""
|
||
|
||
__slots__ = ("handler", "name", "weight", "intercept")
|
||
|
||
def __init__(self, handler: EventHandler, name: str, weight: int, intercept: bool):
|
||
self.handler = handler
|
||
self.name = name
|
||
self.weight = weight
|
||
self.intercept = intercept
|
||
|
||
|
||
# 全局单例
|
||
event_bus = EventBus()
|