python 客户端 sql 查询超时机制

在使用 orm 框架（例如 django orm、peewee 或 sqlalchemy）、驱动库（例如 pymysql）、web 框架（例如 fastapi、flask 或 django），以及服务器（例如 uvicorn 或 gunicorn）时，如何实现 sql 查询超时终止？

例如，使用 uvicorn、fastapi、peewee 和 pymysql 时，某些 sql 查询可能需要很长时间才能完成。但是，这样做存在潜在的安全风险。我们需要设置一个查询时间上限，以防止单个 sql 查询的运行时间超过预期值。

以下是实现此目的的优雅解决方案：

python-pro

高级 Python 特性、异步编程、性能调优、静态类型、内存管理、Python 内部机制及生态库方面的专家。

下载

立即学习“Python免费学习笔记（深入）”；

pymysql

pymysql 提供了以下参数来设置超时：

• "connection_timeout": 指定与数据库建立连接的超时时间（以秒为单位）。
• "read_timeout": 指定从数据库读取数据时使用的超时时间。
• "write_timeout": 指定向数据库写入数据时使用的超时时间。

例如，以下代码将设置所有超时时间为 60 秒：

import pymysql

connection = pymysql.connect(
    connection_timeout=60,
    read_timeout=60,
    write_timeout=60,
)