在无限循环的 Python 脚本中使用 psycopg2 连接到 PostgreSQL 数据库

有一段用 Python 编写的无限循环脚本，它连接到 PostgreSQL 数据库，并在检测到有人出现在电脑摄像头前时，向数据库插入一条记录。需要确定在脚本中连接到数据库的最佳方式，以及是否需要每次检测到有人出现时都重新连接并关闭数据库连接。

解决方案

1. 选择合适的连接方式

共有两种常见的 PostgreSQL 数据库连接方式：

• 使用 psycopg2.connect() 函数每次都创建一个新连接。
• 使用 psycopg2.pool.SimpleConnectionPool 类创建一个连接池，以便重用连接。

2. 在无限循环中使用连接池

使用连接池的好处是它可以显著提高脚本的性能，因为它可以避免每次连接和关闭数据库的开销。使用连接池的方法如下：

import psycopg2

# 创建连接池
pool = psycopg2.pool.SimpleConnectionPool(
    1,  # 最小连接数
    10,  # 最大连接数
    dbname='你的数据库名称',
    user='你的数据库用户名',
    password='你的数据库密码',
    host='你的数据库服务器地址'
)

# 使用连接池获取连接
conn = pool.getconn()

# 执行 SQL 语句
cursor = conn.cursor()
cursor.execute("INSERT INTO people (name) VALUES ('John Doe')")

# 关闭连接
cursor.close()
pool.putconn(conn)

3. 处理连接超时问题

由于连接池会自动关闭闲置的连接，当脚本长时间没有活动时，可能会出现连接超时的问题。为了解决这个问题，可以设置连接超时时间，并在出现超时时重新连接数据库。

import psycopg2

# 创建连接池
pool = psycopg2.pool.SimpleConnectionPool(
    1,  # 最小连接数
    10,  # 最大连接数
    dbname='你的数据库名称',
    user='你的数据库用户名',
    password='你的数据库密码',
    host='你的数据库服务器地址',
    timeout=300  # 连接超时时间（单位：秒）
)

# 使用连接池获取连接
conn = pool.getconn()

# 执行 SQL 语句
cursor = conn.cursor()
cursor.execute("INSERT INTO people (name) VALUES ('John Doe')")

# 关闭连接
cursor.close()
pool.putconn(conn)

4. 定期检查连接池的状态

为了确保连接池正常工作，可以定期检查连接池的状态，并根据需要重新创建连接池。

import psycopg2
import time

# 创建连接池
pool = psycopg2.pool.SimpleConnectionPool(
    1,  # 最小连接数
    10,  # 最大连接数
    dbname='你的数据库名称',
    user='你的数据库用户名',
    password='你的数据库密码',
    host='你的数据库服务器地址'
)

# 定期检查连接池的状态
while True:
    # 检查连接池中的连接数
    num_connections = pool.getnum()

    # 如果连接数为 0，则重新创建连接池
    if num_connections == 0:
        pool = psycopg2.pool.SimpleConnectionPool(
            1,  # 最小连接数
            10,  # 最大连接数
            dbname='你的数据库名称',
            user='你的数据库用户名',
            password='你的数据库密码',
            host='你的数据库服务器地址'
        )

    # 休眠 10 秒
    time.sleep(10)