psycopg2關閉連接池

Question

我正在開發Flask API，我有以下代碼使用Psycopg2建立連接池。 我想我應該考慮在程序終止時關閉連接池，我該怎么做？

@contextmanager
def get_cursor(:
    global connection_pool
    if not cls.connection_pool:
        cls.connection_pool = ThreadedConnectionPool(5, 25, dsn=PoolingWrap.generate_conn_string())

    con = cls.connection_pool.getconn()
    try:
        yield con.cursor(cursor_factory=RealDictCursor)
    finally:
        cls.connection_pool.putconn(con)

Answer 1

我相信只要你正確關閉每個連接中的事務，那么離開全局池就不會有任何問題。 我認為在這種情況下可能發生的最糟糕的情況是數據庫端的某些連接需要花一點時間來確定它們已在客戶端關閉 - 但這不應該導致任何數據一致性類型問題。

但是，如果您確實要在退出之前關閉連接池，則執行此操作的一種方法是注冊atexit函數。

import atexit

@atexit.register
def close_connection_pool():
    global connection_pool
    connection_pool.closeall()