MCP-Golang项目深度测评：为AI插上Go语言翅膀的“万能接口”

mkdir smart-file-analyst && cd smart-file-analyst
go mod init smart-file-analyst
go get github.com/MegaGrindStone/go-mcp

package main

import (
    "context"
    "fmt"
    "os"
    "path/filepath"
    "strings"

    "github.com/MegaGrindStone/go-mcp"
    "github.com/MegaGrindStone/go-mcp/mcp"
)

// 工具1：分析目录文件大小
type AnalyzeDirInput struct {
    DirPath string `json:"dirPath"`
}
func analyzeDirectory(ctx context.Context, params mcp.CallToolParams) (*mcp.CallToolResult, error) {
    var input AnalyzeDirInput
    // 在实际项目中，这里需要更健壮的参数解析
    // 简化演示：假设参数已正确传递
    input.DirPath = params.Arguments["dirPath"].(string)

    var totalSize int64
    var fileList []string
    err := filepath.Walk(input.DirPath, func(path string, info os.FileInfo, err error) error {
        if err != nil || info.IsDir() {
            return nil
        }
        totalSize += info.Size()
        fileList = append(fileList, fmt.Sprintf("- %s (%.2f KB)", path, float64(info.Size())/1024))
        return nil
    })

    if err != nil {
        return mcp.NewCallToolResultError(mcp.ToolError{Message: err.Error()}), nil
    }

    summary := fmt.Sprintf("目录分析报告：\n路径: %s\n文件总数: %d\n总大小: %.2f MB\n文件列表:\n%s",
        input.DirPath, len(fileList), float64(totalSize)/(1024*1024), strings.Join(fileList, "\n"))
    return mcp.NewCallToolResultText(summary), nil
}

// 工具2：查找重复文件名
func findDuplicateNames(ctx context.Context, params mcp.CallToolParams) (*mcp.CallToolResult, error) {
    dirPath := params.Arguments["dirPath"].(string)
    nameMap := make(map[string][]string)

    err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
        if err != nil || info.IsDir() {
            return nil
        }
        fileName := info.Name()
        nameMap[fileName] = append(nameMap[fileName], path)
        return nil
    })

    if err != nil {
        return mcp.NewCallToolResultError(mcp.ToolError{Message: err.Error()}), nil
    }

    var duplicates []string
    for name, paths := range nameMap {
        if len(paths) > 1 {
            duplicates = append(duplicates, fmt.Sprintf("文件 '%s' 出现在:\n  * %s", name, strings.Join(paths, "\n  * ")))
        }
    }

    resultText := "未找到重复文件名的文件。"
    if len(duplicates) > 0 {
        resultText = fmt.Sprintf("发现 %d 组重复文件名:\n\n%s", len(duplicates), strings.Join(duplicates, "\n\n"))
    }
    return mcp.NewCallToolResultText(resultText), nil
}

// 实现工具服务器接口
type ToolServer struct{}
func (t *ToolServer) ListTools(ctx context.Context, req mcp.ListToolsRequest) (mcp.ListToolsResult, error) {
    return mcp.ListToolsResult{
        Tools: []mcp.Tool{
            {
                Name:        "analyze_directory",
                Description: "分析指定目录的文件大小和列表",
                InputSchema: map[string]interface{}{
                    "type": "object",
                    "properties": map[string]interface{}{
                        "dirPath": map[string]interface{}{
                            "type":        "string",
                            "description": "要分析的目录绝对路径",
                        },
                    },
                    "required": []string{"dirPath"},
                },
            },
            {
                Name:        "find_duplicate_filenames",
                Description: "在指定目录中查找具有相同文件名的文件",
                InputSchema: map[string]interface{}{
                    "type": "object",
                    "properties": map[string]interface{}{
                        "dirPath": map[string]interface{}{
                            "type":        "string",
                            "description": "要查找的目录绝对路径",
                        },
                    },
                    "required": []string{"dirPath"},
                },
            },
        },
    }, nil
}
func (t *ToolServer) CallTool(ctx context.Context, req mcp.CallToolRequest) (mcp.CallToolResult, error) {
    switch req.Params.Name {
    case "analyze_directory":
        return analyzeDirectory(ctx, req.Params)
    case "find_duplicate_filenames":
        return findDuplicateNames(ctx, req.Params)
    default:
        return mcp.CallToolResult{}, fmt.Errorf("未知工具: %s", req.Params.Name)
    }
}

func main() {
    // 创建工具服务器实例
    toolServer := &ToolServer{}
    // 创建MCP服务器，使用Stdio传输（便于与桌面客户端集成）
    srv := mcp.NewServer(
        mcp.Info{Name: "SmartFileAnalyst", Version: "1.0.0"},
        mcp.NewStdIO(os.Stdin, os.Stdout),
        mcp.WithToolServer(toolServer),
    )
    // 启动服务器（阻塞）
    if err := srv.Serve(); err != nil {
        fmt.Fprintf(os.Stderr, "服务器错误: %v\n", err)
        os.Exit(1)
    }
}