synoth/ops-mcp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

精简项目：只保留 docker_logs + health + info 三个工具

Browse Source 
- 移除 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
Download Patch File Download Diff File Expand all files Collapse all files
internal/server/server.go
+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
}