精简项目:只保留 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:
yangzhaohan
2026-07-06 10:42:23 +08:00
parent fb2c3a6cae
commit 093511bbdd
23 changed files with 144 additions and 1352 deletions
-17
View File
@@ -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
}
}
+1 -16
View File
@@ -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
View File
@@ -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
}