Python 中如何实现 SQL 查询超时?
在 python client 实现 sql 查询超时
在使用 orm、驱动库、web 框架的现代 python 开发中,及时中止耗时的 sql 查询至关重要。如何优雅地实现这个目标?
方法:利用客户端参数
mysql 驱动库(如 pymysql)提供了连接超时、读取超时和写入超时等参数。通过设置这些参数,可以在客户端层控制 sql 查询的超时时间。
立即学习“Python免费学习笔记(深入)”;
例如,在使用 peewee + pymysql 的情况下,可以在连接初始化时设置以下参数:
import peeweefrom pymysql import connectionsdatabase = peewee.MySQLDatabase("my_database", **{ "connection_timeout": 60, # 60 秒连接超时 "read_timeout": 60, # 60 秒读取超时 "write_timeout": 60 # 60 秒写入超时})
这些参数将确保任何 sql 查询都不会运行超过 60 秒。一旦超时,查询将被中止。
其他注意事项