26c350f378
feat: 🚀 增强会话管理与缓存机制 * 会话 ID 空值兜底，若 `conversation_id` 为空时自动生成 UUID * 在响应头写入 `X-Conversation-ID`，供前端使用，保持同一会话状态 perf: ⚡ 会话状态缓存优化 * 当缓存未命中但 DB 已确认/创建会话后，调用 `SetConversationStatus` 回写 Redis * 缓存写回失败时记录日志，不中断聊天主流程，确保业务流畅性 fix: 🐛 修复历史消息顺序问题与编译错误 * 修复历史消息顺序问题，保证返回的 N 条历史消息按时间正序喂给模型 * 通过反转 `created_at desc` 查询结果的切片，确保模型输入顺序正确 * 修复 `fmt.Errorf` 参数不匹配问题，修正编译错误 * 整理 `agent-cache.go` 为标准 UTF-8 编码，避免 Go 编译报错 `invalid UTF-8 encoding` feat: 🛠️ 独立构建 MCP 服务器 * 使用 `Codex` 构建独立于后端的 MCP 服务器，简化与 Codex 的协作 * 通过该服务器方便 Codex 直接测试和查看 Redis 与 MySQL 中的数据
61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package audit
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type Logger struct {
|
|
mu sync.Mutex
|
|
file *os.File
|
|
}
|
|
|
|
type Record struct {
|
|
Timestamp time.Time `json:"timestamp"`
|
|
Tool string `json:"tool"`
|
|
Caller string `json:"caller"`
|
|
Success bool `json:"success"`
|
|
DurationMs int64 `json:"duration_ms"`
|
|
Meta map[string]any `json:"meta,omitempty"`
|
|
Error string `json:"error,omitempty"`
|
|
}
|
|
|
|
func New(path string) (*Logger, error) {
|
|
dir := filepath.Dir(path)
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
return nil, fmt.Errorf("create audit dir: %w", err)
|
|
}
|
|
f, err := os.OpenFile(path, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0o644)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("open audit log file: %w", err)
|
|
}
|
|
return &Logger{file: f}, nil
|
|
}
|
|
|
|
func (l *Logger) Close() error {
|
|
if l == nil || l.file == nil {
|
|
return nil
|
|
}
|
|
return l.file.Close()
|
|
}
|
|
|
|
func (l *Logger) Log(record Record) {
|
|
if l == nil || l.file == nil {
|
|
return
|
|
}
|
|
if record.Timestamp.IsZero() {
|
|
record.Timestamp = time.Now()
|
|
}
|
|
body, err := json.Marshal(record)
|
|
if err != nil {
|
|
return
|
|
}
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
_, _ = l.file.Write(append(body, '\n'))
|
|
}
|