hhs
6487a8ecab
- Manager.Create 签名新增 userID 参数
- 新增 ConversationSummary 类型和 ListByUser 分页查询
- 新增 UpdateTitle 方法
- MemoryManager 实现：ListByUser 遍历+过滤+排序，UpdateTitle，自动标题生成
- RedisManager 实现：user:{id}:sessions 索引，ListByUser 通过 SMEMBERS 查询
- AppendMessage 自动更新标题（首条 user 消息时，取前 20 字符）
- 更新 ws handler、api/session.go、orchestrator mock 的 Create 调用
366 lines
8.5 KiB
Go
366 lines
8.5 KiB
Go
package session
|
||
|
||
import (
|
||
"context"
|
||
"sort"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/google/uuid"
|
||
|
||
"github.com/hhs/camtalk/internal/logger"
|
||
"github.com/hhs/camtalk/internal/models"
|
||
)
|
||
|
||
const (
|
||
defaultTTL = 30 * time.Minute
|
||
defaultHistorySize = 20
|
||
)
|
||
|
||
// sessionEntry 内部会话条目。
|
||
type sessionEntry struct {
|
||
session models.Session
|
||
history []models.Message
|
||
activeReqID string
|
||
lastActive time.Time
|
||
}
|
||
|
||
// MemoryManager 基于内存的 SessionManager 实现。
|
||
// 适用于 MVP 和无 Redis 的开发环境。
|
||
type MemoryManager struct {
|
||
mu sync.RWMutex
|
||
sessions map[string]*sessionEntry
|
||
ttl time.Duration
|
||
maxHistory int
|
||
stopCleaner chan struct{}
|
||
}
|
||
|
||
// NewMemoryManager 创建内存版 SessionManager。
|
||
// ttl 为会话过期时间，maxHistory 为对话历史上限（0 表示使用默认值 20）。
|
||
func NewMemoryManager(ttl time.Duration, maxHistory int) *MemoryManager {
|
||
if ttl <= 0 {
|
||
ttl = defaultTTL
|
||
}
|
||
if maxHistory <= 0 {
|
||
maxHistory = defaultHistorySize
|
||
}
|
||
|
||
m := &MemoryManager{
|
||
sessions: make(map[string]*sessionEntry),
|
||
ttl: ttl,
|
||
maxHistory: maxHistory,
|
||
stopCleaner: make(chan struct{}),
|
||
}
|
||
|
||
// 启动后台清理 goroutine，每分钟清除过期会话。
|
||
go m.cleanLoop()
|
||
|
||
return m
|
||
}
|
||
|
||
// cleanLoop 后台定期清理过期会话。
|
||
func (m *MemoryManager) cleanLoop() {
|
||
ticker := time.NewTicker(1 * time.Minute)
|
||
defer ticker.Stop()
|
||
for {
|
||
select {
|
||
case <-ticker.C:
|
||
m.cleanExpired()
|
||
case <-m.stopCleaner:
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
// cleanExpired 清除所有过期会话。
|
||
func (m *MemoryManager) cleanExpired() {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
now := time.Now()
|
||
for id, entry := range m.sessions {
|
||
if now.Sub(entry.lastActive) > m.ttl {
|
||
delete(m.sessions, id)
|
||
logger.Log.Debugw("session expired (cleaner)", "session", id)
|
||
}
|
||
}
|
||
}
|
||
|
||
// Stop 停止后台清理 goroutine。应用退出前调用。
|
||
func (m *MemoryManager) Stop() {
|
||
close(m.stopCleaner)
|
||
}
|
||
|
||
// isExpired 检查会话是否过期（调用方需持锁或在已知 entry 存在时调用）。
|
||
func (m *MemoryManager) isExpired(entry *sessionEntry) bool {
|
||
return time.Since(entry.lastActive) > m.ttl
|
||
}
|
||
|
||
// Create 创建新会话。userID 为空表示匿名会话。
|
||
func (m *MemoryManager) Create(_ context.Context, userID string, config models.SessionConfig) (string, error) {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
id := uuid.New().String()
|
||
now := time.Now()
|
||
m.sessions[id] = &sessionEntry{
|
||
session: models.Session{
|
||
ID: id,
|
||
UserID: userID,
|
||
Title: models.DefaultSessionTitle,
|
||
CreatedAt: now,
|
||
UpdatedAt: now,
|
||
Config: config,
|
||
},
|
||
history: make([]models.Message, 0),
|
||
lastActive: now,
|
||
}
|
||
|
||
logger.Log.Debugw("session created", "session", id, "user_id", userID)
|
||
return id, nil
|
||
}
|
||
|
||
// Get 获取会话。
|
||
func (m *MemoryManager) Get(_ context.Context, sessionID string) (*models.Session, error) {
|
||
m.mu.RLock()
|
||
defer m.mu.RUnlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return nil, ErrSessionNotFound
|
||
}
|
||
|
||
sess := entry.session // 复制一份返回
|
||
return &sess, nil
|
||
}
|
||
|
||
// UpdateConfig 更新会话配置。
|
||
func (m *MemoryManager) UpdateConfig(_ context.Context, sessionID string, patch models.SessionConfigPatch) error {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return ErrSessionNotFound
|
||
}
|
||
|
||
patch.Apply(&entry.session.Config)
|
||
entry.lastActive = time.Now()
|
||
|
||
logger.Log.Debugw("session config updated", "session", sessionID)
|
||
return nil
|
||
}
|
||
|
||
// UpdateTitle 更新会话标题。
|
||
func (m *MemoryManager) UpdateTitle(_ context.Context, sessionID string, title string) error {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return ErrSessionNotFound
|
||
}
|
||
|
||
entry.session.Title = title
|
||
entry.session.UpdatedAt = time.Now()
|
||
entry.lastActive = time.Now()
|
||
|
||
logger.Log.Debugw("session title updated", "session", sessionID, "title", title)
|
||
return nil
|
||
}
|
||
|
||
// ListByUser 获取用户的对话列表（分页，按 UpdatedAt 降序）。
|
||
func (m *MemoryManager) ListByUser(_ context.Context, userID string, page, size int) ([]ConversationSummary, int, error) {
|
||
m.mu.RLock()
|
||
defer m.mu.RUnlock()
|
||
|
||
// 收集该用户的所有 session
|
||
var list []ConversationSummary
|
||
for _, entry := range m.sessions {
|
||
if entry.session.UserID != userID {
|
||
continue
|
||
}
|
||
if m.isExpired(entry) {
|
||
continue
|
||
}
|
||
summary := ConversationSummary{
|
||
ID: entry.session.ID,
|
||
Title: entry.session.Title,
|
||
MessageCount: len(entry.history),
|
||
UpdatedAt: entry.lastActive,
|
||
}
|
||
if len(entry.history) > 0 {
|
||
summary.LastMessage = entry.history[len(entry.history)-1].Content
|
||
}
|
||
list = append(list, summary)
|
||
}
|
||
|
||
// 按 UpdatedAt 降序排序
|
||
sort.Slice(list, func(i, j int) bool {
|
||
return list[i].UpdatedAt.After(list[j].UpdatedAt)
|
||
})
|
||
|
||
total := len(list)
|
||
|
||
// 分页
|
||
if page <= 0 {
|
||
page = 1
|
||
}
|
||
if size <= 0 {
|
||
size = 20
|
||
}
|
||
start := (page - 1) * size
|
||
if start >= total {
|
||
return []ConversationSummary{}, total, nil
|
||
}
|
||
end := start + size
|
||
if end > total {
|
||
end = total
|
||
}
|
||
|
||
return list[start:end], total, nil
|
||
}
|
||
|
||
// GetHistory 获取最近 N 轮对话历史。
|
||
func (m *MemoryManager) GetHistory(_ context.Context, sessionID string, limit int) ([]models.Message, error) {
|
||
m.mu.RLock()
|
||
defer m.mu.RUnlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return nil, ErrSessionNotFound
|
||
}
|
||
|
||
if limit <= 0 || limit > len(entry.history) {
|
||
limit = len(entry.history)
|
||
}
|
||
|
||
// 返回最近 limit 条的副本
|
||
result := make([]models.Message, limit)
|
||
copy(result, entry.history[len(entry.history)-limit:])
|
||
return result, nil
|
||
}
|
||
|
||
// AppendMessage 追加一条对话消息，同时刷新 TTL。
|
||
func (m *MemoryManager) AppendMessage(_ context.Context, sessionID string, msg models.Message) error {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return ErrSessionNotFound
|
||
}
|
||
|
||
entry.history = append(entry.history, msg)
|
||
|
||
// 自动更新标题：首条 user 消息时，如果标题为默认值，自动更新为消息前 20 字符
|
||
if msg.Role == "user" && entry.session.Title == models.DefaultSessionTitle {
|
||
entry.session.Title = generateTitle(msg.Content)
|
||
}
|
||
|
||
// 超过上限时裁剪，保留最新的 maxHistory 条
|
||
if len(entry.history) > m.maxHistory {
|
||
entry.history = entry.history[len(entry.history)-m.maxHistory:]
|
||
}
|
||
|
||
now := time.Now()
|
||
entry.lastActive = now
|
||
entry.session.UpdatedAt = now
|
||
return nil
|
||
}
|
||
|
||
// generateTitle 从首条消息生成对话标题（取前 20 个字符）。
|
||
func generateTitle(firstMessage string) string {
|
||
runes := []rune(firstMessage)
|
||
if len(runes) > 20 {
|
||
return string(runes[:20]) + "…"
|
||
}
|
||
return firstMessage
|
||
}
|
||
|
||
// SetActiveRequest 标记当前正在处理的请求 ID。
|
||
func (m *MemoryManager) SetActiveRequest(_ context.Context, sessionID string, requestID string) error {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return ErrSessionNotFound
|
||
}
|
||
|
||
entry.activeReqID = requestID
|
||
entry.lastActive = time.Now()
|
||
return nil
|
||
}
|
||
|
||
// GetActiveRequestID 获取当前活跃请求 ID。
|
||
func (m *MemoryManager) GetActiveRequestID(_ context.Context, sessionID string) (string, error) {
|
||
m.mu.RLock()
|
||
defer m.mu.RUnlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return "", ErrSessionNotFound
|
||
}
|
||
|
||
return entry.activeReqID, nil
|
||
}
|
||
|
||
// ClearActiveRequest 清除活跃请求标记。
|
||
func (m *MemoryManager) ClearActiveRequest(_ context.Context, sessionID string) error {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return ErrSessionNotFound
|
||
}
|
||
|
||
entry.activeReqID = ""
|
||
entry.lastActive = time.Now()
|
||
return nil
|
||
}
|
||
|
||
// Touch 刷新 TTL。
|
||
func (m *MemoryManager) Touch(_ context.Context, sessionID string) error {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
entry, ok := m.sessions[sessionID]
|
||
if !ok || m.isExpired(entry) {
|
||
return ErrSessionNotFound
|
||
}
|
||
|
||
entry.lastActive = time.Now()
|
||
return nil
|
||
}
|
||
|
||
// Destroy 显式销毁会话。
|
||
func (m *MemoryManager) Destroy(_ context.Context, sessionID string) error {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
|
||
if _, ok := m.sessions[sessionID]; !ok {
|
||
return ErrSessionNotFound
|
||
}
|
||
|
||
delete(m.sessions, sessionID)
|
||
logger.Log.Debugw("session destroyed", "session", sessionID)
|
||
return nil
|
||
}
|
||
|
||
// ActiveCount 返回当前活跃会话数。
|
||
func (m *MemoryManager) ActiveCount() int {
|
||
m.mu.RLock()
|
||
defer m.mu.RUnlock()
|
||
|
||
now := time.Now()
|
||
count := 0
|
||
for _, entry := range m.sessions {
|
||
if now.Sub(entry.lastActive) <= m.ttl {
|
||
count++
|
||
}
|
||
}
|
||
return count
|
||
}
|