aury-boot 0.0.23__tar.gz → 0.0.25__tar.gz

{aury_boot-0.0.23 → aury_boot-0.0.25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aury-boot
-Version: 0.0.23
+Version: 0.0.25
 Summary: Aury Boot - 基于 FastAPI 生态的企业级 API 开发框架
 Requires-Python: >=3.13
 Requires-Dist: alembic>=1.17.2

{aury_boot-0.0.23 → aury_boot-0.0.25}/aury/boot/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.0.23'
-__version_tuple__ = version_tuple = (0, 0, 23)
+__version__ = version = '0.0.25'
+__version_tuple__ = version_tuple = (0, 0, 25)
 
 __commit_id__ = commit_id = None

{aury_boot-0.0.23 → aury_boot-0.0.25}/aury/boot/application/app/components.py

@@ -15,6 +15,7 @@ from aury.boot.application.constants import ComponentName, ServiceType
 from aury.boot.application.migrations import MigrationManager
 from aury.boot.common.logging import logger
 from aury.boot.infrastructure.cache import CacheManager
+from aury.boot.infrastructure.channel import ChannelManager
 from aury.boot.infrastructure.database import DatabaseManager
 from aury.boot.infrastructure.events import EventBusManager
 from aury.boot.infrastructure.mq import MQManager
@@ -505,6 +506,49 @@ class MessageQueueComponent(Component):
                 logger.warning(f"消息队列 [{name}] 关闭失败: {e}")
 
 
+class ChannelComponent(Component):
+    """流式通道组件。
+    
+    用于 SSE（Server-Sent Events）和实时通信场景，支持 memory 和 redis 后端。
+    """
+
+    name = ComponentName.CHANNEL
+    enabled = True
+    depends_on: ClassVar[list[str]] = []
+
+    def can_enable(self, config: BaseConfig) -> bool:
+        """当配置了 Channel 实例时启用。"""
+        return self.enabled and bool(config.get_channels())
+
+    async def setup(self, app: FoundationApp, config: BaseConfig) -> None:
+        """初始化流式通道。"""
+        channel_configs = config.get_channels()
+        if not channel_configs:
+            logger.debug("未配置 Channel 实例，跳过通道初始化")
+            return
+        
+        for name, ch_config in channel_configs.items():
+            try:
+                channel_manager = ChannelManager.get_instance(name)
+                if not channel_manager.is_initialized:
+                    await channel_manager.initialize(
+                        backend=ch_config.backend,
+                        url=ch_config.url,
+                    )
+            except Exception as e:
+                logger.warning(f"通道 [{name}] 初始化失败（非关键）: {e}")
+
+    async def teardown(self, app: FoundationApp) -> None:
+        """关闭所有通道实例。"""
+        for name in list(ChannelManager._instances.keys()):
+            try:
+                channel_manager = ChannelManager.get_instance(name)
+                if channel_manager.is_initialized:
+                    await channel_manager.cleanup()
+            except Exception as e:
+                logger.warning(f"通道 [{name}] 关闭失败: {e}")
+
+
 class EventBusComponent(Component):
     """事件总线组件。
     
@@ -555,6 +599,7 @@ FoundationApp.components = [
     StorageComponent,
     TaskComponent,
     MessageQueueComponent,
+    ChannelComponent,
     EventBusComponent,
     SchedulerComponent,
 ]
@@ -563,6 +608,7 @@ FoundationApp.components = [
 __all__ = [
     "AdminConsoleComponent",
     "CacheComponent",
+    "ChannelComponent",
     "DatabaseComponent",
     "EventBusComponent",
     "MessageQueueComponent",

{aury_boot-0.0.23 → aury_boot-0.0.25}/aury/boot/application/constants/components.py

@@ -41,6 +41,9 @@ class ComponentName(str, Enum):
     # 事件总线组件
     EVENT_BUS = "event_bus"
 
+    # 流式通道组件
+    CHANNEL = "channel"
+
     # 迁移组件
     MIGRATIONS = "migrations"
 

{aury_boot-0.0.23 → aury_boot-0.0.25}/aury/boot/commands/templates/project/aury_docs/13-channel.md.tpl

@@ -20,10 +20,7 @@ notification_channel = ChannelManager.get_instance("notification")
 await sse_channel.initialize(backend="memory")
 
 # Redis 后端（多进程/分布式）
-from aury.boot.infrastructure.clients.redis import RedisClient
-redis_client = RedisClient.get_instance()
-await redis_client.initialize(url="redis://localhost:6379/0")
-await notification_channel.initialize(backend="redis", redis_client=redis_client)
+await notification_channel.initialize(backend="redis", url="redis://localhost:6379/0")
 ```
 
 ## 13.2 发布和订阅（Topic 管理）
@@ -85,7 +82,7 @@ async def send_notification(user_id: str, message: str):
 async def space_events(space_id: str):
     \"\"\"订阅空间下所有事件。\"\"\"
     async def event_generator():
-        # 使用 psubscribe 订阅 space:{id}:* 下所有事件
+        # 使用 psubscribe 订阅 space:{{id}}:* 下所有事件
         async for msg in sse_channel.psubscribe(f"space:{{space_id}}:*"):
             yield msg.to_sse()  # 自动转换为 SSE 格式
     
@@ -125,8 +122,8 @@ notify_channel = ChannelManager.get_instance("notify")     # 系统通知
 
 # 分别初始化（可使用不同后端）
 await sse_channel.initialize(backend="memory")  # 单进程即可
-await chat_channel.initialize(backend="redis", redis_client=redis_client)  # 需要跨进程
-await notify_channel.initialize(backend="redis", redis_client=redis_client)
+await chat_channel.initialize(backend="redis", url="redis://localhost:6379/3")  # 需要跨进程
+await notify_channel.initialize(backend="redis", url="redis://localhost:6379/4")
 ```
 
 ## 13.6 环境变量

{aury_boot-0.0.23 → aury_boot-0.0.25}/aury/boot/infrastructure/channel/manager.py

@@ -6,7 +6,6 @@
 from __future__ import annotations
 
 from collections.abc import AsyncIterator
-from typing import TYPE_CHECKING
 
 from aury.boot.common.logging import logger
 
@@ -14,9 +13,6 @@ from .backends.memory import MemoryChannel
 from .backends.redis import RedisChannel
 from .base import ChannelBackend, ChannelMessage, IChannel
 
-if TYPE_CHECKING:
-    from aury.boot.infrastructure.clients.redis import RedisClient
-
 
 class ChannelManager:
     """通道管理器（命名多实例）。
@@ -55,6 +51,8 @@ class ChannelManager:
         self._backend: IChannel | None = None
         self._backend_type: ChannelBackend | None = None
         self._initialized: bool = False
+        self._redis_client = None
+        self._url: str | None = None
 
     @classmethod
     def get_instance(cls, name: str = "default") -> ChannelManager:
@@ -88,14 +86,14 @@ class ChannelManager:
         self,
         backend: ChannelBackend | str = ChannelBackend.MEMORY,
         *,
-        redis_client: RedisClient | None = None,
+        url: str | None = None,
         max_subscribers: int = 1000,
     ) -> ChannelManager:
         """初始化通道（链式调用）。
 
         Args:
             backend: 后端类型
-            redis_client: Redis 客户端（当 backend=redis 时需要）
+            url: Redis 连接 URL（当 backend=redis 时需要）
             max_subscribers: 内存后端的最大订阅者数量
 
         Returns:
@@ -110,13 +108,19 @@ class ChannelManager:
             backend = ChannelBackend(backend.lower())
 
         self._backend_type = backend
+        self._url = url
 
         if backend == ChannelBackend.MEMORY:
             self._backend = MemoryChannel(max_subscribers=max_subscribers)
         elif backend == ChannelBackend.REDIS:
-            if redis_client is None:
-                raise ValueError("Redis 通道需要提供 redis_client 参数")
-            self._backend = RedisChannel(redis_client)
+            if url is None:
+                raise ValueError("Redis 通道需要提供 url 参数")
+            # 内部创建 RedisClient
+            from aury.boot.infrastructure.clients.redis import RedisClient
+
+            self._redis_client = RedisClient()
+            await self._redis_client.configure(url=url).initialize()
+            self._backend = RedisChannel(self._redis_client)
         else:
             raise ValueError(f"不支持的通道后端: {backend}")
 
@@ -214,8 +218,11 @@ class ChannelManager:
         if self._backend:
             await self._backend.close()
             self._backend = None
-            self._initialized = False
-            logger.info(f"通道管理器 [{self.name}] 已关闭")
+        if self._redis_client:
+            await self._redis_client.cleanup()
+            self._redis_client = None
+        self._initialized = False
+        logger.info(f"通道管理器 [{self.name}] 已关闭")
 
     def __repr__(self) -> str:
         """字符串表示。"""