go-svc-mcp/main.go

package main

import (
	"log"
	"net/http"

	"git.x2erp.com/qdy/go-svc-mcp/internal/auth"
	"git.x2erp.com/qdy/go-svc-mcp/internal/handler"
	"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-base/webx"
	"git.x2erp.com/qdy/go-base/webx/router"

	"git.x2erp.com/qdy/go-db/factory/database"
	"git.x2erp.com/qdy/go-db/sqldef"
	mcpsdk "github.com/modelcontextprotocol/go-sdk/mcp"

	_ "git.x2erp.com/qdy/go-svc-mcp/internal/tables" // 导入表定义包，触发 init() 函数
)

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()
	// 创建表
	creteTabel(dbFactory)

	// 创建mongodb
	//mongoDBFactory := container.Create(ctr, mongodb.CreateFactory)
	//mongoDBFactory.TestConnection()

	// 创建mcpservice
	mcpServer := createMcpService(cfg, dbFactory)

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

	//建立httpservice
	webxService := createMHttpService(cfg, dbFactory)

	//等待关闭
	graceful.WaitForShutdown(cfg.GetServiceConfig().ServiceName, ctr, mcpServer.GetHTTPServer(), webxService.GetServer())
}

// 创建httpService
func createMHttpService(cfg config.IConfig, dbFactory *database.DBFactory) *webx.WebService {
	//得到webservice服务工厂
	webxFactory := webx.GetWebServiceFactory()

	//建立hhtpService服务
	webServcie, _ := webxFactory.CreateService(cfg.GetServiceConfig())

	//建立路由-api
	routerService := router.NewWebService(webServcie.GetRouter())
	//注册路由--api
	handler.RegisterRouter(routerService, dbFactory)

	//启动服务
	webServcie.Run()
	return webServcie
}

// createMcpService 创建mcpService
func createMcpService(cfg config.IConfig, dbFactory *database.DBFactory) *mcp.Server {

	// 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.GetMcpServiceConfig().Port,
		ServiceName: cfg.GetMcpServiceConfig().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()

	mcpHandler := mcpsdk.NewStreamableHTTPHandler(func(req *http.Request) *mcpsdk.Server {
		return sdkServer
	}, nil)
	// 包装验证中间件
	handler := auth.AuthMiddleware(mcpHandler, dbFactory)

	mcpServer.Run(handler)

	return mcpServer
}

func creteTabel(factory *database.DBFactory) {

	// 获取数据库连接和类型
	db := factory.GetDB()
	dbType := factory.GetDBType()

	// 创建表同步器
	syncer, err := sqldef.NewTableSyncer(db, dbType)
	if err != nil {
		log.Printf("创建 - 建立器失败: %v", err)
	}

	// 创建表
	if err := syncer.CreateTables(); err != nil {
		log.Printf("建表失败: %v", err)
	}

	log.Println("数据库表建立完成！")
}