如何优雅地对 python 客户端的 sql 查询设置超时时间？

当使用 django orm、peewee、sqlalchemy 等 orm，以及 pymysql 等驱动库、fastapi、flask、django 等 app，以及 uvicorn、gunicorn 等服务器时，有时需要在超过一定阈值后终止 sql 查询，以防止安全隐患。

实现方法

为了优雅地实现此目的，可以使用 pymysql 库的参数：

python-pro

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

下载

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

• 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 查询都会超时并终止。