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

相关推荐
KimLiu2 分钟前
LCODER之Python:使用Django搭建服务端
后端·python·django
胡耀超5 分钟前
3.Python高级数据结构与文本处理
服务器·数据结构·人工智能·windows·python·大模型
1373i22 分钟前
【Python】pytorch安装(使用conda)
pytorch·python·conda
keyinf025 分钟前
python网络爬取个人学习指南-(五)
python
kida_yuan43 分钟前
【从零开始】12. 一切回归原点
python·架构·nlp
大翻哥哥1 小时前
Python 2025:AI工程化与智能代理开发实战
开发语言·人工智能·python
站大爷IP2 小时前
Python文件处理:从基础操作到实战技巧全解析
python
再努力"亿"点点2 小时前
Sklearn(机器学习)实战:鸢尾花数据集处理技巧
开发语言·python
费弗里2 小时前
无需云服务器!通过Plotly Cloud免费快捷部署Dash应用
python·dash
跟橙姐学代码2 小时前
轻松搞定 Python 模块与包导入:新手也能秒懂的入门指南
前端·python·ipython