有什么方法可以处理DB uri中的断开连接吗？

from sqlalchemy import create_engine
from sqlalchemy.pool import QueuePool

# 创建数据库引擎，配置连接池
engine = create_engine(
    'postgresql://username:password@host:port/database',
    poolclass=QueuePool,
    pool_size=10,  # 连接池大小
    max_overflow=20,  # 最大溢出连接数
    echo_pool=True  # 打印连接池日志
)

engine = create_engine(
    'postgresql://username:password@host:port/database?keepalives=1&keepalives_idle=30'
)

engine = create_engine(
    'mysql+pymysql://username:password@host:port/database',
    connect_args={'connect_timeout': 10}
)

import time
from sqlalchemy.exc import DisconnectionError

def execute_query(query):
    retries = 3
    while retries > 0:
        try:
            with engine.connect() as connection:
                result = connection.execute(query)
                return result
        except DisconnectionError:
            retries -= 1
            time.sleep(1)  # 等待1秒后重试
    raise Exception("Failed to execute query after multiple retries")