一文带你学会使用Go语言实现自己的MCP服务端

package main

import (
	"context"
	"errors"
	"github.com/mark3labs/mcp-go/mcp"
	"github.com/mark3labs/mcp-go/server"
	"log"
)

func main() {
	s := server.newmcpserver("calculatorserver", "1.0.0")

	// 添加工具
	calculatortool := mcp.newtool("calculate",
		mcp.withdescription("执行基本的算术运算"),
		mcp.withstring("operation",
			mcp.required(),
			mcp.description("要执行的算术运算类型"),
			mcp.enum("multiply", "divide"),
		),
		mcp.withnumber("x",
			mcp.required(),
			mcp.description("第一个数字"),
		),
		mcp.withnumber("y",
			mcp.required(),
			mcp.description("第二个数字"),
		),
	)

	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 "multiply":
			result = x * y
		case "divide":
			if y == 0 {
				return nil, errors.new("不允许除以零")
			}
			result = x / y
		}

		return mcp.formatnumberresult(result), nil
	})

	//sseserver := server.newsseserver(s, server.withbaseurl("http://localhost:8082"))
	//log.printf("sse server listening on :8082")
	//if err := sseserver.start(":8082"); err != nil {
	//	log.fatalf("server error: %v", err)
	//}
	// 启动基于 stdio 的服务器
	if err := server.servestdio(s); err != nil {
		log.printf("server error: %v\n", err)
	}
}

package main

import (
	"context"
	"fmt"
	"os"
	"strconv"
	"time"

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

func main() {
	if len(os.args) < 2 {
		fmt.println("请提供至少一个参数。")
		return
	}
	// 输出程序名称
	// 遍历并输出所有参数
	a, _ := strconv.parsefloat(os.args[1], 64)
	b, _ := strconv.parsefloat(os.args[2], 64)
        // 这里的路径是上面编译的mcp-server可执行文件
	mcpclient, err := client.newstdiomcpclient("/users/yourusername/workspace/go/src/mcp-server/mcp-server", []string{})
	if err != nil {
		panic(err)
	}
	defer mcpclient.close()

	ctx, cancel := context.withtimeout(context.background(), 30*time.second)
	defer cancel()

	initrequest := mcp.initializerequest{}
	initrequest.params.protocolversion = mcp.latest_protocol_version
	initrequest.params.clientinfo = mcp.implementation{
		name:    "client demo",
		version: "1.0.0",
	}

	initresult, err := mcpclient.initialize(ctx, initrequest)
	if err != nil {
		panic(err)
	}
	fmt.printf("初始化成功，服务器信息: %s %s\n", initresult.serverinfo.name, initresult.serverinfo.version)

	// 调用工具
	toolrequest := mcp.calltoolrequest{
		request: mcp.request{
			method: "tools/call",
		},
	}
	toolrequest.params.name = "calculate"
	toolrequest.params.arguments = map[string]any{
		"operation": "multiply",
		"x":         a,
		"y":         b,
	}

	result, err := mcpclient.calltool(ctx, toolrequest)
	if err != nil {
		panic(err)
	}
	fmt.printf("%f * %f = %s\n", a, b, result.content[0].(mcp.textcontent).text)
}

{
  "mcpservers": {
    "mcp-calculator-server": {
      "command": "/users/yourusername/workspace/go/src/mcp-server/mcp-server"
    }
  }
}

{
  "mcpservers": {
    "mcp-calculator-server": {
      "command": "/users/yourusername/workspace/go/src/mcp-server/mcp-server"
    }
  }
}