upload project source code

2026-04-30 18:49:43 +08:00
commit 9b394ba682
2277 changed files with 660945 additions and 0 deletions
--- a/后端源码/yifan.action-ai.cn/api-bak/app/core/database.py
+++ b/后端源码/yifan.action-ai.cn/api-bak/app/core/database.py
@@ -0,0 +1,126 @@
+# -*- coding: utf-8 -*-
+
+from redis.asyncio import Redis
+from redis import exceptions
+from fastapi import FastAPI
+from sqlalchemy import create_engine, Engine
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker, AsyncSession, AsyncEngine
+
+from app.core.logger import log
+from app.config.setting import settings
+from app.core.exceptions import CustomException
+
+
+def create_engine_and_session(
+    db_url: str = settings.DB_URI
+) -> tuple[Engine, sessionmaker]:
+    """
+    创建同步数据库引擎和会话工厂。
+    
+    参数:
+    - db_url (str): 数据库连接URL,默认从配置中获取。
+    
+    返回:
+    - tuple[Engine, sessionmaker]: 同步数据库引擎和会话工厂。
+    """
+    try:
+        if not settings.SQL_DB_ENABLE:
+            raise CustomException(msg="请先开启数据库连接", data="请启用 app/config/setting.py: SQL_DB_ENABLE")
+        # 同步数据库引擎
+        engine: Engine = create_engine(
+            url=db_url,
+            echo=settings.DATABASE_ECHO,
+            pool_pre_ping=settings.POOL_PRE_PING,
+            pool_recycle=settings.POOL_RECYCLE,
+        )
+    except Exception as e:
+        log.error(f'❌ 数据库连接失败 {e}')
+        raise
+    else:
+        # 同步数据库会话工厂
+        SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+        return engine, SessionLocal
+
+def create_async_engine_and_session(
+    db_url: str = settings.ASYNC_DB_URI
+) -> tuple[AsyncEngine, async_sessionmaker[AsyncSession]]:
+    """
+    获取异步数据库会话连接。
+    
+    返回:
+    - tuple[AsyncEngine, async_sessionmaker[AsyncSession]]: 异步数据库引擎和会话工厂。
+    """
+    try:
+        if not settings.SQL_DB_ENABLE:
+            raise CustomException(msg="请先开启数据库连接", data="请启用 app/config/setting.py: SQL_DB_ENABLE")
+        # 异步数据库引擎
+        async_engine: AsyncEngine = create_async_engine(
+            url=db_url,
+            echo=settings.DATABASE_ECHO,
+            echo_pool=settings.ECHO_POOL,
+            pool_pre_ping=settings.POOL_PRE_PING,
+            future=settings.FUTURE,
+            pool_recycle=settings.POOL_RECYCLE,
+            pool_size=settings.POOL_SIZE,
+            max_overflow=settings.MAX_OVERFLOW,
+            pool_timeout=settings.POOL_TIMEOUT,
+            pool_use_lifo=settings.POOL_USE_LIFO,
+        )
+    except Exception as e:
+        log.error(f'❌ 数据库连接失败 {e}')
+        raise
+    else:
+        # 异步数据库会话工厂
+        AsyncSessionLocal = async_sessionmaker(
+            bind=async_engine,
+            autocommit=settings.AUTOCOMMIT,
+            autoflush=settings.AUTOFETCH,
+            expire_on_commit=settings.EXPIRE_ON_COMMIT,
+            class_=AsyncSession
+        )
+        return async_engine, AsyncSessionLocal
+
+engine, db_session = create_engine_and_session(settings.DB_URI)
+async_engine, async_db_session = create_async_engine_and_session(settings.ASYNC_DB_URI)
+
+async def redis_connect(app: FastAPI, status: str) -> Redis | None:
+    """
+    创建或关闭Redis连接。
+    
+    参数:
+    - app (FastAPI): FastAPI应用实例。
+    - status (bool): 连接状态,True为创建连接,False为关闭连接。
+    
+    返回:
+    - Redis | None: Redis连接实例,如果连接失败则返回None。
+    """
+    if not settings.REDIS_ENABLE:
+        raise CustomException(msg="请先开启Redis连接", data="请启用 app/core/config.py: REDIS_ENABLE")
+
+    if status:
+        try:
+            rd = await Redis.from_url(
+                url=settings.REDIS_URI,
+                encoding='utf-8',
+                decode_responses=True,
+                health_check_interval=20,
+                max_connections=settings.POOL_SIZE,
+                socket_timeout=settings.POOL_TIMEOUT
+            )
+            app.state.redis = rd
+            if await rd.ping():
+                log.info("✅️ Redis连接成功...")
+                return rd
+        except exceptions.AuthenticationError as e:
+            log.error(f"❌ 数据库 Redis 认证失败: {e}")
+            raise
+        except exceptions.TimeoutError as e:
+            log.error(f"❌ 数据库 Redis 连接超时: {e}")
+            raise
+        except exceptions.RedisError as e:
+            log.error(f"❌ 数据库 Redis 连接错误: {e}")
+            raise
+    else:
+        await app.state.redis.aclose()
+        log.info('✅️ Redis连接已关闭')