Python Psycopg2连接池示例

使用psycopg2.pool.SimpleConnectionPool可有效管理数据库连接，避免频繁创建和销毁连接带来的性能开销。1. 创建连接池时指定最小和最大连接数及数据库连接参数，确保连接池初始化成功；2. 通过getconn()获取连接，执行数据库操作后使用putconn()将连接归还池中，禁止直接调用conn.close()；3. SimpleConnectionPool是线程安全的，适用于多线程环境；4. 推荐结合contextmanager实现上下文管理器，确保连接在异常时也能正确归还；5. 连接池应作为全局单例使用，并在程序退出前调用closeall()关闭所有连接。该方案适用于中小型应用，大型服务可考虑SQLAlchemy或asyncpg。

使用 psycopg2 的连接池可以有效管理数据库连接，避免频繁创建和销毁连接带来的性能开销。psycopg2 本身不提供连接池功能，但其官方推荐的 psycopg2.pool 模块（实际上是 psycopg2 包中的 pool 子模块）可以基于 threading 安全地管理连接池。

下面是一个使用 psycopg2.pool.SimpleConnectionPool 的实际例子：

✅ 基本连接池示例

import psycopg2
from psycopg2 import pool

# 创建连接池
try:
    connection_pool = psycopg2.pool.SimpleConnectionPool(
        1,                          # 最小连接数
        10,                         # 最大连接数
        host="localhost",
        database="your_db_name",
        user="your_username",
        password="your_password",
        port="5432"
    )

    if connection_pool:
        print("连接池创建成功")

except Exception as e:
    print("连接池创建失败:", e)

# 从连接池获取连接
conn = connection_pool.getconn()

if conn:
    try:
        cursor = conn.cursor()
        cursor.execute("SELECT version();")
        db_version = cursor.fetchone()
        print("数据库版本:", db_version)

        # 提交事务（如果是写操作）
        conn.commit()

    except Exception as e:
        print("数据库操作出错:", e)
        conn.rollback()
    finally:
        cursor.close()
        # 把连接归还给连接池
        connection_pool.putconn(conn)
else:
    print("无法从连接池获取连接")

# 程序结束时关闭所有连接
# connection_pool.closeall()  # 通常在应用退出时调用

? 关键点说明

1. SimpleConnectionPool 参数

• minconn: 最小连接数（启动时创建的连接数）
• maxconn: 最大连接数（连接池最多可扩展到的连接数）
• 其余参数是标准的 psycopg2.connect() 参数（如 host, user, password 等）

2. 获取和释放连接

• 使用 getconn() 获取一个连接（会自动创建或复用）
• 使用 putconn(conn) 将连接归还给池（不是关闭！）
• 不要手动调用 conn.close()，否则会真正关闭连接，导致连接池失效

3. 线程安全

• SimpleConnectionPool 是线程安全的，适合多线程环境
• 每个线程应独立获取和归还连接

? 更安全的用法（配合上下文管理器）

你可以封装一个上下文管理器，确保连接自动归还：

from contextlib import contextmanager

@contextmanager
def get_db_connection():
    conn = connection_pool.getconn()
    try:
        yield conn
    finally:
        connection_pool.putconn(conn)

# 使用示例
with get_db_connection() as conn:
    cursor = conn.cursor()
    cursor.execute("SELECT current_user;")
    print(cursor.fetchone())
    cursor.close()

这样即使发生异常，连接也会被正确归还。

⚠️ 注意事项

• 连接池对象应是全局或单例的，避免重复创建
• 应用退出时调用 connection_pool.closeall() 释放所有连接
• 高并发场景下，注意设置合理的 maxconn，避免数据库连接数超限
• 如果使用 Flask/Django 等框架，建议结合应用生命周期管理连接池

? 最后：关闭连接池

# 程序结束前
connection_pool.closeall()
print("所有连接已关闭")

基本上就这些。这个例子适合中小型应用。如果是大型服务，建议考虑使用 SQLAlchemy psycopg2 的连接池，或异步方案如 asyncpg。

以上是Python Psycopg2连接池示例的详细内容。更多信息请关注PHP中文网其他相关文章！