aury-boot 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

aury/boot/infrastructure/storage/exceptions.py

@@ -0,0 +1,37 @@
+"""存储相关异常定义。
+
+Infrastructure 层异常，继承自 FoundationError。
+"""
+
+from __future__ import annotations
+
+from aury.boot.common.exceptions import FoundationError
+
+
+class StorageError(FoundationError):
+    """存储相关错误基类。
+    
+    所有存储相关的异常都应该继承此类。
+    """
+    
+    pass
+
+
+class StorageNotFoundError(StorageError):
+    """存储文件不存在错误。"""
+    
+    pass
+
+
+class StorageBackendError(StorageError):
+    """存储后端错误。"""
+    
+    pass
+
+
+__all__ = [
+    "StorageBackendError",
+    "StorageError",
+    "StorageNotFoundError",
+]
+

aury/boot/infrastructure/storage/factory.py

@@ -0,0 +1,88 @@
+"""存储工厂 - 注册机制。
+
+通过注册机制支持多种存储后端。
+"""
+
+from __future__ import annotations
+
+from typing import ClassVar
+
+from aury.boot.common.logging import logger
+
+from aury.sdk.storage.storage import IStorage, LocalStorage
+
+
+class StorageFactory:
+    """存储工厂 - 注册机制。
+    
+    类似缓存工厂的设计，通过注册机制支持多种后端。
+    
+    使用示例:
+        # 注册后端
+        StorageFactory.register("local", LocalStorage)
+        StorageFactory.register("s3", S3Storage)
+        
+        # 创建存储实例
+        storage = await StorageFactory.create("local", base_path="./storage")
+        storage = await StorageFactory.create("s3", access_key_id="...", ...)
+    """
+    
+    _backends: ClassVar[dict[str, type[IStorage]]] = {}
+    
+    @classmethod
+    def register(cls, name: str, backend_class: type[IStorage]) -> None:
+        """注册存储后端。
+        
+        Args:
+            name: 后端名称
+            backend_class: 后端类
+        """
+        cls._backends[name] = backend_class
+        logger.debug(f"注册存储后端: {name} -> {backend_class.__name__}")
+    
+    @classmethod
+    async def create(cls, backend_name: str, **config) -> IStorage:
+        """创建存储实例。
+        
+        Args:
+            backend_name: 后端名称
+            **config: 配置参数
+            
+        Returns:
+            IStorage: 存储实例
+            
+        Raises:
+            ValueError: 后端未注册
+        """
+        if backend_name not in cls._backends:
+            available = ", ".join(cls._backends.keys())
+            raise ValueError(
+                f"存储后端 '{backend_name}' 未注册。"
+                f"可用后端: {available}"
+            )
+        
+        backend_class = cls._backends[backend_name]
+        instance = backend_class(**config)
+        
+        # 如果后端需要初始化
+        if hasattr(instance, "initialize"):
+            await instance.initialize()
+        
+        logger.info(f"创建存储实例: {backend_name}")
+        return instance
+    
+    @classmethod
+    def get_registered(cls) -> list[str]:
+        """获取已注册的后端名称。"""
+        return list(cls._backends.keys())
+
+
+# 注册默认后端（本地）
+StorageFactory.register("local", LocalStorage)
+# 说明：S3/COS/OSS 请直接使用 StorageManager + StorageConfig 或 SDK 中的 S3Storage
+
+
+__all__ = [
+    "StorageFactory",
+]
+

aury/boot/infrastructure/tasks/__init__.py

@@ -0,0 +1,24 @@
+"""异步任务管理器模块。
+
+提供统一的任务队列接口，支持条件注册以避免 API 模式下重复注册。
+"""
+
+from .constants import TaskQueueName, TaskRunMode
+from .exceptions import (
+    TaskError,
+    TaskExecutionError,
+    TaskQueueError,
+)
+from .manager import TaskManager, TaskProxy, conditional_actor
+
+__all__ = [
+    "TaskError",
+    "TaskExecutionError",
+    "TaskManager",
+    "TaskProxy",
+    "TaskQueueError",
+    "TaskQueueName",
+    "TaskRunMode",
+    "conditional_actor",
+]
+

aury/boot/infrastructure/tasks/config.py

@@ -0,0 +1,45 @@
+"""任务队列配置。
+
+Infrastructure 层配置，不依赖 application 层。
+"""
+
+from __future__ import annotations
+
+from pydantic import Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class TaskConfig(BaseSettings):
+    """任务队列基础设施配置。
+    
+    Infrastructure 层直接使用的任务队列配置。
+    
+    环境变量前缀: TASK_
+    示例: TASK_BROKER_URL, TASK_MAX_RETRIES
+    """
+    
+    broker_url: str = Field(
+        default="redis://localhost:6379/0",
+        description="任务队列Broker URL"
+    )
+    max_retries: int = Field(
+        default=3,
+        description="最大重试次数"
+    )
+    time_limit: int = Field(
+        default=3600000,
+        description="任务执行时间限制（毫秒）"
+    )
+    
+    model_config = SettingsConfigDict(
+        env_prefix="TASK_",
+        case_sensitive=False,
+    )
+
+
+__all__ = [
+    "TaskConfig",
+]
+
+
+

aury/boot/infrastructure/tasks/constants.py

@@ -0,0 +1,37 @@
+"""任务队列常量。
+
+任务队列模块内部的常量定义。
+"""
+
+from __future__ import annotations
+
+from enum import Enum
+
+
+class TaskQueueName(str, Enum):
+    """任务队列名称常量。
+
+    框架提供的通用队列名称。业务相关的队列名称应由用户自行定义。
+    """
+
+    DEFAULT = "default"
+    HIGH_PRIORITY = "high_priority"
+    LOW_PRIORITY = "low_priority"
+
+
+class TaskRunMode(str, Enum):
+    """任务运行模式（任务队列模块内部使用）。
+
+    用于控制任务装饰器的行为：
+    - WORKER: Worker 模式（执行者），正常注册为 actor，执行任务
+    - PRODUCER: Producer 模式（生产者），返回 TaskProxy，不注册但可以发送消息
+    """
+
+    WORKER = "worker"
+    PRODUCER = "producer"
+
+
+__all__ = [
+    "TaskQueueName",
+    "TaskRunMode",
+]

aury/boot/infrastructure/tasks/exceptions.py

@@ -0,0 +1,37 @@
+"""任务队列相关异常定义。
+
+Infrastructure 层异常，继承自 FoundationError。
+"""
+
+from __future__ import annotations
+
+from aury.boot.common.exceptions import FoundationError
+
+
+class TaskError(FoundationError):
+    """任务队列相关错误基类。
+    
+    所有任务队列相关的异常都应该继承此类。
+    """
+    
+    pass
+
+
+class TaskQueueError(TaskError):
+    """任务队列错误。"""
+    
+    pass
+
+
+class TaskExecutionError(TaskError):
+    """任务执行错误。"""
+    
+    pass
+
+
+__all__ = [
+    "TaskError",
+    "TaskExecutionError",
+    "TaskQueueError",
+]
+