使用psycopg3搭建数据库连接工具时的连接关闭问题求助

psycopg3的with上下文管理器会在代码块执行完毕后自动关闭连接，而游标是绑定到当前连接的——连接关闭后，游标自然无法再执行任何操作。这是psycopg3与psycopg2在连接生命周期管理上的关键差异，psycopg2中可能因为游标对连接的引用延迟了连接关闭，但psycopg3严格遵循上下文管理器的语义。

方案1：返回连接而非游标，由调用者管理生命周期 #

把函数改为返回连接对象，调用者可以自行创建游标，并且负责在使用完毕后关闭连接（或者用with管理返回的连接）。

修改后的newconnect()：

def newconnect():
    conn = psycopg.connect("host=aaa port=5432 dbname=ccc user=bbb password=xxx")
    print("Connected")
    cur = conn.execute('SELECT version()')
    db_version = cur.fetchone()
    print(db_version)
    return conn

调用方式示例：

# 方式A：手动管理连接关闭
conn = newconnect()
try:
    cur = conn.execute("SELECT * FROM your_table")
    results = cur.fetchall()
    # 处理结果
finally:
    conn.close()

# 方式B：用with管理连接（更推荐）
with newconnect() as conn:
    cur = conn.execute("SELECT * FROM your_table")
    results = cur.fetchall()
    # 处理结果

方案2：将SQL操作逻辑封装在函数内部 #

如果不需要在函数外部复用游标/连接，可以把后续要执行的SQL操作直接放在with块里，让函数完成查询并返回结果，而不是返回游标。

示例：

def execute_query(sql, params=None):
    with psycopg.connect("host=aaa port=5432 dbname=ccc user=bbb password=xxx") as conn:
        print("Connected")
        # 打印版本验证连接
        cur = conn.execute('SELECT version()')
        print(cur.fetchone())
        # 执行传入的SQL
        cur = conn.execute(sql, params)
        return cur.fetchall()

# 调用示例
results = execute_query("SELECT * FROM your_table")

方案3：手动管理连接（不推荐，易忘关闭） #

如果一定要返回游标，可以去掉with语句，手动控制连接的关闭，但必须确保调用者最后关闭连接（否则会造成连接泄漏）。

修改后的函数：

def newconnect():
    conn = psycopg.connect("host=aaa port=5432 dbname=ccc user=bbb password=xxx")
    print("Connected")
    cur = conn.execute('SELECT version()')
    db_version = cur.fetchone()
    print(db_version)
    print("Returning Cursor", cur)
    # 返回游标和连接，让调用者后续关闭连接
    return cur, conn

调用方式：

cur, conn = newconnect()
try:
    cur.execute("SELECT * FROM your_table")
    results = cur.fetchall()
finally:
    conn.close()

最推荐的是方案1或方案2：方案1保留了灵活性，让调用者可以自由操作连接；方案2则把连接管理完全封装在函数内部，避免了外部忘记关闭连接的问题。

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