Python 客户端设置 SQL 查询超时的方法

在使用 Python 客户端进行 SQL 查询时，合理设置查询超时时间至关重要。这不仅可以避免因长时间运行的查询导致程序无响应，还能有效提高系统的稳定性和性能。

对于不同的数据库连接库，设置 SQL 查询超时的方式各有不同。以常用的 psycopg2 库连接 PostgreSQL 数据库为例，在建立数据库连接时，可以通过设置 connect_timeout 参数来控制连接超时时间。而在执行查询操作时，可以使用 cursor.execute() 方法的 timeout 参数来设置查询超时。比如：

import psycopg2

# 建立数据库连接
conn = psycopg2.connect(
    database="your_database",
    user="your_user",
    password="your_password",
    host="your_host",
    port="your_port",
    connect_timeout=10
)

cur = conn.cursor()
try:
    cur.execute("SELECT * FROM your_table", timeout=30)
    rows = cur.fetchall()
    for row in rows:
        print(row)
except psycopg2.OperationalError as e:
    if "query has timed out" in str(e):
        print("查询超时")
    else:
        print(f"其他错误: {e}")
finally:
    cur.close()
    conn.close()

若使用 mysql-connector-python 库连接 MySQL 数据库，虽然该库没有直接提供查询超时的参数设置，但可以通过在查询语句中添加 SQL_NO_CACHE 提示，结合设置连接的 read_timeout 和 write_timeout 来间接控制查询时间。示例代码如下：

import mysql.connector

mydb = mysql.connector.connect(
    host="your_host",
    user="your_user",
    password="your_password",
    database="your_database",
    read_timeout=30,
    write_timeout=30
)

mycursor = mydb.cursor()
try:
    mycursor.execute("SELECT SQL_NO_CACHE * FROM your_table")
    myresult = mycursor.fetchall()
    for x in myresult:
        print(x)
except mysql.connector.Error as e:
    if "MySQL Connection not available" in str(e):
        print("查询超时")
    else:
        print(f"其他错误: {e}")
finally:
    mycursor.close()
    mydb.close()

通过上述方法，在 Python 客户端中合理设置 SQL 查询超时，能够有效保障程序的健壮性，提升用户体验。在实际应用中，需根据数据库的特点和业务需求，灵活调整超时时间，以达到最佳的性能平衡。