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中文網其他相關文章！