MySQL如何使用MySQLdb实现客户端自动重连？|极客笔记

MySQL如何使用MySQLdb实现客户端自动重连？

在使用MySQLdb连接到MySQL时，由于网络问题等原因导致连接断开的情况时，MySQLdb提供了自动重连的功能以便我们减少手动处理重连的过程。

阅读更多：MySQL 教程

安装MySQLdb

在开始之前，我们需要安装MySQLdb模块。可以通过pip进行安装：

pip install mysql-python

连接MySQL

接下来，我们需要连接到MySQL。

import MySQLdb

def connect():
    dbconfig = {'host': 'localhost', 'user': 'root', 'password': '123456', 'db': 'test', 'port': 3306}
    conn = MySQLdb.connect(**dbconfig)
    conn.autocommit(True)
    return conn

conn = connect()
cursor = conn.cursor()

开启自动重连

现在，我们需要设置MySQLdb以在连接断开时自动重连。

from MySQLdb.connections import Connection

class MyConnection(Connection):
    def __init__(self, *args, **kwargs):
        super(MyConnection, self).__init__(*args, **kwargs)

    def _execute(self, *args, **kwargs):
        try:
            return super(MyConnection, self)._execute(*args, **kwargs)
        except MySQLdb.OperationalError as e:
            if e.args[0] in (2006, 2013):
                self.ping(True)
                return super(MyConnection, self)._execute(*args, **kwargs)
            else:
                raise e

def connect():
    dbconfig = {'host': 'localhost', 'user': 'root', 'password': '123456', 'db': 'test', 'port': 3306}
    conn = MyConnection(**dbconfig)
    conn.autocommit(True)
    return conn

测试自动重连

现在，我们可以测试一下自动重连是否可以正常工作了。在连接到MySQL之后，我们可以手动停掉MySQL来测试。

from time import sleep

while True:
    try:
        cursor.execute('select 1 from dual')
        print(cursor.fetchone())
    except Exception as e:
        print('error:', e)
    sleep(1)