Python程序中实现MySQL连接自动重试的方法

在Python开发中，与MySQL数据库进行交互是常见的需求。然而，由于网络波动、数据库服务器负载等原因，连接MySQL数据库时可能会遇到失败的情况。为了提高程序的稳定性和健壮性，实现连接自动重试是一个很好的解决方案。

使用try - except块实现基本重试

最基础的方法是利用try - except块。通过在try块中尝试连接MySQL，在except块中捕获连接异常，并进行重试操作。

import mysql.connector
import time

max_retries = 3
retry_delay = 1

for attempt in range(max_retries):
    try:
        connection = mysql.connector.connect(
            host='your_host',
            user='your_user',
            password='your_password',
            database='your_database'
        )
        print("成功连接到MySQL数据库")
        # 在这里进行数据库操作
        connection.close()
        break
    except mysql.connector.Error as err:
        print(f"连接失败，错误: {err}")
        if attempt < max_retries - 1:
            print(f"重试 {attempt + 1}/{max_retries}，等待 {retry_delay} 秒...")
            time.sleep(retry_delay)
        else:
            print("达到最大重试次数，连接失败。")

使用装饰器实现更优雅的重试

为了使代码结构更清晰，提高可维护性，可以使用装饰器来实现重试逻辑。

import mysql.connector
import time
import functools


def retry_on_mysql_error(max_retries=3, retry_delay=1):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            for attempt in range(max_retries):
                try:
                    return func(*args, **kwargs)
                except mysql.connector.Error as err:
                    print(f"连接失败，错误: {err}")
                    if attempt < max_retries - 1:
                        print(f"重试 {attempt + 1}/{max_retries}，等待 {retry_delay} 秒...")
                        time.sleep(retry_delay)
                    else:
                        print("达到最大重试次数，连接失败。")
        return wrapper
    return decorator


@retry_on_mysql_error(max_retries=3, retry_delay=1)
def connect_to_mysql():
    connection = mysql.connector.connect(
        host='your_host',
        user='your_user',
        password='your_password',
        database='your_database'
    )
    print("成功连接到MySQL数据库")
    # 进行数据库操作
    connection.close()


connect_to_mysql()

通过上述方法，在Python程序中实现MySQL连接自动重试变得简单高效。无论是使用基本的try - except块，还是更高级的装饰器方式，都能有效提升程序在面对数据库连接问题时的稳定性和可靠性，确保业务逻辑的正常运行。