确保psycopg2数据库连接处于活动状态

pg_connection_status 使用pqstatus实现。psycopg不公开该API,因此检查不可用。只有两个地方psycopg调用pqstatus本身是在建立新的连接时,并在开始执行时。所以是的,您需要发出一个简单的SQL语句来查明连接是否仍然存在。

这个问题真的很古老,但仍然会出现在谷歌搜索中,所以我认为知道 psycopg2.connection 实例现在有一个 closed attribute 那将是 0 当连接打开时,大于零(当连接关闭时)。以下示例应演示:

import psycopg2
import subprocess

connection = psycopg2.connect(
    dbname=database,
    user=username,
    password=password,
    host=host,
    port=port
)

print connection.closed # 0

# restart the db externally
subprocess.check_call("sudo /etc/init.d/postgresql restart", shell=True)

# this query will fail because the db is no longer connected
try:
    cur = connection.cursor()
    cur.execute('SELECT 1')
except psycopg2.OperationalError:
    pass

print connection.closed # 2

connection.closed 不反映服务器关闭/断开的连接。它只表示客户端使用 connection.close()

为了确保连接仍然有效,请读取属性 connection.isolation_level . 这将在连接断开时引发一个操作错误,pgcode==“57p01”。

这为数据库的往返增加了一点延迟,但应该比 SELECT 1 或类似的。

import psycopg2
dsn = "dbname=postgres"
conn = psycopg2.connect(dsn)

# ... some time elapses, e.g. connection within a connection pool

try:
    connection.isolation_level
except OperationalError as oe:
    conn = psycopg2.connect(dsn)

c = conn.cursor()
c.execute("SELECT 1")