如何优雅地为 Python 客户端的 SQL 查询设置超时时间?
如何优雅地对 python 客户端的 sql 查询设置超时时间?
当使用 django orm、peewee、sqlalchemy 等 orm,以及 pymysql 等驱动库、fastapi、flask、django 等 app,以及 uvicorn、gunicorn 等服务器时,有时需要在超过一定阈值后终止 sql 查询,以防止安全隐患。
实现方法
为了优雅地实现此目的,可以使用 pymysql 库的参数:
- connection_timeout: 数据库连接超时时间(以秒为单位)
- read_timeout: 数据库读操作超时时间(以秒为单位)
- write_timeout: 数据库写操作超时时间(以秒为单位)
示例
在使用 pymysql 时,可以通过设置以下参数来限制 sql 查询时间:
import pymysql # 打开数据库连接 connection = pymysql.connect( host="localhost", user="root", password="password", database="database_name", connection_timeout=60, read_timeout=60, write_timeout=60, )
通过将 read_timeout 设置为 60 秒,任何超过 60 秒的 sql 查询都会超时并终止。