python - 在 Python 中设置数据库连接超时

Question

我正在创建一个需要访问数据库的 RESTful API。我正在使用 Restish、Oracle 和 SQLAlchemy。但是，我将尝试尽可能笼统地提出我的问题，而不考虑 Restish 或其他 Web API。

我希望能够为执行查询的连接设置超时。这是为了确保放弃长时间运行的查询，并丢弃(或回收)连接。此查询超时可以是一个全局值，这意味着我不需要在每次查询或连接创建时更改它。

给定以下代码:

import cx_Oracle
import sqlalchemy.pool as pool

conn_pool = pool.manage(cx_Oracle)
conn = conn_pool.connect("username/p4ss@dbname")
conn.ping()

try:
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM really_slow_query")
    print cursor.fetchone()
finally:
    cursor.close()

如何修改上面的代码来设置查询超时？ 此超时是否也适用于连接创建？

这类似于 java.sql.Statement 的 setQueryTimeout(int seconds) 方法在 Java 中的作用。

谢谢

Answer 1

对于查询，您可以查看计时器和 conn.cancel() 调用。

这些行中的内容:

t = threading.Timer(timeout,conn.cancel)
t.start()
cursor = conn.cursor()
cursor.execute(query)
res =  cursor.fetchall()
t.cancel()