如何在 Python 中异步操作数据库?aiomysql、asyncpg、aioredis 使用介绍

网站建设4年前发布
24 0 0

,20230306010809a7ac94418c4ac7156730635dc992f2cfcd08b2747,Python 目前已经进化到了 3.8 版本,对操作数据库也提供了相应的异步支持。当我们做一个 Web 服务时,性能的瓶颈绝大部分都在数据库上,如果一个请求从数据库中读数据的时候能够自动切换、去处理其它请求的话,是不是就能提高并发量了呢。,(编者注:原文写于 2020 年 2 月,当时最新为Python 3.8,文章内容现在仍未过时),下面我们来看看如何使用 Python 异步操作 MySQL、PostgreSQL 以及 Redis,以上几个可以说是最常用的数据库了。至于 SQLServer、Oracle,本人没有找到相应的异步驱动,有兴趣可以自己去探索一下。,而操作数据库无非就是增删改查,下面我们来看看如何异步实现它们。,异步操作 MySQL 的话,需要使用一个 aiomysql,直接 pip install aiomysql 即可。,aiomysql 底层依赖于 pymysql,所以 aiomysql 并没有单独实现相应的连接驱动,而是在 pymysql 之上进行了封装。,下面先来看看如何查询记录。,怎么样,是不是很简单呢,和同步库的操作方式其实是类似的。但是很明显,我们在获取记录的时候不会只获取一条,而是会获取多条,获取多条的话使用 await result.fetchall() 即可。,除了 fetchone、fetchall 之外,还有一个 fetchmany,可以获取指定记录的条数。,以上就是通过 aiomysql 查询数据库中的记录,没什么难度。但是值得一提的是,await conn.execute 里面除了可以传递一个原生的 SQL 语句之外,我们还可以借助 SQLAlchemy。,然后是添加记录,我们同样可以借助 SQLAlchemy 帮助我们拼接 SQL 语句。,还是很方便的,但是插入多条记录的话只会返回插入的最后一条记录的信息,所以如果你希望获取每一条的信息,那么就一条一条插入。,修改记录和添加记录是类似的,我们来看一下。,可以看到,记录被成功的修改了。,删除记录就更简单了,直接看代码。,此时数据库中的记录已经全部被删除了。,整体来看还是比较简单的,并且支持的功能也比较全面。,异步操作 PostgreSQL 的话,我们有两个选择,一个是 asyncpg 库,另一个是 aiopg 库。,asyncpg 是自己实现了一套连接驱动,而 aiopg 则是对 psycopg2 进行了封装,个人更推荐 asyncpg,性能和活跃度都比 aiopg 要好。,下面来看看如何使用 asyncpg,首先是安装,直接 pip install asyncpg 即可。,首先是查询记录。,以上我们演示了如何使用 asyncpg 来获取数据库中的记录,我们看到执行 select 语句的话,我们可以使用 conn.fetchrow(query) 来获取满足条件的单条记录,conn.fetch(query) 来获取满足条件的所有记录。,Record 对象,我们说使用 conn.fetchone 查询得到的是一个 Record 对象,使用 conn.fetch 查询得到的是多个 Record 对象组成的列表,那么这个 Rcord 对象怎么用呢?,当然我们也可以借助 SQLAlchemy 帮我们拼接 SQL 语句。,此外,conn.fetch 里面还支持占位符,使用百分号加数字的方式,举个例子:,然后是添加记录,我们看看如何往库里面添加数据。,
,通过 execute 可以插入单条记录,同时返回相关信息,但是说实话这个信息没什么太大用。除了 execute 之外,还有 executemany,用来执行多条插入语句。,注意:如果是执行大量 insert 语句的话,那么 executemany 要比 execute 快很多,但是 executemany 不具备事务功。

© 版权声明

相关文章