如何优雅地实现 Python 客户端 SQL 查询超时?
在使用 python 进行 sql 查询时,确保查询不会无限执行至关重要。本篇文章将探讨如何在 python 客户端设置超时,以防止查询超过设定阈值。
解决方案
一种优雅的解决方案是利用数据库驱动的超时参数。例如,使用 pymysql 驱动,可以通过设置以下参数来实现超时:
connection_timeout=60 # 连接超时(以秒为单位) read_timeout=60 # 读取超时(以秒为单位) write_timeout=60 # 写入超时(以秒为单位)
这些参数指定了查询执行允许的最大时间。如果超过指定时间,则会引发异常。
其他一些 orm 和驱动也提供类似的超时参数:
- django orm: default_timeout
- peewee: mysql_db.connect(timeout=60)
- sqlalchemy: connect_args={"connect_timeout": 60}
注意: