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 成为一种强大且灵活的工具来管理数据库会话。

相关推荐
presenttttt11 分钟前
用Python和OpenCV从零搭建一个完整的双目视觉系统(六 最终篇)
开发语言·python·opencv·计算机视觉
测试19981 小时前
软件测试之压力测试总结
自动化测试·软件测试·python·测试工具·职场和发展·测试用例·压力测试
李昊哲小课2 小时前
销售数据可视化分析项目
python·信息可视化·数据分析·matplotlib·数据可视化·seaborn
烛阴2 小时前
带参数的Python装饰器原来这么简单,5分钟彻底掌握!
前端·python
全干engineer2 小时前
Flask 入门教程:用 Python 快速搭建你的第一个 Web 应用
后端·python·flask·web
nightunderblackcat3 小时前
新手向:Python网络编程,搭建简易HTTP服务器
网络·python·http
李昊哲小课3 小时前
pandas销售数据分析
人工智能·python·数据挖掘·数据分析·pandas
C嘎嘎嵌入式开发3 小时前
python之set详谈
开发语言·python
之歆4 小时前
Python-正则表达式-信息提取-滑动窗口-数据分发-文件加载及分析器-浏览器分析-学习笔记
python·学习·正则表达式
往日情怀酿做酒 V17639296384 小时前
pytorch的介绍以及张量的创建
人工智能·pytorch·python