9.结合AI客户端玩转go MCP项目开发

mkdir my_go_mcp
cd my_go_mcp
mkdir main
cd main
vim main.go

package main

import (
    "context"
    "errors"
    "flag"
    "fmt"

    "github.com/mark3labs/mcp-go/mcp"
    "github.com/mark3labs/mcp-go/server"
)

func main() {
    var transport string
    flag.StringVar(&transport, "t", "stdio", "Transport type (stdio or sse)")
    flag.Parse()
    // Create MCP server
    mcpServer := server.NewMCPServer(
        "Demo 🚀",
        "1.0.0",
    )

    // Add tool
    demoTool := mcp.NewTool("demoMcp",
        mcp.WithDescription("这是一个用于查询Demo信息的MCP服务"),
        mcp.WithString("demo_id",
            mcp.Required(),
            mcp.Description("可以是'demo_id','demoID','demoId'"),
        ),
    )

    mcpServer.AddTool(demoTool, demoHandler)

    if transport == "sse" {
        sseServer := server.NewSSEServer(mcpServer, server.WithBaseURL("http://localhost:8080"))
        fmt.Printf("SSE server listening on :8080\n")
        if err := sseServer.Start(":8080"); err != nil {
            fmt.Printf("Server error: %v\n", err)
        }
    } else {
        if err := server.ServeStdio(mcpServer); err != nil {
            fmt.Printf("Server error: %v\n", err)
        }
    }
}

func demoHandler(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
    demoId, ok := req.Params.Arguments["demo_id"].(string)
    if !ok {
        return nil, errors.New("demo_id must be a string")
    }

    if demoId == "100" {
        demoInfo, err := getDemo100()
        if err != nil {
            return nil, err
        }
        return mcp.NewToolResultText(fmt.Sprintf("demo 信息: %s", string(demoInfo))), nil
    }

    if demoId == "200" {
        demoInfo, err := getDemo200()
        if err != nil {
            return nil, err
        }
        return mcp.NewToolResultText(fmt.Sprintf("demo 信息: %s", string(demoInfo))), nil
    }

    return mcp.NewToolResultText(fmt.Sprintf("demo 信息: %s", "未找到")), nil
}

func getDemo100() (string, error) {
    return "x 等于 155，y 等于 255", nil
}

func getDemo200() (string, error) {
    return "x 等于 88，y 等于 99", nil
}

cd my_go_mcp
go mod init my_go_mcp
go mod tidy

go build -o demoMcp main.go

./demoMcp

./demoMcp -t sse

{
  "mcpServers": {
    "sseMcp": {
      "name": "我的 SSE 服务器",
      "transport": "sse",
      "url": "http://localhost:8080/sse"
    }
  }
}

package main

import (
    "context"
    "errors"
    "fmt"

    "github.com/mark3labs/mcp-go/mcp"
    "github.com/mark3labs/mcp-go/server"
)

func main() {
    // Create a new MCP server
    s := server.NewMCPServer(
        "Calculator Demo",
        "1.0.0",
        server.WithResourceCapabilities(true, true),
        server.WithLogging(),
        server.WithRecovery(),
    )
    // Add a calculator tool
    calculatorTool := mcp.NewTool("calculate",
        mcp.WithDescription("用于计算两个数的加减乘除"),
        mcp.WithString("operation",
            mcp.Required(),
            mcp.Description("用于计算两个数的加减乘除 (add, subtract, multiply, divide) 或 （+，-，*，/）"),
            mcp.Enum("add", "subtract", "multiply", "divide"),
        ),
        mcp.WithNumber("x",
            mcp.Required(),
            mcp.Description("参数1"),
        ),
        mcp.WithNumber("y",
            mcp.Required(),
            mcp.Description("参数2"),
        ),
    )
    // Add the calculator handler
    s.AddTool(calculatorTool, func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
        op := request.Params.Arguments["operation"].(string)
        x := request.Params.Arguments["x"].(float64)
        y := request.Params.Arguments["y"].(float64)
        var result float64
        switch op {
        case "add":
            result = x + y
        case "subtract":
            result = x - y
        case "multiply":
            result = x * y
        case "divide":
            if y == 0 {
                return nil, errors.New("Cannot divide by zero")
            }
            result = x / y
        }
        return mcp.NewToolResultText(fmt.Sprintf("%.2f", result)), nil
    })
    // Start the server
    if err := server.ServeStdio(s); err != nil {
        fmt.Printf("Server error: %v\n", err)
    }
}

go build -i calculateMcp calculate.go

{
  "mcpServers": {
    "calculateMcp": {
      "name": "计算x、y的值",
      "command": "/Users/xxxx你的项目路径/go/mcp/src/my_go_mcp/main/calculate"
    },
    "sseMcp": {
      "name": "获取x、y的值",
      "transport": "sse",
      "url": "http://localhost:8080/sse"
    }
  }
}