学生管理系统代码提交
This commit is contained in:
+36
@@ -0,0 +1,36 @@
|
||||
# database.py
|
||||
# 本文件负责配置数据库连接、创建引擎、会话工厂,并提供依赖注入函数(同步模式)
|
||||
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker, declarative_base
|
||||
|
||||
from config import settings
|
||||
|
||||
# 1. 配置 MySQL 数据库连接 URL(从 config 读取,支持环境变量覆盖)
|
||||
SQLALCHEMY_DATABASE_URL = settings.DATABASE_URL
|
||||
|
||||
# 2. 创建数据库引擎
|
||||
# - pool_pre_ping=True:每次取连接前先 ping,防止使用已断开的连接
|
||||
# - pool_recycle=3600:连接最长复用 1 小时,避免 MySQL wait_timeout 断连
|
||||
engine = create_engine(
|
||||
SQLALCHEMY_DATABASE_URL,
|
||||
pool_pre_ping=True,
|
||||
pool_recycle=3600,
|
||||
echo=settings.DB_ECHO, # True 会在控制台打印所有 SQL,便于调试
|
||||
)
|
||||
|
||||
# 3. 创建会话工厂:不自动提交,需要手动 commit
|
||||
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
|
||||
# 4. 声明式基类,所有模型类都继承自它
|
||||
Base = declarative_base()
|
||||
|
||||
|
||||
# 5. 依赖注入函数:用于 FastAPI 路由中获取数据库会话
|
||||
def get_db():
|
||||
"""每次请求创建一个数据库会话,请求结束后关闭"""
|
||||
db = SessionLocal()
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
||||
Reference in New Issue
Block a user