FastAPI怎么使用依赖注入Depends传递数据库连接-Python教程-PHP中文网

必须用 yield + try/finally，否则连接不释放导致连接池耗尽；直接 return 会使 session 无法自动关闭，引发 timeouterror。

fastapi怎么使用依赖注入depends传递数据库连接

直接用 Depends(get_db) 就能传数据库连接，但必须配 yield + try/finally，否则连接不释放、池子会爆。

为什么不能直接 return 数据库 session

每次 get_db() 被调用时，SessionLocal() 都会新建一个 session 实例。如果只用 return，FastAPI 拿到 session 后就不管了——连接不会关，事务不会 rollback，连接池里的空闲连接数会持续上涨，直到拒绝新连接。

常见错误现象：sqlalchemy.exc.TimeoutError: QueuePool limit of size 5 overflow 10 reached

正确做法是用 yield 让 FastAPI 在路径函数执行完后自动触发清理逻辑：

yield db 把 session 注入到路由参数中
finally: db.close() 确保无论是否出错，session 都被关闭
注意：不是 db.rollback() 或 db.commit() ——这些该由业务层显式控制

get_db 函数怎么写才安全

标准写法（同步版）：

from sqlalchemy.orm import Session
from myapp.db import SessionLocal

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

异步版（搭配 AsyncSession）：

Fastapi Code Review

审查 FastAPI 代码的路由模式、依赖注入、验证和异步处理器。适用于审查 FastAPI 应用、检查 APIRouter 配置、依赖注入等。

下载

from sqlalchemy.ext.asyncio import AsyncSession
from myapp.db import async_session

async def get_db():
    async with async_session() as db:
        yield db

关键点：

不要在 yield 前做 db.begin() ——事务边界应由路径函数或服务层决定
不要在 yield 后写业务逻辑 ——那部分代码在响应已发回客户端后才执行
若依赖里抛了 HTTPException，finally 仍会执行；但若 yield 后直接 raise，客户端可能收不到错误（见下一条）

Depends(get_db) 在路由里怎么用

类型注解必须匹配 yield 的返回值类型，否则 FastAPI 不识别：

@app.get("/users/{user_id}")
def get_user(user_id: int, db: Session = Depends(get_db)):
    return db.query(User).filter(User.id == user_id).first()

注意：

db: Session 中的 Session 类型要和 yield db 的实际类型一致（比如别写成 AsyncSession 却用同步 SessionLocal）
同一个请求内多次 Depends(get_db)（比如嵌套依赖里也用了）只会执行一次 get_db，FastAPI 自动缓存结果
别把 get_db 包进闭包或 lambda ——FastAPI 判定“相同依赖”靠的是 callable 对象身份，不是内容