Version: 0.4.4.dev.260307
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 中的数据
This commit is contained in:
90
infra/smartflow-mcp-server/cmd/server/main.go
Normal file
90
infra/smartflow-mcp-server/cmd/server/main.go
Normal file
@@ -0,0 +1,90 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/audit"
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/config"
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/envutil"
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/mcp"
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/ratelimit"
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/security"
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/store"
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/tools"
|
||||
)
|
||||
|
||||
func main() {
|
||||
if err := envutil.LoadDotEnv(".env"); err != nil {
|
||||
log.Fatalf("load .env failed: %v", err)
|
||||
}
|
||||
|
||||
cfg, err := config.LoadFromEnv()
|
||||
if err != nil {
|
||||
log.Fatalf("load config failed: %v", err)
|
||||
}
|
||||
|
||||
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
||||
defer stop()
|
||||
|
||||
auditLogger, err := audit.New(cfg.AuditLogPath)
|
||||
if err != nil {
|
||||
log.Fatalf("init audit logger failed: %v", err)
|
||||
}
|
||||
defer func() {
|
||||
_ = auditLogger.Close()
|
||||
}()
|
||||
|
||||
mysqlClient, err := store.NewMySQLClient(ctx, cfg.MySQL)
|
||||
if err != nil {
|
||||
log.Fatalf("init mysql failed: %v", err)
|
||||
}
|
||||
defer func() {
|
||||
_ = mysqlClient.Close()
|
||||
}()
|
||||
|
||||
redisClient, err := store.NewRedisClient(ctx, cfg.Redis)
|
||||
if err != nil {
|
||||
log.Fatalf("init redis failed: %v", err)
|
||||
}
|
||||
defer func() {
|
||||
_ = redisClient.Close()
|
||||
}()
|
||||
|
||||
sqlValidator := security.NewSQLValidator(
|
||||
cfg.MySQL.Database,
|
||||
cfg.EnforceWhitelist,
|
||||
cfg.MySQL.AllowedDatabases,
|
||||
cfg.MySQL.AllowedTables,
|
||||
)
|
||||
|
||||
registry, err := tools.NewRegistry(
|
||||
tools.NewMySQLReadOnlyTool(mysqlClient, sqlValidator, cfg.MaxResultRows),
|
||||
tools.NewRedisGetTool(redisClient, cfg.RedisValueMaxItems, cfg.RedisMaxStringBytes),
|
||||
tools.NewRedisScanTool(redisClient, cfg.RedisScanMaxKeys, cfg.RedisScanMaxCount),
|
||||
)
|
||||
if err != nil {
|
||||
log.Fatalf("init tool registry failed: %v", err)
|
||||
}
|
||||
|
||||
limiter := ratelimit.New(cfg.RateLimitRPS, cfg.RateLimitBurst)
|
||||
server := mcp.NewServer(
|
||||
os.Stdin,
|
||||
os.Stdout,
|
||||
registry,
|
||||
auditLogger,
|
||||
limiter,
|
||||
cfg.ServerName,
|
||||
cfg.ServerVersion,
|
||||
cfg.ProtocolVersion,
|
||||
cfg.DefaultCaller,
|
||||
cfg.ToolTimeout,
|
||||
)
|
||||
|
||||
if err := server.Serve(ctx); err != nil {
|
||||
log.Fatalf("mcp server exited with error: %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user