get_db

get_db 函数中使用 yield 关键字,而不是直接返回 (return) 数据库会话对象,是为了实现更复杂的行为,即延迟执行和资源管理。以下是为什么要使用 yield 的几个关键原因:

1. 延迟执行(Lazy Evaluation)

使用 yield 可以创建一个生成器,它不会立即执行函数体中的代码,而是在调用者开始迭代时才逐步执行。对于数据库会话,这意味着会话的创建和关闭可以与请求处理紧密关联,确保每个请求都拥有独立的会话,并且在请求完成后会话被正确关闭。

2. 上下文管理

yield 结合 try...finally 提供了一种简洁的方式来实现上下文管理。当 FastAPI 或其他框架依赖注入系统使用 get_db 作为依赖项时,它们会在路径操作函数执行前后自动管理生成器的生命周期。finally 块确保无论是否发生异常,数据库会话都会被正确关闭。

3. 依赖注入的协同工作

在 FastAPI 等框架中,依赖注入机制能够很好地与这种模式协作。框架会负责调用生成器并处理它的输出。具体来说,FastAPI 在路径操作函数开始之前启动生成器,在路径操作函数完成之后继续执行生成器中剩余的代码(即 finally 块)。这样就保证了每个 HTTP 请求都有一个独立的数据库会话,并且在请求结束时自动关闭。

4. 避免重复代码

如果不用 yield 和生成器模式,可能会在每个需要数据库会话的地方重复编写打开和关闭数据库连接的逻辑。通过将这个逻辑封装到一个生成器函数中,可以简化代码,减少错误,并使代码更容易维护。

示例解释

考虑以下代码:

复制代码
def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()
  • 当 FastAPI 调用 get_db 时,它不会立即创建数据库会话;而是返回一个生成器对象。
  • 当路径操作函数(如 read_users)需要数据库会话时,FastAPI 会从生成器中获取下一个值(这会执行到 yield db),并将数据库会话传递给路径操作函数。
  • 在路径操作函数完成后,FastAPI 继续执行生成器中的代码,进入 finally 块并调用 db.close() 来关闭数据库会话。

这种方式不仅简化了资源管理,还提高了代码的安全性和可读性。因此,yield 是实现这一功能的关键,它使得 get_db 成为一种强大且灵活的工具来管理数据库会话。

相关推荐
PyHaVolask6 分钟前
Python进阶教程:随机数、正则表达式与异常处理
python·正则表达式·异常处理·随机数生成
折翼的恶魔22 分钟前
数据分析:合并二
python·数据分析·pandas
三体世界3 小时前
测试用例全解析:从入门到精通(1)
linux·c语言·c++·python·功能测试·测试用例·测试覆盖率
Python私教3 小时前
Django全栈班v1.04 Python基础语法 20250912 下午
后端·python·django
xchenhao3 小时前
Scikit-Learn 对糖尿病数据集(回归任务)进行全面分析
python·机器学习·回归·数据集·scikit-learn·特征·svm
xchenhao3 小时前
Scikit-learn 对加州房价数据集(回归任务)进行全面分析
python·决策树·机器学习·回归·数据集·scikit-learn·knn
这里有鱼汤3 小时前
发现一个高性能回测框架,Python + Rust,比 backtrader 快 250 倍?小团队必备!
后端·python
☼←安于亥时→❦3 小时前
数据分析之Pandas入门小结
python·pandas
带娃的IT创业者3 小时前
《Python Web部署应知应会》No3:Flask网站的性能优化和实时监测深度实战
前端·python·flask
赴3354 小时前
图像拼接案例,抠图案例
人工智能·python·计算机视觉