go-svc-mcp/main.go

package main

import (
	"log"
	"net/http"
	"os"

	"git.x2erp.com/qdy/go-svc-mcp/internal/auth"
	"git.x2erp.com/qdy/go-svc-mcp/internal/mcp"
	_ "git.x2erp.com/qdy/go-svc-mcp/internal/tools" // 触发工具自动注册

	"git.x2erp.com/qdy/go-base/config"
	"git.x2erp.com/qdy/go-base/container"
	"git.x2erp.com/qdy/go-base/ctx"
	"git.x2erp.com/qdy/go-base/graceful"
	"git.x2erp.com/qdy/go-base/logger"
	"git.x2erp.com/qdy/go-db/factory/database"
	mcpsdk "github.com/modelcontextprotocol/go-sdk/mcp"
)

var (
	appName    = "svc-mcp"
	appVersion = "1"
)

func main() {
	// 0. 初始化日志
	logBootFactory := logger.InitBootLog()

	// 1. 获取配置文件
	cfg := config.GetConfig()
	cfg.SetAppName(appName)
	cfg.SetAppVersion(appVersion)

	// 2. 创建关闭容器
	ctr := container.NewContainer(cfg)

	// 注册日志，实现自动关闭
	container.Reg(ctr, logBootFactory)

	// 3. 创建数据库工厂
	dbFactory := container.Create(ctr, database.CreateDBFactory)
	dbFactory.TestConnection()

	// 4. 创建基础请求上下文（可从配置或认证中提取）
	baseCtx := &ctx.RequestContext{
		TenantID: "default-tenant", // 实际应从认证中间件获取
	}

	// 5. 创建 MCP 服务器
	mcpServer, err := mcp.NewServer(mcp.Config{
		Name:        appName,
		Version:     appVersion,
		Description: "MCP 工具服务，提供自动注册发现和依赖注入",
		DBFactory:   dbFactory,
		BaseCtx:     baseCtx,
		Port:        cfg.GetServiceConfig().Port,
		ServiceName: cfg.GetServiceConfig().ServiceName,
	})
	if err != nil {
		log.Fatalf("Failed to create MCP server: %v", err)
	}
	log.Printf("MCP server created with tools registered")

	// 6. 获取 SDK 服务器实例
	sdkServer := mcpServer.GetSDKServer()

	// 7. 创建 HTTP 处理器（带验证中间件）
	authToken := os.Getenv("MCP_AUTH_TOKEN")
	if authToken == "" {
		authToken = "123" // 仅用于开发，生产环境必须设置
	}
	projectIDHeader := os.Getenv("MCP_PROJECT_ID_HEADER")
	if projectIDHeader == "" {
		projectIDHeader = "X-Project-ID"
	}
	mcpHandler := mcpsdk.NewStreamableHTTPHandler(func(req *http.Request) *mcpsdk.Server {
		return sdkServer
	}, nil)
	// 包装验证中间件
	handler := auth.AuthMiddleware(mcpHandler, authToken, projectIDHeader)

	mcpServer.Run(handler)

	//启用运行日志
	container.Create(ctr, logger.InitRuntimeLogger)

	//等待关闭
	graceful.WaitForShutdown(appName, mcpServer.GetHTTPServer(), ctr)
}