重新连接 MySQL 客户端与 MySQLdb

在数据库连接领域，保持持久连接对于不间断的数据访问至关重要。 MySQLdb 是一个用于连接 MySQL 数据库的流行 Python 库，它提供了一种在连接失败时自动重新连接客户端的方法。

在以前的版本中启用自动重新连接

在早期版本的 MySQLdb 中，可以通过 mysql_options() 函数激活自动重新连接功能，如下所示：

my_bool reconnect = 1
mysql_options(&mysql, MYSQL_OPT_RECONNECT, &reconnect)

但是，随着MySQL Connector/Python的出现，底层实现发生了变化，这种方式不再适用。

解决方案MySQL Connector/Python

要在MySQL Connector/Python中实现自动重连，需要自定义光标对象的行为。以下代码片段演示了这种方法：

import MySQLdb

class DB:
  conn = None

  def connect(self):
    self.conn = MySQLdb.connect()

  def query(self, sql):
    try:
      cursor = self.conn.cursor()
      cursor.execute(sql)
    except (AttributeError, MySQLdb.OperationalError):
      self.connect()
      cursor = self.conn.cursor()
      cursor.execute(sql)
    return cursor

db = DB()
sql = "SELECT * FROM foo"
cur = db.query(sql)
# wait a long time for the Mysql connection to timeout
cur = db.query(sql)
# still works

解释

此代码定义了一个包装 Cursor.execute() 方法的自定义 query() 方法。如果发生连接错误（AttributeError 或 MySQLdb.OperationalError），query() 方法将尝试重新连接到数据库，然后重新执行查询。

通过在游标对象，可以保证你的Python脚本即使MySQL连接间歇性丢失也能继续运行。这种方法提供了一种更强大、更可靠的方法来维护数据库连接。