Python 中如何设置 SQL 查询超时

在使用 Python 进行数据库操作时，设置 SQL 查询超时是一项重要的任务。它可以帮助我们避免长时间运行的查询，从而提高应用程序的性能和稳定性。本文将介绍在 Python 中设置 SQL 查询超时的方法。

一、使用数据库驱动

不同的数据库有各自对应的 Python 驱动，例如，对于 MySQL 数据库，我们常用 mysql-connector-python；对于 PostgreSQL 数据库，常用 psycopg2。在这些驱动中，一般都提供了设置查询超时的参数。

以 mysql-connector-python 为例，我们可以在连接数据库时设置超时时间。如下代码：

import mysql.connector

try:
    cnx = mysql.connector.connect(user='your_username', password='your_password',
                                  host='your_host', database='your_database',
                                  connection_timeout=10)  # 设置连接超时为 10 秒
    cursor = cnx.cursor()
    cursor.execute('SELECT * FROM your_table', timeout=5)  # 设置查询超时为 5 秒
    results = cursor.fetchall()
    for row in results:
        print(row)
    cursor.close()
    cnx.close()
except mysql.connector.Error as err:
    print(f"Error: {err}")

在这个示例中，connection_timeout 用于设置连接数据库的超时时间，timeout 则用于设置执行 SQL 查询的超时时间。

二、使用上下文管理器

上下文管理器可以帮助我们更好地管理资源，同时也能方便地设置查询超时。例如，在使用 psycopg2 操作 PostgreSQL 数据库时：

import psycopg2
from contextlib import contextmanager

@contextmanager
def get_connection():
    try:
        conn = psycopg2.connect(user='your_username', password='your_password',
                                host='your_host', database='your_database')
        yield conn
        conn.commit()
    except psycopg2.Error as e:
        print(f"Error: {e}")
        conn.rollback()
    finally:
        conn.close()

with get_connection() as conn:
    cursor = conn.cursor()
    cursor.execute('SELECT * FROM your_table', timeout=3)  # 设置查询超时为 3 秒
    results = cursor.fetchall()
    cursor.close()

三、异常处理

设置查询超时后，当查询超时时会抛出相应的异常。我们需要对这些异常进行捕获和处理，以确保程序的稳定性。例如：

try:
    cursor.execute('SELECT * FROM your_table', timeout=5)
    results = cursor.fetchall()
except TimeoutError as te:
    print(f"Query timed out: {te}")

通过上述方法，我们可以在 Python 中有效地设置 SQL 查询超时，从而优化数据库操作的性能，避免长时间运行的查询对系统造成不良影响。在实际应用中，应根据具体需求合理设置超时时间，以达到最佳的效果。