Blizzard
a2d184b7ec
新增 sundynix-shared/otelx：otelx.Init(ctx,服务名) 注册 W3C 传播器 + OTLP/HTTP 批量导出（默认 localhost:4318，docker 里的 Jaeger）；OTEL_SDK_DISABLED=true 关导出。 Jaeger 不在线 / 导出器构建失败都不阻断启动（可观测性是增益而非依赖）。 - NATS 跨进程传播（无现成中间件）：bus/trace.go 的 natsHeaderCarrier + inject/extract， PublishTask/CallTool 注入 traceparent，ConsumeTasks/ServeTool 抽出续上 → 链路跨总线连成一棵树。 - 埋点：gateway 挂 otelgin（HTTP server span，链路根）；dispatcher task.execute → node.<kind>（每节点，nctx 下传使工具/LLM 挂到节点下）→ llm.stream/llm.generate； bus 自动出 tool.call(client)↔tool.serve(server) 成对跨服务 span。 - docker-compose 加 jaeger all-in-one（UI :16686，OTLP :4318）。 - 依赖修复：otlptracehttp 触发 genproto 单体(旧)vs 拆分模块 ambiguous import（milvus 拉旧版）， pin genproto 至后拆分版（go.work 工作区全局生效）。 - production_readiness.md 1.1 更新为「已实现」。 验证：真实 input→retriever→agent 任务在 Jaeger 出 14 span / 3 服务的完整树， 跨 NATS（publish→consume）、跨服务（tool.call→tool.serve）均连通， 瓶颈 kb_search 692ms、llm 1597ms 一眼可见；四模块 build+vet+test 全绿。 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
106 lines
4.0 KiB
Go
106 lines
4.0 KiB
Go
// Command server 启动 sundynix-mcp-go —— 第 5 层 Go I/O 型 MCP 工具微服务。
|
||
package main
|
||
|
||
import (
|
||
"context"
|
||
"log"
|
||
"os"
|
||
"os/signal"
|
||
"syscall"
|
||
|
||
sharedbus "github.com/sundynix/sundynix-shared/bus"
|
||
"github.com/sundynix/sundynix-shared/contract"
|
||
|
||
"github.com/sundynix/sundynix-mcp-go/internal/history"
|
||
"github.com/sundynix/sundynix-mcp-go/internal/mcp"
|
||
"github.com/sundynix/sundynix-mcp-go/internal/memory"
|
||
"github.com/sundynix/sundynix-mcp-go/internal/rag"
|
||
"github.com/sundynix/sundynix-mcp-go/internal/search"
|
||
|
||
"github.com/sundynix/sundynix-shared/otelx"
|
||
"github.com/sundynix/sundynix-shared/secrets"
|
||
)
|
||
|
||
func main() {
|
||
secrets.MustHaveKeyInProd() // 生产须配 SUNDYNIX_SECRET_KEY 以解密下发的 api_key 密文
|
||
|
||
// 链路追踪：工具服务端 span 续上 dispatcher 的 trace（tool.call → tool.serve）。
|
||
shutdownTrace, _ := otelx.Init(context.Background(), "sundynix-mcp-go")
|
||
defer func() { _ = shutdownTrace(context.Background()) }()
|
||
|
||
natsURL := envOr("NATS_URL", "nats://localhost:4222")
|
||
pgDSN := envOr("POSTGRES_DSN", "postgres://sundynix:sundynix@localhost:5432/sundynix?sslmode=disable")
|
||
redisAddr := envOr("REDIS_ADDR", "localhost:6379")
|
||
milvusAddr := envOr("MILVUS_ADDR", "localhost:19530")
|
||
embBase := envOr("EMBED_BASE_URL", "") // OpenAI 兼容 embeddings 端点（空=向量检索降级）
|
||
embKey := envOr("EMBED_API_KEY", "")
|
||
embModel := envOr("EMBED_MODEL", "")
|
||
rerankBase := envOr("RERANK_BASE_URL", "") // DashScope 文本重排端点（空=不启用 rerank）
|
||
rerankKey := envOr("RERANK_API_KEY", "")
|
||
rerankModel := envOr("RERANK_MODEL", "")
|
||
neo4jURI := envOr("NEO4J_URI", "neo4j://localhost:7687") // GraphRAG 图谱（连不上则降级）
|
||
neo4jUser := envOr("NEO4J_USER", "neo4j")
|
||
neo4jPass := envOr("NEO4J_PASS", "sundynix")
|
||
|
||
b, err := sharedbus.Connect(natsURL)
|
||
if err != nil {
|
||
log.Fatalf("[mcp_go] nats connect: %v", err)
|
||
}
|
||
defer b.Close()
|
||
log.Printf("[mcp_go] connected %s", natsURL)
|
||
|
||
engine := search.NewHybrid() // LLM Wiki 混合检索：Bleve + Milvus + Neo4j
|
||
mem := memory.Open(pgDSN) // 偏好记忆：sundynix_user_profile（连不上则降级）
|
||
defer mem.Close()
|
||
hist := history.Open(redisAddr) // 会话短期历史：Redis（连不上则降级）
|
||
defer hist.Close()
|
||
|
||
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
||
defer stop()
|
||
|
||
// RAG 核心链：embedding + Milvus(向量) + Bleve(全文) + Neo4j(图谱) + 可选 rerank
|
||
ragEngine := rag.Open(ctx, rag.Config{
|
||
MilvusAddr: milvusAddr,
|
||
EmbedBase: embBase, EmbedKey: embKey, EmbedModel: embModel,
|
||
RerankBase: rerankBase, RerankKey: rerankKey, RerankModel: rerankModel,
|
||
Neo4jURI: neo4jURI, Neo4jUser: neo4jUser, Neo4jPass: neo4jPass,
|
||
})
|
||
defer ragEngine.Close()
|
||
|
||
// 配置控制面：取激活 embedding(向量) + chat(图谱抽取) 配置并订阅热更新。
|
||
applyEmbed := func(cfg *contract.ModelConfig) {
|
||
if cfg != nil {
|
||
ragEngine.SetEmbedding(cfg.BaseURL, cfg.APIKey, cfg.Model)
|
||
}
|
||
}
|
||
applyChat := func(cfg *contract.ModelConfig) {
|
||
if cfg != nil {
|
||
ragEngine.SetChat(cfg.BaseURL, cfg.APIKey, cfg.Model)
|
||
}
|
||
}
|
||
// 先订阅热更新（控制台改配置即生效）。
|
||
if _, err := b.SubscribeConfigUpdated(contract.ConfigKindChat, applyChat); err != nil {
|
||
log.Printf("[mcp_go] subscribe chat config: %v", err)
|
||
}
|
||
if _, err := b.SubscribeConfigUpdated(contract.ConfigKindEmbedding, applyEmbed); err != nil {
|
||
log.Printf("[mcp_go] subscribe embedding config: %v", err)
|
||
}
|
||
// 后台重试拉初始配置：容忍 gateway 晚于本服务启动（避免一次性扑空致 RAG 长期降级）。
|
||
go b.RequestConfigWithRetry(ctx, contract.ConfigKindEmbedding, applyEmbed)
|
||
go b.RequestConfigWithRetry(ctx, contract.ConfigKindChat, applyChat)
|
||
|
||
gw := mcp.NewGateway(b, engine, mem, hist, ragEngine)
|
||
|
||
log.Println("[mcp_go] serving MCP over sundynix.tools.go.* (Ctrl-C to quit)")
|
||
if err := gw.Serve(ctx); err != nil && err != context.Canceled {
|
||
log.Fatalf("[mcp_go] exit: %v", err)
|
||
}
|
||
}
|
||
|
||
func envOr(key, def string) string {
|
||
if v := os.Getenv(key); v != "" {
|
||
return v
|
||
}
|
||
return def
|
||
}
|