122 lines
3.9 KiB
Python
122 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
通信适配器抽象基类
|
||
所有中间件适配器必须继承此类并实现全部抽象方法
|
||
遵循 Factory + Strategy 模式
|
||
"""
|
||
|
||
import asyncio
|
||
import logging
|
||
from abc import ABC, abstractmethod
|
||
from dataclasses import dataclass
|
||
from typing import Callable, Awaitable, List, Optional
|
||
|
||
from uas.comm.message import Message
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 消息处理器类型：同步 or 异步均可
|
||
MessageHandler = Callable[[Message], Optional[Awaitable[None]]]
|
||
|
||
|
||
@dataclass
|
||
class ConnectionConfig:
|
||
"""通用连接配置（由 URL 解析填充）"""
|
||
scheme: str # redis / kafka / dds
|
||
host: str = "localhost"
|
||
port: int = 0
|
||
username: str = ""
|
||
password: str = ""
|
||
database: str = "" # redis db index / kafka group id / dds domain
|
||
options: dict = None # URL query 参数
|
||
|
||
def __post_init__(self):
|
||
if self.options is None:
|
||
self.options = {}
|
||
|
||
|
||
class CommBaseAdapter(ABC):
|
||
"""
|
||
所有通信适配器的抽象基类
|
||
子类只需实现以下 5 个核心方法
|
||
"""
|
||
|
||
def __init__(self, config: ConnectionConfig):
|
||
self.config = config
|
||
self.logger = logging.getLogger(self.__class__.__name__)
|
||
self._running = False
|
||
self._handlers: dict[str, List[MessageHandler]] = {}
|
||
|
||
# ── 生命周期 ──────────────────────────────────────────────────
|
||
|
||
@abstractmethod
|
||
async def connect(self) -> None:
|
||
"""建立连接"""
|
||
...
|
||
|
||
@abstractmethod
|
||
async def disconnect(self) -> None:
|
||
"""断开连接"""
|
||
...
|
||
|
||
@property
|
||
def is_connected(self) -> bool:
|
||
return self._running
|
||
|
||
# ── 核心功能 ──────────────────────────────────────────────────
|
||
|
||
@abstractmethod
|
||
async def publish(self, message: Message) -> bool:
|
||
"""
|
||
发布消息到指定 topic
|
||
:return: True 表示发送成功
|
||
"""
|
||
...
|
||
|
||
@abstractmethod
|
||
async def subscribe(self, topic: str, handler: MessageHandler) -> None:
|
||
"""
|
||
订阅 topic，注册消息处理回调
|
||
支持通配符（由各适配器自行实现）
|
||
"""
|
||
...
|
||
|
||
@abstractmethod
|
||
async def unsubscribe(self, topic: str) -> None:
|
||
"""取消订阅"""
|
||
...
|
||
|
||
# ── 可选扩展 ──────────────────────────────────────────────────
|
||
|
||
async def request(self, message: Message, timeout: float = 5.0) -> Optional[Message]:
|
||
"""
|
||
Request/Reply 模式（默认未实现，子类可覆盖）
|
||
"""
|
||
raise NotImplementedError(f"{self.__class__.__name__} 不支持 Request/Reply 模式")
|
||
|
||
async def health_check(self) -> bool:
|
||
"""健康检查，默认返回连接状态"""
|
||
return self._running
|
||
|
||
# ── 内部工具 ──────────────────────────────────────────────────
|
||
|
||
def _register_handler(self, topic: str, handler: MessageHandler):
|
||
self._handlers.setdefault(topic, []).append(handler)
|
||
|
||
async def _dispatch(self, message: Message):
|
||
"""将消息分发给所有匹配的 handler"""
|
||
handlers = self._handlers.get(message.topic, [])
|
||
for handler in handlers:
|
||
try:
|
||
result = handler(message)
|
||
if asyncio.iscoroutine(result):
|
||
await result
|
||
except Exception as e:
|
||
self.logger.error(f"Handler 执行异常 topic={message.topic}: {e}", exc_info=True)
|
||
|
||
async def __aenter__(self):
|
||
await self.connect()
|
||
return self
|
||
|
||
async def __aexit__(self, *args):
|
||
await self.disconnect() |