317 lines
9.8 KiB
Go
317 lines
9.8 KiB
Go
package agentsvc
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"log"
|
||
"sort"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"github.com/LoveLosita/smartflow/backend/pkg"
|
||
"github.com/LoveLosita/smartflow/backend/respond"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// GetConversationHistory 返回指定会话的聊天历史。
|
||
//
|
||
// 职责边界：
|
||
// 1. 负责会话 ID 归一化、会话归属校验，以及“先 Redis、后 DB”的读取编排；
|
||
// 2. 负责把缓存消息 / DB 记录统一转换为 API 响应 DTO；
|
||
// 3. 不负责补写会话标题，也不负责修改聊天主链路的缓存写入策略。
|
||
func (s *AgentService) GetConversationHistory(ctx context.Context, userID int, chatID string) ([]model.GetConversationHistoryItem, error) {
|
||
normalizedChatID := strings.TrimSpace(chatID)
|
||
if normalizedChatID == "" {
|
||
return nil, respond.MissingParam
|
||
}
|
||
|
||
// 1. 先做归属校验：
|
||
// 1.1 Redis 历史缓存只按 chat_id 分桶，不能单靠缓存判断用户归属；
|
||
// 1.2 因此先查会话是否属于当前用户，避免命中别人会话缓存时产生越权读取；
|
||
// 1.3 若会话不存在，统一返回 gorm.ErrRecordNotFound，交由 API 层映射为参数错误。
|
||
exists, err := s.repo.IfChatExists(ctx, userID, normalizedChatID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if !exists {
|
||
return nil, gorm.ErrRecordNotFound
|
||
}
|
||
|
||
// 2. 优先读取“会话历史视图缓存”：
|
||
// 2.1 这层缓存专门服务 conversation-history，字段口径与前端展示一致；
|
||
// 2.2 与 Agent 上下文热缓存解耦，避免为了历史多版本而拖慢首 token；
|
||
// 2.3 若命中则直接返回，miss 再回源 DB。
|
||
if s.cacheDAO != nil {
|
||
items, cacheErr := s.cacheDAO.GetConversationHistoryFromCache(ctx, userID, normalizedChatID)
|
||
if cacheErr != nil {
|
||
log.Printf("读取会话历史视图缓存失败 chat_id=%s: %v", normalizedChatID, cacheErr)
|
||
} else if items != nil {
|
||
return items, nil
|
||
}
|
||
}
|
||
|
||
// 3. Redis miss 时回源 DB：
|
||
// 3.1 复用现有 GetUserChatHistories 读取最近 N 条历史，保证“重试版本、落库主键、创建时间”口径稳定；
|
||
// 3.2 再把 DB 结果转换成接口 DTO，作为历史视图缓存回填；
|
||
// 3.3 失败时直接上抛，由 API 层统一处理。
|
||
histories, err := s.repo.GetUserChatHistories(ctx, userID, pkg.HistoryFetchLimitByModel("worker"), normalizedChatID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
items := buildConversationHistoryItemsFromDB(histories)
|
||
|
||
if s.cacheDAO != nil {
|
||
if setErr := s.cacheDAO.SetConversationHistoryToCache(ctx, userID, normalizedChatID, items); setErr != nil {
|
||
log.Printf("回填会话历史视图缓存失败 chat_id=%s: %v", normalizedChatID, setErr)
|
||
}
|
||
}
|
||
|
||
return items, nil
|
||
}
|
||
|
||
// appendConversationHistoryCacheOptimistically 把“刚生成但尚未完成 DB 持久化确认”的消息追加到历史视图缓存。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只服务前端会话历史展示，不参与 Agent 上下文热缓存；
|
||
// 2. 优先复用现有历史视图缓存，miss 时再用 DB 历史做一次启动兜底；
|
||
// 3. 不保证最终权威性，最终仍以 DB 落库成功后的缓存失效与回源结果为准。
|
||
func (s *AgentService) appendConversationHistoryCacheOptimistically(
|
||
ctx context.Context,
|
||
userID int,
|
||
chatID string,
|
||
newItems ...model.GetConversationHistoryItem,
|
||
) {
|
||
if s == nil || s.cacheDAO == nil {
|
||
return
|
||
}
|
||
normalizedChatID := strings.TrimSpace(chatID)
|
||
if userID <= 0 || normalizedChatID == "" || len(newItems) == 0 {
|
||
return
|
||
}
|
||
if ctx == nil {
|
||
ctx = context.Background()
|
||
}
|
||
|
||
// 1. 优先取历史视图缓存，避免每轮乐观追加都回源 DB。
|
||
items, err := s.cacheDAO.GetConversationHistoryFromCache(ctx, userID, normalizedChatID)
|
||
if err != nil {
|
||
log.Printf("读取会话历史视图缓存失败 chat_id=%s: %v", normalizedChatID, err)
|
||
return
|
||
}
|
||
|
||
// 2. 缓存 miss 时，用当前 DB 已有历史做一次基线兜底。
|
||
// 2.1 这样即便本轮是“缓存刚被 retry 补种操作删掉”，也不会只留下最新两条消息；
|
||
// 2.2 失败策略：DB 兜底失败只记日志并跳过，不阻塞主回复流程。
|
||
if items == nil {
|
||
histories, hisErr := s.repo.GetUserChatHistories(ctx, userID, pkg.HistoryFetchLimitByModel("worker"), normalizedChatID)
|
||
if hisErr != nil {
|
||
log.Printf("乐观追加历史缓存时回源 DB 失败 chat_id=%s: %v", normalizedChatID, hisErr)
|
||
return
|
||
}
|
||
items = buildConversationHistoryItemsFromDB(histories)
|
||
}
|
||
|
||
merged := append([]model.GetConversationHistoryItem(nil), items...)
|
||
for _, item := range newItems {
|
||
merged = appendConversationHistoryItemIfMissing(merged, item)
|
||
}
|
||
sortConversationHistoryItems(merged)
|
||
merged = attachConversationRetryTotals(merged)
|
||
|
||
if err = s.cacheDAO.SetConversationHistoryToCache(ctx, userID, normalizedChatID, merged); err != nil {
|
||
log.Printf("乐观追加会话历史视图缓存失败 chat_id=%s: %v", normalizedChatID, err)
|
||
}
|
||
}
|
||
|
||
// buildConversationHistoryItemsFromDB 把数据库聊天记录转换为接口响应。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只透传 DB 已有字段，不尝试补算 reasoning_content；
|
||
// 2. message_content / role 为空时兜底为空串与 system，避免空指针影响接口；
|
||
// 3. 保持 DAO 返回的时间正序，前端可直接渲染。
|
||
func buildConversationHistoryItemsFromDB(histories []model.ChatHistory) []model.GetConversationHistoryItem {
|
||
items := make([]model.GetConversationHistoryItem, 0, len(histories))
|
||
for _, history := range histories {
|
||
content := ""
|
||
if history.MessageContent != nil {
|
||
content = strings.TrimSpace(*history.MessageContent)
|
||
}
|
||
|
||
role := "system"
|
||
if history.Role != nil {
|
||
role = normalizeConversationHistoryRole(*history.Role)
|
||
}
|
||
|
||
items = append(items, model.GetConversationHistoryItem{
|
||
ID: history.ID,
|
||
Role: role,
|
||
Content: content,
|
||
CreatedAt: history.CreatedAt,
|
||
ReasoningContent: strings.TrimSpace(derefConversationHistoryText(history.ReasoningContent)),
|
||
ReasoningDurationSeconds: history.ReasoningDurationSeconds,
|
||
RetryGroupID: cloneConversationStringPointer(history.RetryGroupID),
|
||
RetryIndex: cloneConversationIntPointer(history.RetryIndex),
|
||
})
|
||
}
|
||
return attachConversationRetryTotals(items)
|
||
}
|
||
|
||
func derefConversationHistoryText(text *string) string {
|
||
if text == nil {
|
||
return ""
|
||
}
|
||
return *text
|
||
}
|
||
|
||
func attachConversationRetryTotals(items []model.GetConversationHistoryItem) []model.GetConversationHistoryItem {
|
||
if len(items) == 0 {
|
||
return items
|
||
}
|
||
groupTotals := make(map[string]int)
|
||
for _, item := range items {
|
||
if item.RetryGroupID == nil || item.RetryIndex == nil {
|
||
continue
|
||
}
|
||
groupID := strings.TrimSpace(*item.RetryGroupID)
|
||
if groupID == "" {
|
||
continue
|
||
}
|
||
if *item.RetryIndex > groupTotals[groupID] {
|
||
groupTotals[groupID] = *item.RetryIndex
|
||
}
|
||
}
|
||
for idx := range items {
|
||
groupIDPtr := items[idx].RetryGroupID
|
||
if groupIDPtr == nil {
|
||
continue
|
||
}
|
||
groupID := strings.TrimSpace(*groupIDPtr)
|
||
total := groupTotals[groupID]
|
||
if total <= 0 {
|
||
continue
|
||
}
|
||
totalCopy := total
|
||
items[idx].RetryTotal = &totalCopy
|
||
}
|
||
return items
|
||
}
|
||
|
||
func cloneConversationStringPointer(src *string) *string {
|
||
if src == nil {
|
||
return nil
|
||
}
|
||
text := strings.TrimSpace(*src)
|
||
if text == "" {
|
||
return nil
|
||
}
|
||
return &text
|
||
}
|
||
|
||
func cloneConversationIntPointer(src *int) *int {
|
||
if src == nil || *src <= 0 {
|
||
return nil
|
||
}
|
||
value := *src
|
||
return &value
|
||
}
|
||
|
||
func normalizeConversationHistoryRole(role string) string {
|
||
switch strings.ToLower(strings.TrimSpace(role)) {
|
||
case "user":
|
||
return "user"
|
||
case "assistant":
|
||
return "assistant"
|
||
default:
|
||
return "system"
|
||
}
|
||
}
|
||
|
||
func buildOptimisticConversationHistoryItem(
|
||
role string,
|
||
content string,
|
||
reasoningContent string,
|
||
reasoningDurationSeconds int,
|
||
retryMeta *chatRetryMeta,
|
||
createdAt time.Time,
|
||
) model.GetConversationHistoryItem {
|
||
item := model.GetConversationHistoryItem{
|
||
Role: normalizeConversationHistoryRole(role),
|
||
Content: strings.TrimSpace(content),
|
||
ReasoningContent: strings.TrimSpace(reasoningContent),
|
||
ReasoningDurationSeconds: reasoningDurationSeconds,
|
||
}
|
||
if !createdAt.IsZero() {
|
||
t := createdAt
|
||
item.CreatedAt = &t
|
||
}
|
||
if retryMeta != nil {
|
||
item.RetryGroupID = retryMeta.GroupIDPtr()
|
||
item.RetryIndex = retryMeta.IndexPtr()
|
||
item.RetryTotal = retryMeta.IndexPtr()
|
||
}
|
||
return item
|
||
}
|
||
|
||
func appendConversationHistoryItemIfMissing(
|
||
items []model.GetConversationHistoryItem,
|
||
item model.GetConversationHistoryItem,
|
||
) []model.GetConversationHistoryItem {
|
||
targetKey := conversationHistoryItemSignature(item)
|
||
for _, existed := range items {
|
||
if conversationHistoryItemSignature(existed) == targetKey {
|
||
return items
|
||
}
|
||
}
|
||
return append(items, item)
|
||
}
|
||
|
||
func conversationHistoryItemSignature(item model.GetConversationHistoryItem) string {
|
||
if item.ID > 0 {
|
||
return fmt.Sprintf("id:%d", item.ID)
|
||
}
|
||
|
||
groupID := ""
|
||
if item.RetryGroupID != nil {
|
||
groupID = strings.TrimSpace(*item.RetryGroupID)
|
||
}
|
||
retryIndex := 0
|
||
if item.RetryIndex != nil {
|
||
retryIndex = *item.RetryIndex
|
||
}
|
||
createdAt := ""
|
||
if item.CreatedAt != nil {
|
||
createdAt = item.CreatedAt.UTC().Format(time.RFC3339Nano)
|
||
}
|
||
|
||
return fmt.Sprintf(
|
||
"%s|%s|%s|%s|%d|%d|%s",
|
||
strings.TrimSpace(item.Role),
|
||
strings.TrimSpace(item.Content),
|
||
strings.TrimSpace(item.ReasoningContent),
|
||
groupID,
|
||
retryIndex,
|
||
item.ReasoningDurationSeconds,
|
||
createdAt,
|
||
)
|
||
}
|
||
|
||
func sortConversationHistoryItems(items []model.GetConversationHistoryItem) {
|
||
sort.SliceStable(items, func(i, j int) bool {
|
||
left := conversationHistoryTimestamp(items[i])
|
||
right := conversationHistoryTimestamp(items[j])
|
||
if left.Equal(right) {
|
||
return conversationHistoryItemSignature(items[i]) < conversationHistoryItemSignature(items[j])
|
||
}
|
||
return left.Before(right)
|
||
})
|
||
}
|
||
|
||
func conversationHistoryTimestamp(item model.GetConversationHistoryItem) time.Time {
|
||
if item.CreatedAt == nil {
|
||
return time.Time{}
|
||
}
|
||
return *item.CreatedAt
|
||
}
|