如何为Python/Pandas执行的Oracle查询设置超时？

问题原因 #

pandas的read_sql()方法本身不支持timeout参数，直接传入该参数会触发TypeError，因为该方法的参数列表中未定义此选项。

可行的超时设置方案 #

1. 在cx_Oracle连接层面设置超时 #

创建cx_Oracle连接时，可直接配置连接超时与查询超时参数：

import cx_Oracle
import pandas as pd

# 创建带超时配置的连接
conn = cx_Oracle.connect(
    user="你的用户名",
    password="你的密码",
    dsn="你的Oracle连接DSN",
    connect_timeout=10,  # 连接阶段超时时间（秒）
    timeout=1           # 查询执行超时时间（秒）
)

# 正常执行read_sql查询
df = pd.read_sql(f"select count(1) from {oracledb_table_name}", conn)

2. 单独为cursor设置查询超时 #

如果不想修改全局连接的超时配置，可以针对单个cursor设置：

def get_query_dataframe(conn, sql, pd):
    # 创建cursor并设置超时
    cursor = conn.cursor()
    cursor.settimeout(1)  # 设置当前cursor的查询超时（秒）
    
    # 执行查询并转换为DataFrame
    cursor.execute(sql)
    columns = [col[0] for col in cursor.description]
    data = cursor.fetchall()
    df = pd.DataFrame(data, columns=columns)
    
    cursor.close()
    return df

3. 用Python signal模块实现超时（仅Unix系统可用） #

如果Oracle原生超时配置不适用，可借助Python的signal模块强制中断超时查询：

import signal
import pandas as pd

class TimeoutError(Exception):
    pass

def timeout_handler(signum, frame):
    raise TimeoutError("查询执行超时")

def get_query_dataframe(conn, sql, pd):
    # 设置超时信号
    signal.signal(signal.SIGALRM, timeout_handler)
    signal.alarm(1)  # 1秒后触发超时
    
    try:
        df = pd.read_sql(sql, conn)
        signal.alarm(0)  # 查询完成后取消超时
        return df
    except TimeoutError:
        conn.close()
        raise Exception("查询超时，已中断连接")

注意：该方法仅在Unix/Linux系统生效，Windows系统不支持signal模块的此用法；若查询已开始执行，需手动中断连接释放资源。

内容的提问来源于stack exchange，提问作者shaZaam