infomankit 0.3.23__py3-none-any.whl

infoman/service/core/lifespan.py

@@ -0,0 +1,132 @@
+# service/core/lifespan.py
+"""
+应用生命周期管理（精简版）
+
+职责：
+- 协调各个服务的启动和关闭
+- 不包含具体的连接逻辑
+- 支持可选依赖的优雅降级
+"""
+
+from contextlib import asynccontextmanager
+from fastapi import FastAPI
+from loguru import logger
+
+from infoman.config import settings
+
+# ========== 数据库（必需） ==========
+if settings.USE_PRO_ORM:
+    from infoman.service.infrastructure.db_relation.manager_pro import db_manager
+    _DB_MANAGER_TYPE = "pro"
+else:
+    from infoman.service.infrastructure.db_relation.manager import db_manager
+    _DB_MANAGER_TYPE = "basic"
+
+# ========== Redis（可选） ==========
+try:
+    from infoman.service.infrastructure.db_cache.manager import RedisManager
+    REDIS_AVAILABLE = True
+except ImportError:
+    logger.warning("⚠️ Redis 依赖未安装，缓存功能将不可用 (需要: redis, fastapi-cache2)")
+    RedisManager = None
+    REDIS_AVAILABLE = False
+
+# ========== 向量数据库（可选） ==========
+try:
+    from infoman.service.infrastructure.db_vector.manager import VectorDBManager
+    VECTOR_AVAILABLE = True
+except ImportError:
+    logger.warning("⚠️ 向量数据库依赖未安装，向量搜索功能将不可用 (需要: qdrant-client)")
+    VectorDBManager = None
+    VECTOR_AVAILABLE = False
+
+# ========== 消息队列（可选） ==========
+try:
+    from infoman.service.infrastructure.mq import NATSManager
+    MQ_AVAILABLE = True
+except ImportError:
+    logger.warning("⚠️ 消息队列依赖未安装，NATS 功能将不可用 (需要: nats-py)")
+    NATSManager = None
+    MQ_AVAILABLE = False
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """应用生命周期管理（支持可选依赖）"""
+
+    # ========== 启动 ==========
+    logger.info(f"🚀 应用启动中 [{settings.APP_NAME} v{settings.APP_VERSION}]")
+    logger.info(f"   环境: {settings.ENV}")
+    logger.info(f"   数据库管理器: {_DB_MANAGER_TYPE}")
+    logger.info(f"   可选功能: Redis={REDIS_AVAILABLE}, Vector={VECTOR_AVAILABLE}, MQ={MQ_AVAILABLE}")
+
+    # 初始化管理器（仅限已安装的）
+    managers = {}
+
+    # 数据库（必需）
+    managers['db'] = db_manager
+    app.state.db_manager = db_manager
+
+    # Redis（可选）
+    if REDIS_AVAILABLE:
+        redis_manager = RedisManager()
+        managers['redis'] = redis_manager
+        app.state.redis_manager = redis_manager
+
+    # 向量数据库（可选）
+    if VECTOR_AVAILABLE:
+        vector_manager = VectorDBManager()
+        managers['vector'] = vector_manager
+        app.state.vector_manager = vector_manager
+
+    # 消息队列（可选）
+    if MQ_AVAILABLE:
+        nats_manager = NATSManager()
+        managers['mq'] = nats_manager
+        app.state.nats_manager = nats_manager
+
+    try:
+        # 1. 数据库
+        await db_manager.startup(app)
+
+        # 2. Redis（如果可用）
+        if 'redis' in managers:
+            await managers['redis'].startup(app)
+
+        # 3. 向量数据库（如果可用）
+        if 'vector' in managers:
+            await managers['vector'].startup(app)
+
+        # 4. 消息队列（如果可用）
+        if 'mq' in managers:
+            await managers['mq'].startup(app)
+
+        logger.success("✅ 所有服务启动完成")
+
+    except Exception as e:
+        logger.error(f"❌ 服务启动失败: {e}")
+        raise
+
+    # ========== 运行 ==========
+    yield
+
+    # ========== 关闭 ==========
+    logger.info("⏹️ 应用关闭中...")
+
+    try:
+        # 按相反顺序关闭（仅关闭已启动的）
+        if 'mq' in managers:
+            await managers['mq'].shutdown()
+
+        if 'vector' in managers:
+            await managers['vector'].shutdown()
+
+        if 'redis' in managers:
+            await managers['redis'].shutdown()
+
+        await db_manager.shutdown()
+
+        logger.success("✅ 所有服务已关闭")
+
+    except Exception as e:
+        logger.error(f"❌ 服务关闭失败: {e}")

infoman/service/core/monitor.py

@@ -0,0 +1,57 @@
+from prometheus_fastapi_instrumentator import Instrumentator, metrics
+from prometheus_client import Counter, Histogram, Gauge
+
+# ===== 1. HTTP 指标（自动收集）=====
+instrumentator = Instrumentator(
+    should_group_status_codes=True,
+    should_ignore_untemplated=True,
+    should_respect_env_var=True,
+    should_instrument_requests_inprogress=True,
+    excluded_handlers=["/health", "/metrics", "/doc", "/redoc", "/openapi.json"],
+)
+
+instrumentator.add(metrics.default())
+instrumentator.add(metrics.request_size())
+instrumentator.add(metrics.response_size())
+
+# ===== 2. 业务指标（手动收集）=====
+
+# 用户操作
+user_operations = Counter(
+    "user_operations_total", "Total user operations", ["operation_type", "status"]
+)
+
+# 数据库查询
+db_query_duration = Histogram(
+    "db_query_duration_seconds",
+    "Database query duration",
+    ["query_type"],
+    buckets=[0.001, 0.01, 0.05, 0.1, 0.5, 1.0, 2.5, 5.0],
+)
+
+# 缓存
+cache_hits = Counter("cache_hits_total", "Cache hits", ["cache_type"])
+cache_misses = Counter("cache_misses_total", "Cache misses", ["cache_type"])
+
+# 连接数
+active_db_connections = Gauge("active_db_connections", "Active database connections")
+active_redis_connections = Gauge("active_redis_connections", "Active Redis connections")
+
+# 队列
+task_queue_length = Gauge("task_queue_length", "Task queue length")
+
+# 在线用户
+online_users = Gauge("online_users", "Online users count")
+
+# ===== 3. 导出所有指标（方便使用）=====
+__all__ = [
+    "instrumentator",
+    "user_operations",
+    "db_query_duration",
+    "cache_hits",
+    "cache_misses",
+    "active_db_connections",
+    "active_redis_connections",
+    "task_queue_length",
+    "online_users",
+]

infoman/service/core/response.py

@@ -0,0 +1,37 @@
+# -*- coding:utf-8 -*-
+"""
+# Time       ：2023/12/8 18:23
+# Author     ：Maxwell
+# version    ：python 3.9
+# Description：
+"""
+from infoman.service.exception.exception import AppError
+from infoman.config import settings as config
+from fastapi.responses import ORJSONResponse
+import orjson
+
+
+class ProRJSONResponse(ORJSONResponse):
+    def render(self, content) -> bytes:
+        return orjson.dumps(
+            content,
+            option=(
+                orjson.OPT_NON_STR_KEYS | orjson.OPT_SERIALIZE_NUMPY | orjson.OPT_UTC_Z
+            ),
+        )
+
+
+def response(code, msg, data=None):
+    result = {"code": code, "message": msg, "data": data}
+    return result
+
+
+def success(data=None, msg=""):
+    return response(200, msg, data)
+
+
+def failed(error: AppError):
+    msg = error.message_en
+    if not config.DEFAULT_LANGUAGE_IS_EN:
+        msg = error.message
+    return response(code=error.code, msg=msg)

infoman/service/exception/__init__.py

@@ -0,0 +1,7 @@
+# !/usr/bin/env python
+# -*-coding:utf-8 -*-
+"""
+# Time       ：2023/12/9 13:48
+# Author     ：Maxwell
+# Description：
+"""

infoman/service/exception/error.py

@@ -0,0 +1,274 @@
+from enum import Enum
+from dataclasses import dataclass
+from typing import Optional, Dict, Any
+from http import HTTPStatus
+
+
+@dataclass
+class ErrorProperties:
+    code: int
+    message: str
+    message_en: str
+    status_code: int = HTTPStatus.INTERNAL_SERVER_ERROR
+    details: Optional[Dict[str, Any]] = None
+
+
+class AppError(Enum):
+
+    @property
+    def code(self) -> int:
+        return self.value.code
+
+    @property
+    def message(self) -> str:
+        return self.value.message
+
+    @property
+    def message_en(self) -> str:
+        return self.value.message_en
+
+    @property
+    def status_code(self) -> int:
+        return self.value.status_code
+
+    @property
+    def details(self) -> Optional[Dict[str, Any]]:
+        return self.value.details
+
+    def with_details(self, details: Dict[str, Any]) -> ErrorProperties:
+        return ErrorProperties(
+            code=self.code,
+            message=self.message,
+            message_en=self.message_en,
+            status_code=self.status_code,
+            details=details,
+        )
+
+
+class AppSystemError(AppError):
+    UNKNOWN_ERROR = ErrorProperties(
+        100000, "系统未知异常", "Unknown system error", HTTPStatus.INTERNAL_SERVER_ERROR
+    )
+
+    SYSTEM_ERROR = ErrorProperties(
+        100001,
+        "系统内部错误",
+        "Internal server error",
+        HTTPStatus.INTERNAL_SERVER_ERROR,
+    )
+
+    TIMEOUT_ERROR = ErrorProperties(
+        100002, "系统处理超时", "System processing timeout", HTTPStatus.GATEWAY_TIMEOUT
+    )
+
+    SERVICE_UNAVAILABLE = ErrorProperties(
+        100003,
+        "服务暂时不可用",
+        "Service temporarily unavailable",
+        HTTPStatus.SERVICE_UNAVAILABLE,
+    )
+
+    RESOURCE_EXHAUSTED = ErrorProperties(
+        100004,
+        "系统资源不足",
+        "System resources exhausted",
+        HTTPStatus.TOO_MANY_REQUESTS,
+    )
+
+    CONFIGURATION_ERROR = ErrorProperties(
+        100005,
+        "系统配置错误",
+        "System configuration error",
+        HTTPStatus.INTERNAL_SERVER_ERROR,
+    )
+
+
+class RequestError(AppError):
+    VALIDATION_ERROR = ErrorProperties(
+        200000,
+        "请求数据验证失败",
+        "Request data validation failed",
+        HTTPStatus.BAD_REQUEST,
+    )
+
+    INVALID_PARAMETER = ErrorProperties(
+        200001, "无效的请求参数", "Invalid request parameter", HTTPStatus.BAD_REQUEST
+    )
+
+    MISSING_PARAMETER = ErrorProperties(
+        200002, "缺少必要参数", "Missing required parameter", HTTPStatus.BAD_REQUEST
+    )
+
+    INVALID_FORMAT = ErrorProperties(
+        200003, "请求格式错误", "Invalid request format", HTTPStatus.BAD_REQUEST
+    )
+
+    REQUEST_TOO_LARGE = ErrorProperties(
+        200004,
+        "请求体过大",
+        "Request body too large",
+        HTTPStatus.REQUEST_ENTITY_TOO_LARGE,
+    )
+
+    RATE_LIMITED = ErrorProperties(
+        200005, "请求频率超限", "Request rate limited", HTTPStatus.TOO_MANY_REQUESTS
+    )
+
+    UNSUPPORTED_MEDIA_TYPE = ErrorProperties(
+        200006,
+        "不支持的媒体类型",
+        "Unsupported media type",
+        HTTPStatus.UNSUPPORTED_MEDIA_TYPE,
+    )
+
+
+class DatabaseError(AppError):
+    VALIDATION_ERROR = ErrorProperties(
+        300000, "数据验证失败", "Data validation failed", HTTPStatus.BAD_REQUEST
+    )
+
+    INTEGRITY_ERROR = ErrorProperties(
+        300001,
+        "数据完整性约束冲突",
+        "Data integrity constraint violation",
+        HTTPStatus.CONFLICT,
+    )
+
+    RECORD_NOT_FOUND = ErrorProperties(
+        300002, "记录不存在", "Record not found", HTTPStatus.NOT_FOUND
+    )
+
+    DUPLICATE_ENTRY = ErrorProperties(
+        300003, "记录已存在", "Duplicate record", HTTPStatus.CONFLICT
+    )
+
+    OPERATIONAL_ERROR = ErrorProperties(
+        300004,
+        "数据库操作错误",
+        "Database operational error",
+        HTTPStatus.INTERNAL_SERVER_ERROR,
+    )
+
+    CONNECTION_ERROR = ErrorProperties(
+        300005,
+        "数据库连接错误",
+        "Database connection error",
+        HTTPStatus.SERVICE_UNAVAILABLE,
+    )
+
+    TRANSACTION_ERROR = ErrorProperties(
+        300006,
+        "事务处理错误",
+        "Transaction processing error",
+        HTTPStatus.INTERNAL_SERVER_ERROR,
+    )
+
+    QUERY_TIMEOUT = ErrorProperties(
+        300007, "查询超时", "Query timeout", HTTPStatus.GATEWAY_TIMEOUT
+    )
+
+
+class BusinessError(AppError):
+
+    INVALID_OPERATION = ErrorProperties(
+        400000, "无效的操作", "Invalid operation", HTTPStatus.BAD_REQUEST
+    )
+
+    STATE_CONFLICT = ErrorProperties(
+        400001, "状态冲突", "State conflict", HTTPStatus.CONFLICT
+    )
+
+    RESOURCE_LOCKED = ErrorProperties(
+        400002, "资源已锁定", "Resource locked", HTTPStatus.LOCKED
+    )
+
+    QUOTA_EXCEEDED = ErrorProperties(
+        400003, "配额超限", "Quota exceeded", HTTPStatus.FORBIDDEN
+    )
+
+    PAYMENT_REQUIRED = ErrorProperties(
+        400004, "需要付款", "Payment required", HTTPStatus.PAYMENT_REQUIRED
+    )
+
+    BUSINESS_RULE_VIOLATION = ErrorProperties(
+        400005, "违反业务规则", "Business rule violation", HTTPStatus.BAD_REQUEST
+    )
+
+
+class SecurityError(AppError):
+    """
+    安全相关错误
+
+    包括认证、授权、访问控制等错误
+    错误代码范围：500000-599999
+    """
+
+    UNAUTHORIZED = ErrorProperties(
+        500000, "未授权访问", "Unauthorized access", HTTPStatus.UNAUTHORIZED
+    )
+
+    FORBIDDEN = ErrorProperties(
+        500001, "禁止访问", "Access forbidden", HTTPStatus.FORBIDDEN
+    )
+
+    INVALID_CREDENTIALS = ErrorProperties(
+        500002, "无效的凭证", "Invalid credentials", HTTPStatus.UNAUTHORIZED
+    )
+
+    TOKEN_EXPIRED = ErrorProperties(
+        500003, "令牌已过期", "Token expired", HTTPStatus.UNAUTHORIZED
+    )
+
+    INVALID_TOKEN = ErrorProperties(
+        500004, "无效的令牌", "Invalid token", HTTPStatus.UNAUTHORIZED
+    )
+
+    PERMISSION_DENIED = ErrorProperties(
+        500005, "权限不足", "Permission denied", HTTPStatus.FORBIDDEN
+    )
+
+    ACCOUNT_LOCKED = ErrorProperties(
+        500006, "账户已锁定", "Account locked", HTTPStatus.FORBIDDEN
+    )
+
+    IP_RESTRICTED = ErrorProperties(
+        500007, "IP地址受限", "IP address restricted", HTTPStatus.FORBIDDEN
+    )
+
+
+class ExternalServiceError(AppError):
+    """
+    外部服务错误
+
+    包括第三方API、外部依赖服务等错误
+    错误代码范围：600000-699999
+    """
+
+    SERVICE_UNAVAILABLE = ErrorProperties(
+        600000, "外部服务不可用", "External service unavailable", HTTPStatus.BAD_GATEWAY
+    )
+
+    REQUEST_FAILED = ErrorProperties(
+        600001,
+        "外部服务请求失败",
+        "External service request failed",
+        HTTPStatus.BAD_GATEWAY,
+    )
+
+    RESPONSE_ERROR = ErrorProperties(
+        600002,
+        "外部服务响应错误",
+        "External service response error",
+        HTTPStatus.BAD_GATEWAY,
+    )
+
+    TIMEOUT = ErrorProperties(
+        600003, "外部服务超时", "External service timeout", HTTPStatus.GATEWAY_TIMEOUT
+    )
+
+    INVALID_RESPONSE = ErrorProperties(
+        600004,
+        "外部服务返回无效数据",
+        "Invalid data returned from external service",
+        HTTPStatus.BAD_GATEWAY,
+    )

infoman/service/exception/exception.py

@@ -0,0 +1,25 @@
+# !/usr/bin/env python
+# -*-coding:utf-8 -*-
+
+"""
+# Time       ：2025/6/21 21:51
+# Author     ：Maxwell
+# Description：
+"""
+from infoman.service.exception.error import AppError
+from infoman.config.settings import settings
+
+
+class AppException(Exception):
+
+    def __init__(self, error: AppError):
+        self.error_code = error.code
+        self.message = error.message_en
+        if not settings.DEFAULT_LANGUAGE_IS_EN:
+            self.message = error.message
+        super().__init__(error.message)
+
+    def with_content(self, place_holder, content):
+        if self.message:
+            self.message = self.message.replace(place_holder, content)
+        return self