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 中的数据
131 lines
3.2 KiB
Go
131 lines
3.2 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/store"
|
|
)
|
|
|
|
type RedisGetTool struct {
|
|
client *store.RedisClient
|
|
valueMaxItems int
|
|
maxStringBytes int
|
|
}
|
|
|
|
func NewRedisGetTool(client *store.RedisClient, valueMaxItems int, maxStringBytes int) *RedisGetTool {
|
|
return &RedisGetTool{client: client, valueMaxItems: valueMaxItems, maxStringBytes: maxStringBytes}
|
|
}
|
|
|
|
func (t *RedisGetTool) Name() string {
|
|
return "redis_get"
|
|
}
|
|
|
|
func (t *RedisGetTool) Description() string {
|
|
return "Get a Redis key by name and return its type and value."
|
|
}
|
|
|
|
func (t *RedisGetTool) InputSchema() map[string]any {
|
|
return map[string]any{
|
|
"type": "object",
|
|
"properties": map[string]any{
|
|
"key": map[string]any{
|
|
"type": "string",
|
|
"description": "Redis key",
|
|
},
|
|
},
|
|
"required": []string{"key"},
|
|
"additionalProperties": false,
|
|
}
|
|
}
|
|
|
|
func (t *RedisGetTool) Execute(ctx context.Context, args map[string]any) (map[string]any, error) {
|
|
key, ok := args["key"].(string)
|
|
if !ok || key == "" {
|
|
return nil, fmt.Errorf("key must be a non-empty string")
|
|
}
|
|
res, err := t.client.GetWithType(ctx, key, t.valueMaxItems, t.maxStringBytes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return map[string]any{
|
|
"exists": res.Exists,
|
|
"key": res.Key,
|
|
"type": res.Type,
|
|
"value": res.Value,
|
|
"truncated": res.Truncated,
|
|
"durationMs": res.DurationMs,
|
|
}, nil
|
|
}
|
|
|
|
type RedisScanTool struct {
|
|
client *store.RedisClient
|
|
maxKeys int
|
|
maxScanCount int
|
|
}
|
|
|
|
func NewRedisScanTool(client *store.RedisClient, maxKeys int, maxScanCount int) *RedisScanTool {
|
|
return &RedisScanTool{client: client, maxKeys: maxKeys, maxScanCount: maxScanCount}
|
|
}
|
|
|
|
func (t *RedisScanTool) Name() string {
|
|
return "redis_scan"
|
|
}
|
|
|
|
func (t *RedisScanTool) Description() string {
|
|
return "Scan Redis keys by pattern with capped result size."
|
|
}
|
|
|
|
func (t *RedisScanTool) InputSchema() map[string]any {
|
|
return map[string]any{
|
|
"type": "object",
|
|
"properties": map[string]any{
|
|
"pattern": map[string]any{
|
|
"type": "string",
|
|
"description": "Pattern, for example user:*",
|
|
},
|
|
"count": map[string]any{
|
|
"type": "number",
|
|
"description": "Optional scan count hint",
|
|
},
|
|
},
|
|
"required": []string{"pattern"},
|
|
"additionalProperties": false,
|
|
}
|
|
}
|
|
|
|
func (t *RedisScanTool) Execute(ctx context.Context, args map[string]any) (map[string]any, error) {
|
|
pattern, ok := args["pattern"].(string)
|
|
if !ok || pattern == "" {
|
|
return nil, fmt.Errorf("pattern must be a non-empty string")
|
|
}
|
|
|
|
count := int64(20)
|
|
if rawCount, ok := args["count"]; ok {
|
|
number, ok := rawCount.(float64)
|
|
if !ok {
|
|
return nil, fmt.Errorf("count must be a number")
|
|
}
|
|
if number <= 0 {
|
|
return nil, fmt.Errorf("count must be > 0")
|
|
}
|
|
count = int64(number)
|
|
}
|
|
if count > int64(t.maxScanCount) {
|
|
count = int64(t.maxScanCount)
|
|
}
|
|
|
|
res, err := t.client.ScanKeys(ctx, pattern, count, t.maxKeys)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return map[string]any{
|
|
"pattern": res.Pattern,
|
|
"keys": res.Keys,
|
|
"returned": res.Returned,
|
|
"nextCursor": res.NextCursor,
|
|
"truncated": res.Truncated,
|
|
"durationMs": res.DurationMs,
|
|
}, nil
|
|
}
|