372 lines
13 KiB
Python
372 lines
13 KiB
Python
"""
|
||
对话列表管理服务
|
||
负责管理用户的对话列表（会话列表）
|
||
"""
|
||
|
||
import logging
|
||
from typing import Optional, Dict, Any, List
|
||
from datetime import datetime
|
||
|
||
from database import query, insert, update, get_db_type
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class ConversationService:
|
||
"""对话列表管理服务"""
|
||
|
||
@staticmethod
|
||
def generate_id() -> int:
|
||
"""生成唯一ID（使用时间戳+随机数）"""
|
||
import time
|
||
import random
|
||
return int(time.time() * 1000000) + random.randint(1000, 9999)
|
||
|
||
@staticmethod
|
||
async def create_conversation(
|
||
system_user_id: int,
|
||
context_id: str,
|
||
title: str = "新对话",
|
||
description: Optional[str] = None
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
创建新对话
|
||
|
||
Args:
|
||
system_user_id: 系统用户ID
|
||
context_id: 会话上下文ID
|
||
title: 对话标题
|
||
description: 对话描述（可选）
|
||
|
||
Returns:
|
||
创建的对话信息
|
||
"""
|
||
try:
|
||
conv_id = ConversationService.generate_id()
|
||
|
||
sql = """
|
||
INSERT INTO conversation_list
|
||
(id, context_id, system_user_id, title, description,
|
||
message_count, created_at, updated_at, is_active)
|
||
VALUES (%s, %s, %s, %s, %s, 0, NOW(), NOW(), TRUE)
|
||
"""
|
||
|
||
params = (conv_id, context_id, system_user_id, title, description)
|
||
|
||
insert(sql, params)
|
||
logger.info(f"✅ 创建对话成功: user={system_user_id}, context={context_id}, title={title}")
|
||
|
||
return {
|
||
"id": conv_id,
|
||
"context_id": context_id,
|
||
"system_user_id": system_user_id,
|
||
"title": title,
|
||
"description": description,
|
||
"message_count": 0,
|
||
"created_at": datetime.now().isoformat(),
|
||
"updated_at": datetime.now().isoformat(),
|
||
"is_active": True
|
||
}
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ 创建对话失败: {e}", exc_info=True)
|
||
raise Exception(f"创建对话失败: {str(e)}")
|
||
|
||
@staticmethod
|
||
async def get_conversation_by_context(
|
||
context_id: str
|
||
) -> Optional[Dict[str, Any]]:
|
||
"""
|
||
根据context_id获取对话信息
|
||
|
||
Args:
|
||
context_id: 会话上下文ID
|
||
|
||
Returns:
|
||
对话信息，不存在则返回None
|
||
"""
|
||
try:
|
||
sql = """
|
||
SELECT id, context_id, system_user_id, title, description,
|
||
message_count, last_message, created_at, updated_at, is_active
|
||
FROM conversation_list
|
||
WHERE context_id = %s
|
||
LIMIT 1
|
||
"""
|
||
|
||
results = query(sql, (context_id,))
|
||
|
||
if results:
|
||
return results[0]
|
||
return None
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ 获取对话信息失败: {e}", exc_info=True)
|
||
return None
|
||
|
||
@staticmethod
|
||
async def update_conversation(
|
||
context_id: str,
|
||
title: Optional[str] = None,
|
||
last_message: Optional[str] = None,
|
||
increment_message_count: bool = False
|
||
) -> bool:
|
||
"""
|
||
更新对话信息
|
||
|
||
Args:
|
||
context_id: 会话上下文ID
|
||
title: 新标题（可选）
|
||
last_message: 最后一条消息（可选）
|
||
increment_message_count: 是否增加消息计数
|
||
|
||
Returns:
|
||
是否成功更新
|
||
"""
|
||
try:
|
||
db_type = get_db_type()
|
||
|
||
# 对于StarRocks，使用DELETE+INSERT策略
|
||
if db_type == 'starrocks':
|
||
# 先获取当前记录
|
||
conv = await ConversationService.get_conversation_by_context(context_id)
|
||
if not conv:
|
||
logger.warning(f"⚠️ 对话不存在: context={context_id}")
|
||
return False
|
||
|
||
# 更新字段
|
||
if title is not None:
|
||
conv['title'] = title
|
||
if last_message is not None:
|
||
preview = last_message[:200] if len(last_message) > 200 else last_message
|
||
conv['last_message'] = preview
|
||
if increment_message_count:
|
||
conv['message_count'] = conv.get('message_count', 0) + 1
|
||
|
||
# 删除旧记录
|
||
delete_sql = "DELETE FROM conversation_list WHERE context_id = %s"
|
||
update(delete_sql, (context_id,))
|
||
|
||
# 插入新记录
|
||
insert_sql = """
|
||
INSERT INTO conversation_list
|
||
(id, system_user_id, updated_at, context_id, title, description,
|
||
message_count, last_message, created_at, is_active)
|
||
VALUES (%s, %s, NOW(), %s, %s, %s, %s, %s, %s, %s)
|
||
"""
|
||
insert(insert_sql, (
|
||
conv['id'],
|
||
conv['system_user_id'],
|
||
conv['context_id'],
|
||
conv['title'],
|
||
conv.get('description'),
|
||
conv['message_count'],
|
||
conv.get('last_message'),
|
||
conv['created_at'],
|
||
conv.get('is_active', True)
|
||
))
|
||
|
||
logger.info(f"✅ 更新对话成功(StarRocks): context={context_id}")
|
||
return True
|
||
|
||
# 对于MySQL，使用正常的UPDATE
|
||
else:
|
||
# 构建动态更新SQL
|
||
update_fields = []
|
||
params = []
|
||
|
||
if title is not None:
|
||
update_fields.append("title = %s")
|
||
params.append(title)
|
||
|
||
if last_message is not None:
|
||
# 限制预览长度为200字符
|
||
preview = last_message[:200] if len(last_message) > 200 else last_message
|
||
update_fields.append("last_message = %s")
|
||
params.append(preview)
|
||
|
||
if increment_message_count:
|
||
update_fields.append("message_count = message_count + 1")
|
||
|
||
# 总是更新 updated_at
|
||
update_fields.append("updated_at = NOW()")
|
||
|
||
if not update_fields:
|
||
return True # 没有字段需要更新
|
||
|
||
params.append(context_id)
|
||
|
||
sql = f"""
|
||
UPDATE conversation_list
|
||
SET {', '.join(update_fields)}
|
||
WHERE context_id = %s
|
||
"""
|
||
|
||
affected = update(sql, tuple(params))
|
||
|
||
if affected > 0:
|
||
logger.info(f"✅ 更新对话成功: context={context_id}")
|
||
return True
|
||
else:
|
||
logger.warning(f"⚠️ 对话不存在: context={context_id}")
|
||
return False
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ 更新对话失败: {e}", exc_info=True)
|
||
return False
|
||
|
||
@staticmethod
|
||
async def get_user_conversations(
|
||
system_user_id: int,
|
||
limit: int = 50,
|
||
only_active: bool = True
|
||
) -> List[Dict[str, Any]]:
|
||
"""
|
||
获取用户的对话列表
|
||
|
||
Args:
|
||
system_user_id: 系统用户ID
|
||
limit: 返回数量限制
|
||
only_active: 是否只返回活跃对话
|
||
|
||
Returns:
|
||
对话列表（按更新时间倒序）
|
||
"""
|
||
try:
|
||
if only_active:
|
||
sql = """
|
||
SELECT id, context_id, system_user_id, title, description,
|
||
message_count, last_message, created_at, updated_at, is_active
|
||
FROM conversation_list
|
||
WHERE system_user_id = %s AND is_active = TRUE
|
||
ORDER BY updated_at DESC
|
||
LIMIT %s
|
||
"""
|
||
else:
|
||
sql = """
|
||
SELECT id, context_id, system_user_id, title, description,
|
||
message_count, last_message, created_at, updated_at, is_active
|
||
FROM conversation_list
|
||
WHERE system_user_id = %s
|
||
ORDER BY updated_at DESC
|
||
LIMIT %s
|
||
"""
|
||
|
||
results = query(sql, (system_user_id, limit))
|
||
return results
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ 获取对话列表失败: {e}", exc_info=True)
|
||
return []
|
||
|
||
@staticmethod
|
||
async def delete_conversation(
|
||
context_id: str,
|
||
system_user_id: int,
|
||
soft_delete: bool = True
|
||
) -> bool:
|
||
"""
|
||
删除对话
|
||
|
||
Args:
|
||
context_id: 会话上下文ID
|
||
system_user_id: 系统用户ID（用于权限验证）
|
||
soft_delete: 是否软删除（仅标记为不活跃）
|
||
|
||
Returns:
|
||
是否成功删除
|
||
"""
|
||
try:
|
||
db_type = get_db_type()
|
||
|
||
# StarRocks只支持硬删除（DELETE），不支持UPDATE
|
||
if db_type == 'starrocks' or not soft_delete:
|
||
# 硬删除：物理删除记录
|
||
sql = """
|
||
DELETE FROM conversation_list
|
||
WHERE context_id = %s AND system_user_id = %s
|
||
"""
|
||
affected = update(sql, (context_id, system_user_id))
|
||
|
||
if affected > 0:
|
||
logger.info(f"✅ 删除对话成功: context={context_id}")
|
||
return True
|
||
else:
|
||
logger.warning(f"⚠️ 对话不存在或无权限: context={context_id}")
|
||
return False
|
||
|
||
# MySQL支持软删除
|
||
else:
|
||
# 软删除：仅标记为不活跃（需要先获取记录再重新插入）
|
||
conv = await ConversationService.get_conversation_by_context(context_id)
|
||
if not conv or conv['system_user_id'] != system_user_id:
|
||
logger.warning(f"⚠️ 对话不存在或无权限: context={context_id}")
|
||
return False
|
||
|
||
# 删除旧记录
|
||
delete_sql = """
|
||
DELETE FROM conversation_list
|
||
WHERE context_id = %s AND system_user_id = %s
|
||
"""
|
||
update(delete_sql, (context_id, system_user_id))
|
||
|
||
# 插入标记为不活跃的记录
|
||
insert_sql = """
|
||
INSERT INTO conversation_list
|
||
(id, system_user_id, updated_at, context_id, title, description,
|
||
message_count, last_message, created_at, is_active)
|
||
VALUES (%s, %s, NOW(), %s, %s, %s, %s, %s, %s, FALSE)
|
||
"""
|
||
insert(insert_sql, (
|
||
conv['id'],
|
||
conv['system_user_id'],
|
||
conv['context_id'],
|
||
conv['title'],
|
||
conv.get('description'),
|
||
conv['message_count'],
|
||
conv.get('last_message'),
|
||
conv['created_at']
|
||
))
|
||
|
||
logger.info(f"✅ 软删除对话成功: context={context_id}")
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ 删除对话失败: {e}", exc_info=True)
|
||
return False
|
||
|
||
@staticmethod
|
||
async def ensure_conversation_exists(
|
||
system_user_id: int,
|
||
context_id: str,
|
||
title: str = "新对话"
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
确保对话存在，不存在则创建
|
||
|
||
Args:
|
||
system_user_id: 系统用户ID
|
||
context_id: 会话上下文ID
|
||
title: 对话标题
|
||
|
||
Returns:
|
||
对话信息
|
||
"""
|
||
# 先尝试获取
|
||
conv = await ConversationService.get_conversation_by_context(context_id)
|
||
|
||
if conv:
|
||
return conv
|
||
|
||
# 不存在则创建
|
||
return await ConversationService.create_conversation(
|
||
system_user_id=system_user_id,
|
||
context_id=context_id,
|
||
title=title
|
||
)
|
||
|
||
|
||
# 创建全局服务实例
|
||
conversation_service = ConversationService()
|
||
|