pymysql重连机制

import pymysql
from pymysql.constants import CLIENT

def get_connection():
    return pymysql.connect(
        host='your_host',
        user='your_user',
        password='your_password',
        db='your_db',
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor,
        client_flag=CLIENT.MULTI_STATEMENTS,
        connect_timeout=5  # 设置连接超时时间
    )

def execute_query(query):
    retries = 3  # 设置重试次数
    while retries > 0:
        try:
            connection = get_connection()
            with connection.cursor() as cursor:
                cursor.execute(query)
                result = cursor.fetchall()
            connection.commit()
            connection.close()
            return result
        except pymysql.OperationalError as e:
            if e.args[0] in (2006, 2013):  # MySQL server has gone away 或 Lost connection to MySQL server during query
                retries -= 1
                print(f"Connection lost, retrying ({retries} attempts left)...")
            else:
                raise  # 其他错误，直接抛出
        finally:
            if 'connection' in locals() and connection.open:
                connection.close()
    raise Exception("Failed to execute query after multiple retries")

# 使用示例
try:
    results = execute_query("SELECT * FROM your_table")
    print(results)
except Exception as e:
    print(f"An error occurred: {e}")