aury-boot 0.0.24__tar.gz → 0.0.26__tar.gz

{aury_boot-0.0.24 → aury_boot-0.0.26}/PKG-INFO

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

{aury_boot-0.0.24 → aury_boot-0.0.26}/aury/boot/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.0.24'
-__version_tuple__ = version_tuple = (0, 0, 24)
+__version__ = version = '0.0.26'
+__version_tuple__ = version_tuple = (0, 0, 26)
 
 __commit_id__ = commit_id = None

{aury_boot-0.0.24 → aury_boot-0.0.26}/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.24 → aury_boot-0.0.26}/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.24 → aury_boot-0.0.26}/aury/boot/commands/templates/project/aury_docs/13-channel.md.tpl

@@ -9,10 +9,34 @@
 
 ## 13.1 基本用法
 
+### 通过环境变量自动初始化（推荐）
+
+配置环境变量后，`ChannelComponent` 会在应用启动时自动初始化：
+
+```bash
+# .env
+CHANNEL__SSE__BACKEND=memory
+CHANNEL__NOTIFICATION__BACKEND=redis
+CHANNEL__NOTIFICATION__URL=redis://localhost:6379/3
+```
+
+```python
+from aury.boot.infrastructure.channel import ChannelManager
+
+# 直接获取已初始化的实例
+sse_channel = ChannelManager.get_instance("sse")
+notification_channel = ChannelManager.get_instance("notification")
+
+# 直接使用
+await sse_channel.publish("user:123", {{"event": "hello"}})
+```
+
+### 手动初始化
+
 ```python
 from aury.boot.infrastructure.channel import ChannelManager
 
-# 命名多实例（推荐）- 不同业务场景使用不同实例
+# 命名多实例 - 不同业务场景使用不同实例
 sse_channel = ChannelManager.get_instance("sse")
 notification_channel = ChannelManager.get_instance("notification")
 
@@ -82,7 +106,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 格式
     
@@ -112,21 +136,7 @@ await sse_channel.publish(f"space:{{space_id}}:comment_added", {{
 
 Redis 后端使用 Redis 原生 `PSUBSCRIBE`，内存后端使用 `fnmatch` 实现。
 
-## 13.5 应用场景示例
-
-```python
-# 不同业务场景使用不同的命名实例
-sse_channel = ChannelManager.get_instance("sse")           # 前端 SSE 推送
-chat_channel = ChannelManager.get_instance("chat")         # 聊天消息
-notify_channel = ChannelManager.get_instance("notify")     # 系统通知
-
-# 分别初始化（可使用不同后端）
-await sse_channel.initialize(backend="memory")  # 单进程即可
-await chat_channel.initialize(backend="redis", url="redis://localhost:6379/3")  # 需要跨进程
-await notify_channel.initialize(backend="redis", url="redis://localhost:6379/4")
-```
-
-## 13.6 环境变量
+## 13.5 环境变量
 
 ```bash
 # 默认实例