PyMySQL 多线程连接问题解析与解决方案

问题现象

在使用 PyMySQL 连接 MySQL 数据库时，开发者可能会遇到各种看似随机的错误，包括但不限于：

• OSError: [Errno 9] Bad file descriptor
• AttributeError: 'NoneType' object has no attribute 'settimeout'
• pymysql.err.InternalError: Packet sequence number wrong

这些错误通常不会持续出现，而是在特定条件下随机发生，给调试带来很大困难。

问题根源

经过深入分析，这些问题的根本原因是数据库连接的线程安全性问题。PyMySQL 的连接对象不是线程安全的，这意味着：

1. 同一个连接对象不能同时被多个线程使用
2. 连接对象在创建后不能被 fork 到其他进程中使用

这与 Python 数据库 API 规范(PEP 249)中关于线程安全性的定义一致，该规范明确指出："线程可以共享模块，但不能共享连接"。

解决方案

方案一：每个线程创建独立连接

最简单的解决方案是为每个线程创建独立的数据库连接：

def worker():
    # 每个线程创建自己的连接
    connection = pymysql.connect(...)
    try:
        # 使用连接
        with connection.cursor() as cursor:
            cursor.execute(...)
    finally:
        connection.close()

方案二：使用连接池

对于高并发场景，频繁创建和关闭连接会影响性能，此时可以使用数据库连接池。Python 中有多个成熟的连接池实现，如 DBUtils 等。

使用连接池的示例：

from dbutils.pooled_db import PooledDB

# 创建连接池
pool = PooledDB(
    creator=pymysql,
    maxconnections=10,
    host='localhost',
    user='user',
    password='password',
    database='dbname'
)

def worker():
    # 从池中获取连接
    connection = pool.connection()
    try:
        with connection.cursor() as cursor:
            cursor.execute(...)
    finally:
        connection.close()  # 实际上是归还到连接池

最佳实践

1. 避免共享连接：绝对不要在多个线程间共享同一个连接对象
2. 及时关闭连接：使用完毕后及时关闭连接或归还到连接池
3. 连接复用：对于频繁访问数据库的场景，使用连接池提高性能
4. 错误处理：实现完善的错误处理机制，包括连接失败、超时等情况

总结

PyMySQL 的连接线程安全性问题是数据库编程中的常见陷阱。理解并正确处理这一问题，可以避免许多难以调试的随机错误。对于简单的应用，可以为每个线程创建独立连接；对于高并发场景，建议使用专业的连接池管理数据库连接，既能保证线程安全，又能提高性能。