diff --git a/AI/WxMini/Utils/MySQLUtil.py b/AI/WxMini/Utils/MySQLUtil.py index b2fb6bba..25b9b368 100644 --- a/AI/WxMini/Utils/MySQLUtil.py +++ b/AI/WxMini/Utils/MySQLUtil.py @@ -28,6 +28,7 @@ async def init_mysql_pool(): # 保存聊天记录到 MySQL async def save_chat_to_mysql(mysql_pool, person_id, prompt, result, audio_url, duration): async with mysql_pool.acquire() as conn: + await conn.ping() # 重置连接 async with conn.cursor() as cur: await cur.execute( "INSERT INTO t_chat_log (person_id, user_input, model_response,audio_url,duration,create_time) VALUES (%s, %s, %s, %s, %s,NOW())", @@ -39,6 +40,7 @@ async def save_chat_to_mysql(mysql_pool, person_id, prompt, result, audio_url, d # 清空表 async def truncate_chat_log(mysql_pool): async with mysql_pool.acquire() as conn: + await conn.ping() # 重置连接 async with conn.cursor() as cur: await cur.execute("TRUNCATE TABLE t_chat_log") await conn.commit() @@ -62,6 +64,7 @@ async def get_chat_log_by_session(mysql_pool, current_user, person_id, page=1, p raise ValueError("MySQL 连接池未初始化") async with mysql_pool.acquire() as conn: + await conn.ping() # 重置连接 async with conn.cursor(DictCursor) as cur: # 使用 DictCursor # 查询总记录数 await cur.execute( @@ -122,6 +125,7 @@ async def get_last_chat_log_id(mysql_pool, person_id): :return: 最后一条记录的 id,如果未找到则返回 None """ async with mysql_pool.acquire() as conn: + await conn.ping() # 重置连接 async with conn.cursor() as cur: await cur.execute( "SELECT id FROM t_chat_log WHERE person_id = %s ORDER BY id DESC LIMIT 1", @@ -134,6 +138,7 @@ async def get_last_chat_log_id(mysql_pool, person_id): # 更新为危险的记录 async def update_risk(mysql_pool, person_id, risk_memo): async with mysql_pool.acquire() as conn: + await conn.ping() # 重置连接 async with conn.cursor() as cur: # 1. 获取此人员的最后一条记录 id last_id = await get_last_chat_log_id(mysql_pool, person_id) @@ -159,6 +164,7 @@ async def get_user_by_login_name(mysql_pool, login_name: str) -> Optional[Dict]: :return: 用户信息(字典形式) """ async with mysql_pool.acquire() as conn: + await conn.ping() # 重置连接 async with conn.cursor() as cursor: sql = "SELECT * FROM t_base_person WHERE login_name = %s" await cursor.execute(sql, (login_name,)) @@ -181,6 +187,7 @@ async def get_chat_logs_by_risk_flag(mysql_pool, risk_flag: int, offset: int, pa :return: 聊天记录列表和总记录数 """ async with mysql_pool.acquire() as conn: + await conn.ping() # 重置连接 async with conn.cursor() as cursor: # 查询符合条件的记录 sql = """