psycopg2 ThreadedConnectionPool如何检测损坏但未关闭的连接

确实存在连接已完全损坏但未被标记为关闭的场景，psycopg2提供了明确的异常类型可直接识别这类无效连接，无需手动匹配pgcode或错误文本。

• 半开连接：网络闪断、防火墙/负载均衡超时切断空闲连接、数据库服务重启/主动踢掉连接时，客户端未收到TCP断开通知，此时conn.closed属性仍显示为正常状态，但执行任何数据库操作都会失败。
• 连接状态不可逆损坏：事务状态异常崩溃、连接内部缓冲区出错、服务端会话被意外终止但客户端未收到关闭信号，这类连接无法通过回滚操作恢复正常，但也不会被自动标记为已关闭。

不需要硬编码匹配pgcode或错误文本，以下两类异常抛出时即可判定连接无法复用：

• psycopg2.OperationalError：覆盖所有连接层面的故障，包括网络断开、连接被重置、服务端不可达、会话被终止等场景，是最常触发的连接损坏异常。
• psycopg2.InterfaceError：客户端侧连接对象本身状态异常时抛出，比如内部资源被意外释放、连接状态机错乱，这类异常下连接100%无法复用。

注意：部分连接损坏场景下客户端根本没收到服务端返回的错误包，不存在pgcode、pgerror属性，硬编码匹配错误字段会出现漏判。

原示例代码存在笔误（commit是连接对象的方法，不是游标方法），同时异常处理逻辑没有区分连接损坏和普通SQL错误，修正后代码如下：

from psycopg2.pool import ThreadedConnectionPool
import psycopg2

DSN = "postgresql://User:password@localhost/db"
pool = ThreadedConnectionPool(8, 10, DSN)

def execute(query):
    need_discard = False
    conn = pool.getconn()
    try:
        cur = conn.cursor()
        cur.execute(query)
        conn.commit()
        return cur.fetchall() if cur.description else None
    except (psycopg2.OperationalError, psycopg2.InterfaceError):
        # 连接损坏，标记为需要丢弃，不要执行回滚（坏连接回滚会触发二次报错）
        need_discard = True
    except Exception:
        # 普通SQL错误（语法错误、约束冲突等），连接本身正常，回滚后可归还复用
        conn.rollback()
    finally:
        pool.putconn(conn, close=need_discard)

补充说明：不需要在获取连接后主动执行SELECT 1做心跳检测，这类操作会额外增加一次数据库请求开销，靠异常捕获识别坏连接的性能损耗最低，可覆盖所有损坏场景。

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