增加stdio 调用模式

Author: weibaohui

main.go

@@ -19,6 +19,9 @@ func main() {
 	// example.Example()
 	// mcp.RunMCPServer("kom mcp server", "0.0.1", 9096)
 
+	// SSE调用时，可以在MCP client 请求中，在header中注入认证信息
+	// kom执行时，如果注册了callback，那么可以在callback中获取到ctx，从ctx上可以拿到注入的认证信息
+	// 有了认证信息，就可以在callback中，进行权限的逻辑控制了
 	authKey := "username"
 	authRoleKey := "role"
 	var ctxFn = func(ctx context.Context, r *http.Request) context.Context {
@@ -44,6 +47,7 @@ func main() {
 		},
 		AuthKey:     authKey,
 		AuthRoleKey: authRoleKey,
+		Mode:        metadata.MCPServerModeBoth, // 同时开启STDIO 和 SSE
 	}
 	mcp.RunMCPServerWithOption(&cfg)
 }

mcp/metadata/type.go

@@ -22,7 +22,15 @@ type ServerConfig struct {
 	Metadata      map[string]string // 元数据
 	AuthKey       string            // 认证key
 	AuthRoleKey   string            // 认证key
+	Mode          MCPServerMode     // 运行模式 sse,stdio
 }
+type MCPServerMode string
+
+const (
+	MCPServerModeSSE   MCPServerMode = "sse"
+	MCPServerModeStdio MCPServerMode = "stdio"
+	MCPServerModeBoth  MCPServerMode = "both"
+)
 
 type ResourceInfo struct {
 	Group      string

mcp/server.go

@@ -78,12 +78,20 @@ func RunMCPServerWithOption(cfg *metadata.ServerConfig) {
 	storageclass.RegisterTools(s, config)
 	ingressclass.RegisterTools(s, config)
 	yaml.RegisterTools(s, config)
+
+	if cfg.Mode == metadata.MCPServerModeStdio || cfg.Mode == metadata.MCPServerModeBoth {
+		// Start the stdio server
+		if err := server.ServeStdio(s); err != nil {
+			fmt.Printf("Server error: %v\n", err)
+		}
+	}
+
 	// 创建 SSE 服务器
 	sseServer := server.NewSSEServer(s, config.SSEOption...)
-
 	// 启动服务器
 	err := sseServer.Start(fmt.Sprintf(":%d", config.Port))
 	if err != nil {
 		klog.Errorf("MCP Server error: %v\n", err)
 	}
+
 }