如何使用 Python 客户端设置 SQL 查询超时？

在 python 客户端实现 sql 查询超时

在使用 python orm（如 django orm、peewee、sqlalchemy）与数据库驱动（如 pymysql）以及 fastapi、flask、django 等应用框架时，存在一个潜在的安全隐患：sql 查询可能需要长时间运行，导致应用程序挂起。为了解决这一问题，我们需要设置一个查询时间上限，以便在查询超过一定时间后自动终止。

一种优雅的实现方式是利用数据库驱动库中的超时设置：

对于 pymysql，可以设置以下参数：

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

例如，要将查询超时时间设置为 60 秒，可以如下设置：

import pymysql# 连接到数据库connection = pymysql.connect(    host='localhost',    user='user',    password='password',    database='database',    connection_timeout=60,    read_timeout=60,    write_timeout=60)

通过这种方式，当 sql 查询运行时间超过 60 秒时，数据库连接将自动终止，从而避免应用程序挂起。这是在客户端设置查询超时的优雅且有效的方法，而无需修改 mysql 服务器配置。