初始化

ds/config.py

@@ -0,0 +1,17 @@
+import configparser
+import os
+
+# 读取配置文件路径
+config_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../config.ini")
+
+# 初始化配置解析器
+config = configparser.ConfigParser()
+
+# 读取配置文件
+config.read(config_path, encoding="utf-8")
+
+# 暴露配置项（方便其他文件调用）
+SERVER_CONFIG = config["server"]
+MYSQL_CONFIG = config["mysql"]
+JWT_CONFIG = config["jwt"]
+LIVE_CONFIG = config["live"]

ds/db.py

@@ -0,0 +1,46 @@
+import mysql.connector
+from mysql.connector import Error
+
+from .config import MYSQL_CONFIG
+
+
+class Database:
+    """MySQL 连接池管理类"""
+    pool_config = {
+        "host": MYSQL_CONFIG.get("host", "localhost"),
+        "port": int(MYSQL_CONFIG.get("port", 3306)),
+        "user": MYSQL_CONFIG.get("user", "root"),
+        "password": MYSQL_CONFIG.get("password", ""),
+        "database": MYSQL_CONFIG.get("database", ""),
+        "charset": MYSQL_CONFIG.get("charset", "utf8mb4"),
+        "pool_name": "fastapi_pool",
+        "pool_size": 5,
+        "pool_reset_session": True
+    }
+
+    @classmethod
+    def get_connection(cls):
+        """获取数据库连接"""
+        try:
+            # 从连接池获取连接
+            conn = mysql.connector.connect(**cls.pool_config)
+            if conn.is_connected():
+                return conn
+        except Error as e:
+            # 抛出数据库连接错误（会被全局异常处理器捕获）
+            raise Exception(f"MySQL 连接失败: {str(e)}") from e
+
+    @classmethod
+    def close_connection(cls, conn, cursor=None):
+        """关闭连接和游标"""
+        try:
+            if cursor:
+                cursor.close()
+            if conn and conn.is_connected():
+                conn.close()
+        except Error as e:
+            raise Exception(f"MySQL 连接关闭失败: {str(e)}") from e
+
+
+# 暴露数据库操作工具
+db = Database()