aury-boot 0.0.34__tar.gz → 0.0.36__tar.gz

{aury_boot-0.0.34 → aury_boot-0.0.36}/PKG-INFO

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

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.0.34'
-__version_tuple__ = version_tuple = (0, 0, 34)
+__version__ = version = '0.0.36'
+__version_tuple__ = version_tuple = (0, 0, 36)
 
 __commit_id__ = commit_id = None

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/application/app/base.py

@@ -280,7 +280,7 @@ class FoundationApp(FastAPI):
         title: str = "Aury Service",
         version: str = "1.0.0",
         description: str | None = None,
-        intercept_loggers: list[str] | None = None,
+        logger_levels: list[tuple[str, str]] | None = None,
         **kwargs: Any,
     ) -> None:
         """初始化应用。
@@ -290,9 +290,8 @@ class FoundationApp(FastAPI):
             title: 应用标题
             version: 应用版本
             description: 应用描述
-            intercept_loggers: 额外需要拦截的标准 logging logger 名称列表，
-                会追加到框架默认列表 (uvicorn, sqlalchemy.engine 等)。
-                也可通过配置 LOG__INTERCEPT_LOGGERS 设置。
+            logger_levels: 需要设置特定级别的 logger 列表，格式: [("name", "LEVEL"), ...]
+                例如: [("sse_starlette", "WARNING"), ("httpx", "INFO")]
             **kwargs: 传递给 FastAPI 的其他参数
         """
         # 加载配置
@@ -304,11 +303,6 @@ class FoundationApp(FastAPI):
         frame = sys._getframe(1)
         self._caller_module = frame.f_globals.get("__name__", "__main__")
 
-        # 合并 intercept_loggers：参数 + 配置
-        merged_intercept = list(config.log.intercept_loggers)
-        if intercept_loggers:
-            merged_intercept.extend(intercept_loggers)
-
         # 初始化日志（必须在其他操作之前）
         setup_logging(
             log_level=config.log.level,
@@ -318,7 +312,7 @@ class FoundationApp(FastAPI):
             retention_days=config.log.retention_days,
             enable_file_rotation=config.log.enable_file_rotation,
             enable_console=config.log.enable_console,
-            intercept_loggers=merged_intercept,
+            logger_levels=logger_levels,
         )
         
         # 注册 access 日志（HTTP 请求日志）

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/application/config/settings.py

@@ -401,14 +401,6 @@ class LogSettings(BaseModel):
         default=False,
         description="是否记录 WebSocket 消息内容（注意性能和敏感数据）"
     )
-    intercept_loggers: list[str] = Field(
-        default_factory=list,
-        description=(
-            "额外需要由 loguru 接管的标准 logging logger 名称列表。"
-            "框架默认已拦截 uvicorn、uvicorn.error、uvicorn.access、sqlalchemy.engine，"
-            "此处配置会追加到默认列表。"
-        ),
-    )
 
 
 class ServiceSettings(BaseModel):
@@ -538,24 +530,25 @@ class MessageQueueSettings(BaseModel):
     """消息队列配置。
     
     环境变量格式: MQ__{FIELD}
-    示例: MQ__BROKER_URL, MQ__DEFAULT_QUEUE, MQ__SERIALIZER
+    示例: MQ__BACKEND, MQ__URL, MQ__BROKER_URL
     
     与 Task（任务队列）的区别：
     - Task: 基于 Dramatiq，用于异步任务处理（API + Worker 模式）
     - MQ: 通用消息队列，用于服务间通信、事件驱动架构
     
     支持的后端：
-    - Redis: redis://localhost:6379/0
-    - RabbitMQ: amqp://guest:guest@localhost:5672//
+    - redis: redis://localhost:6379/0
+    - redis_stream: redis://localhost:6379/0
+    - rabbitmq: amqp://guest:guest@localhost:5672//
     """
     
-    enabled: bool = Field(
-        default=False,
-        description="是否启用消息队列组件"
+    backend: str = Field(
+        default="",
+        description="消息队列后端 (redis/redis_stream/rabbitmq)，空字符串表示不启用"
     )
-    broker_url: str | None = Field(
+    url: str | None = Field(
         default=None,
-        description="消息队列代理 URL"
+        description="连接 URL"
     )
     default_queue: str = Field(
         default="default",
@@ -1009,6 +1002,7 @@ class BaseConfig(BaseSettings):
     # ========== 异步与事件 ==========
     task: TaskSettings = Field(default_factory=TaskSettings)
     event: EventSettings = Field(default_factory=EventSettings)
+    mq: MessageQueueSettings = Field(default_factory=MessageQueueSettings)
     
     # ========== 微服务通信 ==========
     # RPC 客户端配置（调用其他服务）
@@ -1116,10 +1110,18 @@ class BaseConfig(BaseSettings):
         """获取所有消息队列实例配置。
         
         从环境变量解析 MQ__{INSTANCE}__{FIELD} 格式的配置。
+        如果没有配置多实例，返回从单实例配置转换的 default 实例。
         """
         if self._mqs is None:
             loader = MultiInstanceConfigLoader("MQ", MQInstanceConfig)
             self._mqs = loader.load()
+            if not self._mqs and self.mq.backend:
+                self._mqs = {
+                    "default": MQInstanceConfig(
+                        backend=self.mq.backend,
+                        url=self.mq.url,
+                    )
+                }
         return self._mqs
     
     def get_events(self) -> dict[str, EventInstanceConfig]:

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/commands/scheduler.py

@@ -84,9 +84,12 @@ def run_scheduler(
         module = __import__(module_path, fromlist=[app_name])
         application = getattr(module, app_name)
 
-        # 设置日志上下文
-        from aury.boot.common.logging import set_service_context
-        set_service_context("scheduler")
+        # 设置日志（必须在其他操作之前）
+        from aury.boot.common.logging import setup_logging
+        setup_logging(
+            log_level=getattr(application, "_config", None) and application._config.log.level or "INFO",
+            service_type="scheduler",
+        )
 
         # 获取调度器
         from aury.boot.infrastructure.scheduler import SchedulerManager

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/commands/server/app.py

@@ -9,6 +9,7 @@ import sys
 from typing import TYPE_CHECKING
 
 import typer
+import uvicorn
 
 if TYPE_CHECKING:
     from aury.boot.application.app.base import FoundationApp
@@ -416,7 +417,6 @@ def dev(
         os_module.environ["AURIMYTH_RELOAD"] = "1"
         
         # 热重载模式下，直接使用 uvicorn，传递 app 字符串路径
-        import uvicorn
         uvicorn.run(
             app=app_module_path,
             host=server_host,
@@ -500,19 +500,22 @@ def prod(
     typer.echo("   热重载: ❌")
     typer.echo("   调试模式: ❌")
     
+    # 获取 app 模块路径（多进程模式需要字符串格式）
+    app_module_path = app_path or _detect_app_module()
+    typer.echo(f"   应用模块: {app_module_path}")
+    
     try:
-        server = ApplicationServer(
-            app=app_instance,
+        # 多进程模式必须使用字符串路径，否则子进程无法重新加载应用
+        uvicorn.run(
+            app=app_module_path,
             host=server_host,
             port=server_port,
             workers=server_workers,
             reload=False,
             loop="auto",
             http="auto",
-            debug=False,
             access_log=True,
         )
-        server.run()
     except KeyboardInterrupt:
         typer.echo("\n👋 服务器已停止")
     except Exception as e:

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/commands/templates/project/main.py.tpl

@@ -25,15 +25,17 @@ config = AppConfig()
 #   - HEALTH_CHECK_PATH: 健康检查路径（默认 /api/health）
 #   - HEALTH_CHECK_ENABLED: 是否启用（默认 true）
 #
-# 日志拦截：
-#   框架默认拦截 uvicorn/sqlalchemy.engine，可通过 intercept_loggers 追加额外的 logger
+# 日志：
+#   框架自动全局接管所有 logging，无需配置
+#   要查看 TRACE 级别日志，设置 LOG__LEVEL=TRACE
+#   要屏蔽某些库的 DEBUG 日志，使用 logger_levels 参数
 #
 app = FoundationApp(
     title="{project_name}",
     version="0.1.0",
     description="{project_name} - 基于 Aury Boot",
     config=config,
-    intercept_loggers=[],
+    # logger_levels=[("sse_starlette", "WARNING")],  # 可选：设置特定库的日志级别
 )
 
 # 注册 API 路由

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/commands/worker.py

@@ -100,9 +100,12 @@ def run_worker(
         module = __import__(module_path, fromlist=[app_name])
         application = getattr(module, app_name)
 
-        # 设置日志上下文
-        from aury.boot.common.logging import set_service_context
-        set_service_context("worker")
+        # 设置日志（必须在其他操作之前）
+        from aury.boot.common.logging import setup_logging
+        setup_logging(
+            log_level=getattr(application, "_config", None) and application._config.log.level or "INFO",
+            service_type="worker",
+        )
 
         # 尝试导入 dramatiq
         try:

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/common/logging/__init__.py

@@ -41,14 +41,13 @@ from aury.boot.common.logging.format import (
     log_exception,
 )
 from aury.boot.common.logging.setup import (
-    DEFAULT_INTERCEPT_LOGGERS,
+    TRACE,
     register_log_sink,
-    setup_intercept,
     setup_logging,
 )
 
 __all__ = [
-    "DEFAULT_INTERCEPT_LOGGERS",
+    "TRACE",
     "ServiceContext",
     "format_exception_java_style",
     "get_class_logger",
@@ -61,7 +60,6 @@ __all__ = [
     "register_log_sink",
     "set_service_context",
     "set_trace_id",
-    "setup_intercept",
     "setup_logging",
 ]
 

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/common/logging/setup.py

@@ -11,6 +11,25 @@ from typing import Any
 
 from loguru import logger
 
+# =============================================================================
+# TRACE Level 支持
+# =============================================================================
+# 标准 logging 没有 TRACE，需要手动添加
+# TRACE (5) < DEBUG (10)，用于超细粒度调试（如每个 streaming chunk）
+TRACE = 5
+logging.addLevelName(TRACE, "TRACE")
+
+
+def _add_trace_method() -> None:
+    """为标准 logging.Logger 添加 trace() 方法。"""
+    def trace(self: logging.Logger, msg: str, *args: Any, **kwargs: Any) -> None:
+        if self.isEnabledFor(TRACE):
+            self._log(TRACE, msg, args, **kwargs)
+    
+    logging.Logger.trace = trace  # type: ignore[attr-defined]
+
+_add_trace_method()
+
 from aury.boot.common.logging.context import (
     ServiceContext,
     _to_service_context,
@@ -89,31 +108,27 @@ def register_log_sink(
     logger.debug(f"注册日志 sink: {name} (filter_key={filter_key})")
 
 
-# 默认拦截的标准 logging 日志记录器
-# - uvicorn: Uvicorn 服务器日志
-# - uvicorn.error: Uvicorn 错误日志
-# - sqlalchemy.engine: SQLAlchemy SQL 语句日志
-# 注意：uvicorn.access 不拦截，因为框架有自己的 RequestLoggingMiddleware
-DEFAULT_INTERCEPT_LOGGERS = [
-    "uvicorn",
-    "uvicorn.error",
-    "sqlalchemy.engine",
-]
-
 
 class _InterceptHandler(logging.Handler):
     """将标准 logging 日志转发到 loguru 的处理器。"""
 
     def emit(self, record: logging.LogRecord) -> None:
         # 获取对应的 loguru 级别
+        # loguru 原生支持 TRACE，标准 logging 的 TRACE(5) 会自动映射
         try:
             level = logger.level(record.levelname).name
         except ValueError:
             level = record.levelno
 
         # 查找调用者的帧深度
-        frame, depth = logging.currentframe(), 2
-        while frame and frame.f_code.co_filename == logging.__file__:
+        # 跳过 logging 模块和本文件的所有帧
+        frame = logging.currentframe()
+        depth = 0
+        while frame is not None:
+            filename = frame.f_code.co_filename
+            # 跳过 logging 模块、本文件、loguru 内部
+            if "logging" not in filename and "loguru" not in filename:
+                break
             frame = frame.f_back
             depth += 1
 
@@ -122,35 +137,22 @@ class _InterceptHandler(logging.Handler):
         )
 
 
-def _setup_intercept(logger_names: list[str]) -> None:
-    """让 loguru 接管指定的标准 logging 日志记录器。"""
-    handler = _InterceptHandler()
-    for name in logger_names:
-        std_logger = logging.getLogger(name)
-        std_logger.handlers = [handler]
-        std_logger.setLevel(logging.DEBUG)
-        std_logger.propagate = False
-
-
-def setup_intercept(logger_names: list[str] | None = None) -> None:
-    """拦截标准 logging 日志记录器并转发到 loguru。
-
-    用于独立脚本/CLI 入口点（不使用 FoundationApp 时）。
-    FoundationApp 会自动调用此函数，无需手动调用。
-
+def _setup_global_intercept(logger_levels: list[tuple[str, str]] | None = None) -> None:
+    """全局接管所有标准 logging，转发到 loguru。
+    
+    这样任何使用 logging.getLogger() 的库都会自动被接管。
+    
     Args:
-        logger_names: 额外需要拦截的 logger 名称列表，
-            会追加到默认列表 (uvicorn, sqlalchemy.engine 等)。
-
-    使用示例::
-
-        from aury.boot.common.logging import setup_logging, setup_intercept
-
-        setup_logging(log_level="DEBUG")
-        setup_intercept(["my_package", "third_party_lib"])
+        logger_levels: 需要设置特定级别的 logger 列表，格式: [("name", "LEVEL"), ...]
+            例如: [("sse_starlette", "WARNING"), ("httpx", "INFO")]
     """
-    to_intercept = DEFAULT_INTERCEPT_LOGGERS + (logger_names or [])
-    _setup_intercept(to_intercept)
+    logging.root.handlers = [_InterceptHandler()]
+    logging.root.setLevel(TRACE)  # 接收所有级别，包括 TRACE
+    
+    # 对指定的 logger 设置特定级别
+    if logger_levels:
+        for name, level in logger_levels:
+            logging.getLogger(name).setLevel(level.upper())
 
 
 def setup_logging(
@@ -162,7 +164,7 @@ def setup_logging(
     retention_days: int = 7,
     rotation_size: str = "50 MB",
     enable_console: bool = True,
-    intercept_loggers: list[str] | None = None,
+    logger_levels: list[tuple[str, str]] | None = None,
 ) -> None:
     """设置日志配置。
 
@@ -177,7 +179,7 @@ def setup_logging(
     可通过 register_log_sink() 注册额外的日志文件（如 access.log）。
 
     Args:
-        log_level: 日志级别（DEBUG/INFO/WARNING/ERROR/CRITICAL）
+        log_level: 日志级别（DEBUG/INFO/WARNING/ERROR/CRITICAL/TRACE）
         log_dir: 日志目录（默认：./logs）
         service_type: 服务类型（app/scheduler/worker）
         enable_file_rotation: 是否启用日志轮转
@@ -185,8 +187,8 @@ def setup_logging(
         retention_days: 日志保留天数（默认：7 天）
         rotation_size: 单文件大小上限（默认：50 MB）
         enable_console: 是否输出到控制台
-        intercept_loggers: 额外需要拦截的标准 logging logger 名称列表，
-            会追加到默认列表 (uvicorn, sqlalchemy.engine 等)。
+        logger_levels: 需要设置特定级别的 logger 列表，格式: [("name", "LEVEL"), ...]
+            例如: [("sse_starlette", "WARNING"), ("httpx", "INFO")]
     """
     log_level = log_level.upper()
     log_dir = log_dir or "logs"
@@ -273,16 +275,15 @@ def setup_logging(
             filter=lambda record, c=ctx: record["extra"].get("service") == c,
         )
 
-    # 拦截标准 logging 日志并转发到 loguru
-    to_intercept = DEFAULT_INTERCEPT_LOGGERS + (intercept_loggers or [])
-    _setup_intercept(to_intercept)
+    # 全局拦截标准 logging 日志并转发到 loguru
+    # 所有使用 logging.getLogger() 的库自动被接管
+    _setup_global_intercept(logger_levels=logger_levels)
 
-    logger.info(f"日志系统初始化完成 | 服务: {service_type} | 级别: {log_level} | 目录: {log_dir}")
+    logger.info
 
 
 __all__ = [
-    "DEFAULT_INTERCEPT_LOGGERS",
+    "TRACE",
     "register_log_sink",
-    "setup_intercept",
     "setup_logging",
 ]

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/infrastructure/monitoring/alerting/rules.py

@@ -107,14 +107,14 @@ class AlertRule:
         # 检查路径
         if self._path_regex:
             endpoint = event.metadata.get("endpoint", "")
-            if not self._path_regex.match(endpoint):
+            if not self._path_regex.fullmatch(endpoint):
                 return False
         
         # 检查排除路径
         if self._exclude_regexes:
             endpoint = event.metadata.get("endpoint", "")
             for exclude_regex in self._exclude_regexes:
-                if exclude_regex.match(endpoint):
+                if exclude_regex.fullmatch(endpoint):
                     return False  # 匹配到排除规则，不触发告警
         
         # 检查阈值（对于 slow_* 类型）

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/infrastructure/monitoring/tracing/processor.py

@@ -181,7 +181,23 @@ class AlertingSpanProcessor:
             or name
         )
         
-        return any(regex.match(path) for regex in self._exclude_regexes)
+        # 检查所有可能的路径来源
+        paths_to_check = [path]
+        
+        # 也检查 span name 中的路径（可能包含 HTTP 方法和后缀）
+        # 例如 "GET /api/v1/spaces/{space_id}/subscribe http receive"
+        if name and name != path:
+            # 尝试提取 span name 中的路径部分
+            parts = name.split()
+            for part in parts:
+                if part.startswith("/"):
+                    paths_to_check.append(part)
+        
+        for p in paths_to_check:
+            if any(regex.fullmatch(p) for regex in self._exclude_regexes):
+                return True
+        
+        return False
     
     def _emit_slow_alert(
         self,

{aury_boot-0.0.34 → aury_boot-0.0.36}/aury/boot/infrastructure/mq/backends/__init__.py

@@ -2,8 +2,10 @@
 
 from .rabbitmq import RabbitMQ
 from .redis import RedisMQ
+from .redis_stream import RedisStreamMQ
 
 __all__ = [
     "RabbitMQ",
     "RedisMQ",
+    "RedisStreamMQ",
 ]