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 中的数据
96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/config"
|
|
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/security"
|
|
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/store"
|
|
)
|
|
|
|
func TestIntegrationMySQLReadOnlyTool(t *testing.T) {
|
|
if os.Getenv("MCP_IT_RUN") != "1" {
|
|
t.Skip("set MCP_IT_RUN=1 to run integration tests")
|
|
}
|
|
|
|
port := 3306
|
|
if p := os.Getenv("MYSQL_PORT"); p != "" {
|
|
if n, err := strconv.Atoi(p); err == nil {
|
|
port = n
|
|
}
|
|
}
|
|
|
|
mysqlCfg := config.MySQLConfig{
|
|
Host: os.Getenv("MYSQL_HOST"),
|
|
Port: port,
|
|
User: os.Getenv("MYSQL_USER"),
|
|
Password: os.Getenv("MYSQL_PASSWORD"),
|
|
Database: os.Getenv("MYSQL_DATABASE"),
|
|
Params: "charset=utf8mb4&parseTime=true&loc=Local",
|
|
}
|
|
if mysqlCfg.Host == "" || mysqlCfg.User == "" || mysqlCfg.Database == "" {
|
|
t.Skip("missing MYSQL_* env for integration test")
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
client, err := store.NewMySQLClient(ctx, mysqlCfg)
|
|
if err != nil {
|
|
t.Fatalf("mysql not available: %v", err)
|
|
}
|
|
defer func() { _ = client.Close() }()
|
|
|
|
validator := security.NewSQLValidator(mysqlCfg.Database, false, nil, nil)
|
|
tool := NewMySQLReadOnlyTool(client, validator, 10)
|
|
res, err := tool.Execute(ctx, map[string]any{"sql": "SELECT 1 AS ok"})
|
|
if err != nil {
|
|
t.Fatalf("tool execute failed: %v", err)
|
|
}
|
|
if res["rowCount"].(int) < 1 {
|
|
t.Fatalf("expected rowCount >= 1")
|
|
}
|
|
}
|
|
|
|
func TestIntegrationRedisTools(t *testing.T) {
|
|
if os.Getenv("MCP_IT_RUN") != "1" {
|
|
t.Skip("set MCP_IT_RUN=1 to run integration tests")
|
|
}
|
|
|
|
db := 0
|
|
if p := os.Getenv("REDIS_DB"); p != "" {
|
|
if n, err := strconv.Atoi(p); err == nil {
|
|
db = n
|
|
}
|
|
}
|
|
redisCfg := config.RedisConfig{
|
|
Addr: os.Getenv("REDIS_ADDR"),
|
|
Password: os.Getenv("REDIS_PASSWORD"),
|
|
DB: db,
|
|
}
|
|
if redisCfg.Addr == "" {
|
|
t.Skip("REDIS_ADDR is empty")
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
client, err := store.NewRedisClient(ctx, redisCfg)
|
|
if err != nil {
|
|
t.Fatalf("redis not available: %v", err)
|
|
}
|
|
defer func() { _ = client.Close() }()
|
|
|
|
getTool := NewRedisGetTool(client, 10, 128)
|
|
if _, err := getTool.Execute(ctx, map[string]any{"key": "__integration_missing_key__"}); err != nil {
|
|
t.Fatalf("redis_get failed: %v", err)
|
|
}
|
|
|
|
scanTool := NewRedisScanTool(client, 10, 10)
|
|
if _, err := scanTool.Execute(ctx, map[string]any{"pattern": "*", "count": float64(5)}); err != nil {
|
|
t.Fatalf("redis_scan failed: %v", err)
|
|
}
|
|
}
|