psycopg2.pool 在多线程中如何使用

在多线程中使用 `psycopg2.pool`，需要创建一个连接池对象并将其传递给每个工作线程。在每个工作线程中，可以从连接池中获取一个连接对象，使用该连接对象执行数据库操作，然后将该连接对象归还给连接池。

以下是一个示例代码，展示如何在多线程中使用 `psycopg2.pool`：

import psycopg2
from psycopg2 import pool
import threading

class DBPool:
    def __init__(self, minconn, maxconn, **kwargs):
        self._pool = psycopg2.pool.SimpleConnectionPool(minconn, maxconn, **kwargs)

    def getConnection(self):
        return self._pool.getconn()

    def releaseConnection(self, conn):
        self._pool.putconn(conn)

def worker(dbpool):
    conn = dbpool.getConnection()
    # 使用连接对象执行数据库操作
    conn.cursor().execute('SELECT * FROM mytable')
    dbpool.releaseConnection(conn)

dbpool = DBPool(1, 10, host='localhost', database='mydb', user='myuser', password='mypassword')

threads = []
for i in range(10):
    t = threading.Thread(target=worker, args=(dbpool,))
    threads.append(t)
    t.start()

for t in threads:
    t.join()

dbpool._pool.closeall()

在上面的示例中，我们首先定义了一个 `DBPool` 类，该类封装了 `psycopg2.pool.SimpleConnectionPool` 对象，并提供了获取连接和释放连接的方法。在 `worker` 函数中，我们从连接池中获取一个连接对象，使用该连接对象执行数据库操作，然后将其归还给连接池。最后，我们创建了10个工作线程，并将连接池对象传递给每个线程。每个线程都会从连接池中获取一个连接对象，使用该连接对象执行数据库操作，然后将其归还给连接池。

需要注意的是，当使用 `psycopg2.pool` 连接池时，需要在每个线程中获取连接对象，并在使用完毕后将其归还给连接池。如果不归还连接对象，那么连接池中的连接将会被耗尽，从而导致连接失败。