mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-17 10:27:36 +08:00
169488851f
主要变更：
1. **数据库集成**:
* 引入 MySQL 数据库支持，使用 SQLAlchemy 和 `databases` 库持久化存储应用程序设置。
* 添加了 `app/database` 目录，包含数据库连接、模型和初始化逻辑。
* 更新 `requirements.txt` 添加数据库相关依赖 (`pymysql`, `sqlalchemy`, `aiomysql`, `databases`, `python-dotenv`)。
2. **配置管理重构**:
* 重构 `ConfigService` (`app/service/config/config_service.py`)，使其从数据库加载和保存设置，并支持从 `.env` 文件同步初始配置到数据库。
* 修改 `Settings` 模型 (`app/config/config.py`) 以包含数据库连接信息，并添加了从数据库加载/同步配置的逻辑。
* 配置相关的路由 (`app/router/config_routes.py`) 更新为异步，并调用新的 `ConfigService` 方法。
* `KeyManager` (`app/service/key/key_manager.py`) 现在可以在配置更新后重置和重新初始化。
3. **错误日志查看器**:
* 新增 `/logs` 页面 (`app/templates/error_logs.html`) 用于展示应用程序错误日志。
* 添加了相应的路由 (`app/router/log_routes.py`)、静态资源 (`app/static/css/error_logs.css`, `app/static/js/error_logs.js`) 和日志记录器 (`app/log/logger.py`)。
* 在配置页面和密钥管理页面的导航栏中添加了指向日志页面的链接。
4. **异步操作**:
* 将配置服务和相关路由转换为异步 (`async def`) 以支持异步数据库操作。
5. **其他**:
* 更新了应用程序初始化逻辑 (`app/core/application.py`, `app/core/initialization.py`) 以包含数据库连接的建立和关闭。
50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
"""
|
|
数据库连接池模块
|
|
"""
|
|
from databases import Database
|
|
from sqlalchemy import create_engine, MetaData
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
|
|
from app.config.config import settings
|
|
from app.log.logger import get_database_logger
|
|
|
|
logger = get_database_logger()
|
|
|
|
# 数据库URL
|
|
DATABASE_URL = f"mysql+pymysql://{settings.MYSQL_USER}:{settings.MYSQL_PASSWORD}@{settings.MYSQL_HOST}:{settings.MYSQL_PORT}/{settings.MYSQL_DATABASE}"
|
|
|
|
# 创建数据库引擎
|
|
engine = create_engine(DATABASE_URL)
|
|
|
|
# 创建元数据对象
|
|
metadata = MetaData()
|
|
|
|
# 创建基类
|
|
Base = declarative_base(metadata=metadata)
|
|
|
|
# 创建数据库连接池
|
|
database = Database(DATABASE_URL)
|
|
|
|
|
|
async def connect_to_db():
|
|
"""
|
|
连接到数据库
|
|
"""
|
|
try:
|
|
await database.connect()
|
|
logger.info("Connected to database")
|
|
except Exception as e:
|
|
logger.error(f"Failed to connect to database: {str(e)}")
|
|
raise
|
|
|
|
|
|
async def disconnect_from_db():
|
|
"""
|
|
断开数据库连接
|
|
"""
|
|
try:
|
|
await database.disconnect()
|
|
logger.info("Disconnected from database")
|
|
except Exception as e:
|
|
logger.error(f"Failed to disconnect from database: {str(e)}")
|