go get "github.com/mark3labs/mcp-go"

//新建mcp 服務(wù)
mcpServer := server.NewMCPServer("ziyi Mcp Server", "1.0.0")
//對(duì)外提供 stdio mcp server服務(wù)
if err := server.ServeStdio(mcpServer); err != nil {
	panic(err)
}

//新增一個(gè)mcp server對(duì)外暴露的靜態(tài)資源（固定URI），比如：對(duì)外暴露項(xiàng)目的操作手冊(cè)以及部署方式等，就可以通過靜態(tài)資源README文件的方式來告訴大模型
resource := mcp.NewResource(
	"docs://readme",
	"項(xiàng)目的README文件",
	mcp.WithResourceDescription("這是一個(gè)項(xiàng)目的README文件"),
	mcp.WithMIMEType("text/markdown"),
)
// 添加對(duì)靜態(tài)資源的處理器
mcpServer.AddResource(resource, func(ctx context.Context, request mcp.ReadResourceRequest) ([]mcp.ResourceContents, error) {
	content, err := os.ReadFile("README.md")
	if err != nil {
		return nil, err
	}

	return []mcp.ResourceContents{
		mcp.TextResourceContents{
			URI:      "docs://readme",
			MIMEType: "text/markdown",
			Text:     string(content),
		},
	}, nil
})

//給該mcp server添加計(jì)算能力(Tools)
// 1. 描述該工具，以及調(diào)用該工具調(diào)用的參數(shù)機(jī)器含義
calculatorTool := mcp.NewTool("calculate",
	mcp.WithDescription("進(jìn)行基礎(chǔ)的數(shù)學(xué)運(yùn)算"),
	mcp.WithString("operation",
		mcp.Required(),
		mcp.Description("The arithmetic operation to perform"),
		mcp.Enum("add", "subtract", "multiply", "divide"),
	),
	mcp.WithNumber("x",
		mcp.Required(),
		mcp.Description("First number"),
	),
	mcp.WithNumber("y",
		mcp.Required(),
		mcp.Description("Second number"),
	),
)
// 2. 實(shí)現(xiàn)工具的具體處理邏輯，類比大模型function_calling中的func部分
mcpServer.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("Division by zero is not allowed")
		}
		result = x / y
	}

	return mcp.FormatNumberResult(result), nil
})

// 給mcpServer添加提示詞模版
mcpServer.AddPrompt(mcp.NewPrompt("打招呼",
	mcp.WithPromptDescription("A friendly greeting prompt"),
	mcp.WithArgument("name",
		mcp.ArgumentDescription("Name of the person to greet"),
	),
), func(ctx context.Context, request mcp.GetPromptRequest) (*mcp.GetPromptResult, error) {
	name := request.Params.Arguments["name"]
	if name == "" {
		name = "friend"
	}

	return mcp.NewGetPromptResult(
		"A friendly greeting",
		[]mcp.PromptMessage{
			mcp.NewPromptMessage(
				mcp.RoleAssistant,
				mcp.NewTextContent(fmt.Sprintf("Hello, %s! How can I help you today?", name)),
			),
		},
	), nil
})

package main

import (
	"context"
	"errors"
	"fmt"
	"github.com/mark3labs/mcp-go/mcp"
	"github.com/mark3labs/mcp-go/server"
	"io"
	"net"
	"net/http"
)

func main() {
	// Create MCP server
	s := server.NewMCPServer(
		"ip-mcp",
		"1.0.0",
	)

	// Add tool
	tool := mcp.NewTool("ip_query",
		mcp.WithDescription("query geo location of an IP address"),
		mcp.WithString("ip",
			mcp.Required(),
			mcp.Description("IP address to query"),
		),
	)

	// Add tool handler
	s.AddTool(tool, ipQueryHandler)

	// Start the stdio server
	if err := server.ServeStdio(s); err != nil {
		fmt.Printf("Server error: %v\n", err)
	}
}

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

	parsedIP := net.ParseIP(ip)
	if parsedIP == nil {
		return nil, errors.New("invalid IP address")
	}

	resp, err := http.Get("https://ip.rpcx.io/api/ip?ip=" + ip)
	if err != nil {
		return nil, fmt.Errorf("Error fetching IP information: %v", err)
	}
	defer resp.Body.Close()

	data, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("Error reading response body: %v", err)
	}
	fmt.Printf("call ip: %s  Response body: %s\n", ip, string(data))
	return mcp.NewToolResultText(string(data)), nil
}

# 執(zhí)行命令編譯mcp server
go build -o mcp-ip main.go