python 创建数据库类 使用连接池

以下是一个使用连接池的Python数据库类示例：

import psycopg2
from psycopg2 import pool

class Database:
    __connection_pool = None

    @staticmethod
    def initialise(**kwargs):
        Database.__connection_pool = pool.SimpleConnectionPool(1, 10, **kwargs)

    @classmethod
    def get_connection(cls):
        return cls.__connection_pool.getconn()

    @classmethod
    def return_connection(cls, connection):
        cls.__connection_pool.putconn(connection)

    @classmethod
    def close_all_connections(cls):
        cls.__connection_pool.closeall()

class CursorFromConnectionPool:
    def __init__(self):
        self.connection = None
        self.cursor = None

    def __enter__(self):
        self.connection = Database.get_connection()
        self.cursor = self.connection.cursor()
        return self.cursor

    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_val is not None:
            self.connection.rollback()
        else:
            self.cursor.close()
            self.connection.commit()
        Database.return_connection(self.connection)

if __name__ == '__main__':
    Database.initialise(database="your_database_name", user="your_username", password="your_password", host="your_host")
    with CursorFromConnectionPool() as cursor:
        cursor.execute("CREATE TABLE IF NOT EXISTS books (id SERIAL PRIMARY KEY, title VARCHAR(100), author VARCHAR(100), year INTEGER)")

在这个示例中，我们使用了`psycopg2`库作为我们的Python数据库驱动程序，并使用了连接池来管理数据库连接。我们创建了一个`Database`类来管理连接池，一个`CursorFromConnectionPool`类来创建使用连接池的游标。

在`Database`类中，我们使用`psycopg2`库的`SimpleConnectionPool`方法来创建连接池。我们还创建了一个静态方法`initialise`，它接受数据库连接参数，并使用它们初始化连接池。我们还有一个静态方法`get_connection`，它从连接池中获取连接。我们还有一个静态方法`return_connection`，它将连接返回到连接池中。最后，我们有一个静态方法`close_all_connections`，它关闭连接池中的所有连接。

在`CursorFromConnectionPool`类中，我们使用上下文管理器来获取连接和游标。在`__enter__`方法中，我们从连接池中获取连接，并使用该连接创建游标。在`__exit__`方法中，我们检查是否有异常，并根据情况回滚或提交事务。最后，我们将连接返回到连接池中。

在我们的示例中，我们使用`Database.initialise`方法初始化了连接池，并使用`CursorFromConnectionPool`类来创建游标。我们使用游标来执行SQL语句，这里我们创建了一个名为`books`的表。