mysql 连接断开

import pymysql
from pymysql.constants import CLIENT

def connect_to_mysql():
    return pymysql.connect(
        host='localhost',
        user='user',
        password='password',
        db='database',
        client_flag=CLIENT.MULTI_STATEMENTS,
        connect_timeout=10,
        read_timeout=30,
        write_timeout=30
    )

def execute_query(query):
    while True:
        try:
            conn = connect_to_mysql()
            with conn.cursor() as cursor:
                cursor.execute(query)
                result = cursor.fetchall()
            conn.commit()
            return result
        except pymysql.err.OperationalError as e:
            if e.args[0] in (2006, 2013):  # 2006: MySQL server has gone away, 2013: Lost connection to MySQL server during query
                print("Connection lost, retrying...")
                continue
            else:
                raise
        finally:
            conn.close()

# 示例查询
query = "SELECT * FROM table_name"
result = execute_query(query)
print(result)