heroku - 在 psycopg2 中设置事务\查询超时？

Question

有没有办法在 psycopg2 中设置超时用于数据库事务还是用于数据库查询？

示例用例:
Heroku 将 django Web 请求限制为 30 秒，之后 Heroku 终止请求，不允许 django 优雅地回滚任何尚未返回的事务。这可以在 postgres 上留下未完成的交易。您可以在数据库中配置超时，但这也会限制非 Web 相关的查询，例如维护脚本分析等。在这种情况下，最好通过中间件 ( or via django ) 设置超时。

Answer 1

您可以使用 options 参数在连接时设置超时。语法有点奇怪:

>>> import psycopg2
>>> cnn = psycopg2.connect("dbname=test options='-c statement_timeout=1000'")
>>> cur = cnn.cursor()
>>> cur.execute("select pg_sleep(2000)")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
psycopg2.extensions.QueryCanceledError: canceling statement due to statement timeout

也可以使用 env 变量进行设置:

>>> import os
>>> os.environ['PGOPTIONS'] = '-c statement_timeout=1000'
>>> import psycopg2
>>> cnn = psycopg2.connect("dbname=test")
>>> cur = cnn.cursor()
>>> cur.execute("select pg_sleep(2000)")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
psycopg2.extensions.QueryCanceledError: canceling statement due to statement timeout