parent
af96259f57
commit
7a12b2bbea
@ -0,0 +1,167 @@
|
||||
import logging
|
||||
import asyncio
|
||||
import uuid
|
||||
from aiomysql import create_pool
|
||||
from datetime import datetime
|
||||
|
||||
# MYSQL配置信息
|
||||
MYSQL_HOST = "10.10.14.203"
|
||||
MYSQL_PORT = 3306
|
||||
MYSQL_USER = "root"
|
||||
MYSQL_PASSWORD = "Password123@mysql"
|
||||
MYSQL_DB_NAME = "xiaozhi_esp32_server"
|
||||
|
||||
# 配置日志
|
||||
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# MySQL 配置
|
||||
MYSQL_CONFIG = {
|
||||
"host": MYSQL_HOST,
|
||||
"port": MYSQL_PORT,
|
||||
"user": MYSQL_USER,
|
||||
"password": MYSQL_PASSWORD,
|
||||
"db": MYSQL_DB_NAME,
|
||||
"minsize": 1,
|
||||
"maxsize": 20,
|
||||
"autocommit": True,
|
||||
"charset": "utf8mb4",
|
||||
}
|
||||
|
||||
|
||||
async def init_mysql_pool():
|
||||
"""初始化 MySQL 连接池"""
|
||||
try:
|
||||
pool = await create_pool(**MYSQL_CONFIG)
|
||||
logger.info("MySQL连接池创建成功")
|
||||
return pool
|
||||
except Exception as e:
|
||||
logger.error(f"创建MySQL连接池失败: {str(e)}")
|
||||
raise
|
||||
|
||||
|
||||
async def truncate_chat_log(pool):
|
||||
"""清空聊天历史表"""
|
||||
try:
|
||||
async with pool.acquire() as conn:
|
||||
await conn.ping()
|
||||
async with conn.cursor() as cur:
|
||||
await cur.execute("TRUNCATE TABLE t_chat_history")
|
||||
logger.info("表 t_chat_history 已清空")
|
||||
except Exception as e:
|
||||
logger.error(f"清空表失败: {str(e)}")
|
||||
raise
|
||||
|
||||
|
||||
async def insert_chat_log(pool, chat_text, chat_wav, ai_text, person_id, person_name):
|
||||
"""插入聊天记录"""
|
||||
try:
|
||||
async with pool.acquire() as conn:
|
||||
await conn.ping()
|
||||
async with conn.cursor() as cur:
|
||||
sql = """
|
||||
INSERT INTO t_chat_history
|
||||
(chat_text, chat_wav, ai_text, person_id, person_name)
|
||||
VALUES (%s, %s, %s, %s, %s)
|
||||
"""
|
||||
await cur.execute(sql, (chat_text, chat_wav, ai_text, person_id, person_name))
|
||||
logger.info(f"成功插入聊天记录: {person_name}")
|
||||
return cur.lastrowid
|
||||
except Exception as e:
|
||||
logger.error(f"插入聊天记录失败: {str(e)}")
|
||||
raise
|
||||
|
||||
|
||||
async def update_chat_log(pool, chat_id, chat_text=None, chat_wav=None, ai_text=None, person_name=None):
|
||||
"""更新聊天记录"""
|
||||
try:
|
||||
async with pool.acquire() as conn:
|
||||
await conn.ping()
|
||||
async with conn.cursor() as cur:
|
||||
# 构建更新语句
|
||||
updates = []
|
||||
params = []
|
||||
if chat_text is not None:
|
||||
updates.append("chat_text = %s")
|
||||
params.append(chat_text)
|
||||
if chat_wav is not None:
|
||||
updates.append("chat_wav = %s")
|
||||
params.append(chat_wav)
|
||||
if ai_text is not None:
|
||||
updates.append("ai_text = %s")
|
||||
params.append(ai_text)
|
||||
if person_name is not None:
|
||||
updates.append("person_name = %s")
|
||||
params.append(person_name)
|
||||
|
||||
if not updates:
|
||||
logger.warning("没有提供要更新的字段")
|
||||
return
|
||||
|
||||
params.append(chat_id)
|
||||
sql = f"UPDATE t_chat_history SET {', '.join(updates)} WHERE id = %s"
|
||||
await cur.execute(sql, params)
|
||||
logger.info(f"成功更新聊天记录 ID: {chat_id}")
|
||||
except Exception as e:
|
||||
logger.error(f"更新聊天记录失败: {str(e)}")
|
||||
raise
|
||||
|
||||
|
||||
async def delete_chat_log(pool, chat_id):
|
||||
"""删除聊天记录"""
|
||||
try:
|
||||
async with pool.acquire() as conn:
|
||||
await conn.ping()
|
||||
async with conn.cursor() as cur:
|
||||
sql = "DELETE FROM t_chat_history WHERE id = %s"
|
||||
await cur.execute(sql, (chat_id,))
|
||||
logger.info(f"成功删除聊天记录 ID: {chat_id}")
|
||||
except Exception as e:
|
||||
logger.error(f"删除聊天记录失败: {str(e)}")
|
||||
raise
|
||||
|
||||
|
||||
async def main():
|
||||
"""主函数"""
|
||||
pool = None
|
||||
try:
|
||||
# 初始化连接池
|
||||
pool = await init_mysql_pool()
|
||||
|
||||
# 清空表
|
||||
await truncate_chat_log(pool)
|
||||
|
||||
# 测试插入
|
||||
person_id = str(uuid.uuid4())
|
||||
chat_id = await insert_chat_log(
|
||||
pool=pool,
|
||||
chat_text="你好，小智",
|
||||
chat_wav="test.wav",
|
||||
ai_text="你好，有什么我可以帮你的吗？",
|
||||
person_id=person_id,
|
||||
person_name="测试用户"
|
||||
)
|
||||
logger.info(f"插入的聊天记录ID: {chat_id}")
|
||||
|
||||
# 测试更新
|
||||
await update_chat_log(
|
||||
pool=pool,
|
||||
chat_id=chat_id,
|
||||
ai_text="你好，很高兴为你服务！",
|
||||
person_name="更新后的用户"
|
||||
)
|
||||
|
||||
# 测试删除
|
||||
await delete_chat_log(pool, chat_id)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"程序执行出错: {str(e)}")
|
||||
finally:
|
||||
if pool:
|
||||
pool.close()
|
||||
await pool.wait_closed()
|
||||
logger.info("MySQL连接池已关闭")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
asyncio.run(main())
|
||||
Loading…
Reference in new issue