SQLAlchemy 是 Python 中最流行的数据库工具之一,在新版本中引入了对异步操作的支持。这为使用异步框架(如 FastAPI)开发应用程序带来了极大的便利。在这篇文章中,简单介绍下 SQLAlchemy 是如何利用 Greenlet 实现异步操作的。
什么是 Greenlet?
Greenlet 是一个轻量级的第三方协程库,最初由 PyPy 项目开发。它允许在单个操作系统线程内执行多个任务(协程),这些任务可以相互间切换,而无需线程的上下文切换开销。Greenlet 提供了更细粒度的控制,可以在协程之间显式切换。
SQLAlchemy 中的异步支持
在 SQLAlchemy 1.4 中,引入了对异步操作的支持。这一功能的核心是通过 asyncio
和 greenlet
实现的。asyncio
是 Python 的内置库,用于编写异步代码,而 greenlet
则用于在协程之间同步和异步代码切换。
使用 Greenlet 桥接上下层异步接口
SQLAlchemy 的异步实现很巧妙,它通过 Greenlet 桥接了上层的异步接口和底层的异步驱动接口,使得中间层可以继续使用同步代码执行。
当你执行一个异步数据库操作时,SQLAlchemy会进行以下过程:
- SQLAlchemy 会启动一个 Greenlet,将该操作交给 Greenlet 处理。
- 在 Greenlet 中,代码以同步方式继续执行,直到需要调用底层异步接口。
- 调用底层异步接口时,SQLAlchemy 会返回一个协程对象,此时 Greenlet 会切换出去,并将控制权返回给上层的异步接口。
- 上层异步接口接收到返回的协程对象后,使用
await
关键字进入 Python 的异步环境继续执行。当下次需要调用异步接口时,重复上述过程,形成一个闭环。
这种机制使得 SQLAlchemy 只需要适配少量代码,提供强大的异步数据库操作支持。
官方给出的示意图
Greenlet 切换到同步的源代码
结论
我们可以利用这一思想,让同步代码在真正的异步环境中运行。只需正确封装上下层接口,就能在同步代码中享受异步的并发能力。