精简项目:只保留 docker_logs + health + info 三个工具
- 移除 database/middleware 等复杂模块,先聚焦核心功能 - docker_logs:封装 docker logs --tail N <container> - 修复 go.mod 版本为 1.24 匹配 Docker 构建镜像 - Dockerfile 改用 alpine + docker-cli,避免 scratch 无 shell 问题 - docker-compose.yml 简化为单服务定义 Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -1,17 +0,0 @@
|
||||
package server
|
||||
|
||||
// Config 是 server 包所需的配置子集,避免循环依赖。
|
||||
// 实际值由 main.go 从 config.Config 映射过来。
|
||||
type Config struct {
|
||||
Server struct {
|
||||
Transport string
|
||||
Addr string
|
||||
}
|
||||
Docker struct {
|
||||
Enabled bool
|
||||
}
|
||||
Databases map[string]struct {
|
||||
Driver string
|
||||
DSN string
|
||||
}
|
||||
}
|
||||
@@ -19,18 +19,16 @@ func NewRegistry() *Registry {
|
||||
return &Registry{tools: make(map[string]tool.Tool)}
|
||||
}
|
||||
|
||||
// Register 注册一个工具。如果工具名重复则报错。
|
||||
func (r *Registry) Register(t tool.Tool) error {
|
||||
name := t.Name()
|
||||
if _, exists := r.tools[name]; exists {
|
||||
return fmt.Errorf("tool %q already registered", name)
|
||||
}
|
||||
r.tools[name] = t
|
||||
slog.Info("tool registered", "name", name, "desc", t.Description())
|
||||
slog.Info("tool registered", "name", name)
|
||||
return nil
|
||||
}
|
||||
|
||||
// InitializeAll 调用所有工具的 Initialize,任一失败则终止。
|
||||
func (r *Registry) InitializeAll(ctx context.Context) error {
|
||||
for name, t := range r.tools {
|
||||
slog.Info("initializing tool", "name", name)
|
||||
@@ -41,18 +39,15 @@ func (r *Registry) InitializeAll(ctx context.Context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// RegisterAll 将所有工具注册到 MCP Server。
|
||||
func (r *Registry) RegisterAll(mcpServer *server.MCPServer) error {
|
||||
for name, t := range r.tools {
|
||||
if err := t.Register(mcpServer); err != nil {
|
||||
return fmt.Errorf("register tool %q: %w", name, err)
|
||||
}
|
||||
slog.Info("tool handlers registered", "name", name)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ShutdownAll 调用所有工具的 Shutdown,收集所有错误。
|
||||
func (r *Registry) ShutdownAll(ctx context.Context) []error {
|
||||
var errs []error
|
||||
for name, t := range r.tools {
|
||||
@@ -64,7 +59,6 @@ func (r *Registry) ShutdownAll(ctx context.Context) []error {
|
||||
return errs
|
||||
}
|
||||
|
||||
// HealthCheckAll 检查所有工具的连通性,用于 system health 工具。
|
||||
func (r *Registry) HealthCheckAll(ctx context.Context) map[string]error {
|
||||
result := make(map[string]error, len(r.tools))
|
||||
for name, t := range r.tools {
|
||||
@@ -72,12 +66,3 @@ func (r *Registry) HealthCheckAll(ctx context.Context) map[string]error {
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// List 返回所有已注册工具的名称。
|
||||
func (r *Registry) List() []string {
|
||||
names := make([]string, 0, len(r.tools))
|
||||
for name := range r.tools {
|
||||
names = append(names, name)
|
||||
}
|
||||
return names
|
||||
}
|
||||
|
||||
+10
-53
@@ -4,7 +4,6 @@ import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
@@ -16,12 +15,11 @@ import (
|
||||
|
||||
const version = "0.1.0"
|
||||
|
||||
// Run 启动 MCP Server,处理信号优雅关闭。
|
||||
func Run(cfg *Config, tools ...tool.Tool) error {
|
||||
// Run 启动 MCP Server。
|
||||
func Run(transport string, tools ...tool.Tool) error {
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
// 监听终止信号
|
||||
sigCh := make(chan os.Signal, 1)
|
||||
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
|
||||
go func() {
|
||||
@@ -30,7 +28,6 @@ func Run(cfg *Config, tools ...tool.Tool) error {
|
||||
cancel()
|
||||
}()
|
||||
|
||||
// 创建注册中心
|
||||
registry := NewRegistry()
|
||||
for _, t := range tools {
|
||||
if err := registry.Register(t); err != nil {
|
||||
@@ -38,13 +35,11 @@ func Run(cfg *Config, tools ...tool.Tool) error {
|
||||
}
|
||||
}
|
||||
|
||||
// 系统工具总是最后注册,确保它能看到所有工具
|
||||
sysTool := tool.NewSystemTool(version, registry)
|
||||
if err := registry.Register(sysTool); err != nil {
|
||||
return fmt.Errorf("register system tool: %w", err)
|
||||
return fmt.Errorf("register system: %w", err)
|
||||
}
|
||||
|
||||
// 初始化所有工具
|
||||
if err := registry.InitializeAll(ctx); err != nil {
|
||||
return fmt.Errorf("initialize: %w", err)
|
||||
}
|
||||
@@ -54,58 +49,20 @@ func Run(cfg *Config, tools ...tool.Tool) error {
|
||||
}
|
||||
}()
|
||||
|
||||
// 创建 MCP Server
|
||||
mcpServer := server.NewMCPServer(
|
||||
"ops-mcp",
|
||||
version,
|
||||
server.WithLogging(),
|
||||
)
|
||||
mcpServer := server.NewMCPServer("ops-mcp", version, server.WithLogging())
|
||||
|
||||
// 注册所有工具处理器
|
||||
if err := registry.RegisterAll(mcpServer); err != nil {
|
||||
return fmt.Errorf("register all: %w", err)
|
||||
}
|
||||
|
||||
// 选择传输模式
|
||||
switch cfg.Server.Transport {
|
||||
switch transport {
|
||||
case "stdio":
|
||||
return runStdio(ctx, mcpServer)
|
||||
slog.Info("starting MCP server", "transport", "stdio")
|
||||
return server.ServeStdio(mcpServer)
|
||||
case "sse":
|
||||
return runSSE(ctx, mcpServer, cfg)
|
||||
slog.Info("starting MCP server", "transport", "sse", "addr", ":8080")
|
||||
return server.NewSSEServer(mcpServer).Start(":8080")
|
||||
default:
|
||||
return fmt.Errorf("unknown transport: %s (expect stdio or sse)", cfg.Server.Transport)
|
||||
return fmt.Errorf("unknown transport: %s", transport)
|
||||
}
|
||||
}
|
||||
|
||||
func runStdio(ctx context.Context, mcpServer *server.MCPServer) error {
|
||||
slog.Info("starting MCP server", "transport", "stdio")
|
||||
go func() {
|
||||
<-ctx.Done()
|
||||
slog.Info("shutting down stdio server")
|
||||
}()
|
||||
return server.ServeStdio(mcpServer)
|
||||
}
|
||||
|
||||
func runSSE(ctx context.Context, mcpServer *server.MCPServer, cfg *Config) error {
|
||||
sseServer := server.NewSSEServer(mcpServer)
|
||||
|
||||
addr := cfg.Server.Addr
|
||||
if addr == "" {
|
||||
addr = ":8080"
|
||||
}
|
||||
|
||||
slog.Info("starting MCP server", "transport", "sse", "addr", addr)
|
||||
|
||||
httpServer := &http.Server{Addr: addr, Handler: sseServer}
|
||||
|
||||
go func() {
|
||||
<-ctx.Done()
|
||||
slog.Info("shutting down HTTP server")
|
||||
httpServer.Shutdown(context.Background())
|
||||
}()
|
||||
|
||||
if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user