一文弄懂用Go實現(xiàn)MCP服務(wù)的示例代碼

更新時間：2025年04月21日 10:23:58 作者：王中陽講編程

本文主要介紹了一文弄懂用Go實現(xiàn)MCP服務(wù)的示例代碼,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

為什么MCP如此重要？

在過去，如果想要讓AI處理特定的數(shù)據(jù)，通常只能依賴于預(yù)訓(xùn)練數(shù)據(jù)或者手動上傳數(shù)據(jù)，這既麻煩又低效。即便對于強大的AI模型而言，也存在數(shù)據(jù)隔離的問題，無法直接訪問新的數(shù)據(jù)源，每次更新數(shù)據(jù)都需要重新訓(xùn)練或上傳?，F(xiàn)在，MCP解決了這個問題，它使得AI不再局限于靜態(tài)知識庫，而是能夠像人類一樣調(diào)用搜索引擎、訪問本地文件、連接API服務(wù)等，極大提升了AI的動態(tài)交互能力。

MCP總體架構(gòu)

MCP的核心是“客戶端-服務(wù)器”架構(gòu)，其中MCP客戶端可以連接到多個服務(wù)器?？蛻舳耸侵赶Ｍㄟ^MCP訪問數(shù)據(jù)的應(yīng)用程序，如CLI工具、IDE插件或AI應(yīng)用。

使用mcp-go構(gòu)建MCP服務(wù)端與客戶端

要開始使用Go語言構(gòu)建MCP項目，首先需要安裝mcp-go庫，這是Go語言實現(xiàn)的Model Context Protocol庫，支持LLM應(yīng)用與外部數(shù)據(jù)源和工具之間的無縫集成。

go get github.com/mark3labs/mcp-go

構(gòu)建MCP服務(wù)端

接下來，我們將演示如何使用mcp-go提供的server模塊來構(gòu)建一個通過stdio方式連接的MCP服務(wù)器。

創(chuàng)建Server對象

s := server.NewMCPServer("My Server", "1.0.0")

添加工具（Tools）

例如，我們可以創(chuàng)建一個簡單的計算器工具，這次我們實現(xiàn)乘法和除法功能：

calculatorTool := mcp.NewTool("calculate",
    mcp.WithDescription("執(zhí)行基本的算術(shù)運算"),
    mcp.WithString("operation",
        mcp.Required(),
        mcp.Description("要執(zhí)行的算術(shù)運算類型"),
        mcp.Enum("multiply", "divide"), // 修改為僅支持乘法和除法
    ),
    mcp.WithNumber("x",
        mcp.Required(),
        mcp.Description("第一個數(shù)字"),
    ),
    mcp.WithNumber("y",
        mcp.Required(),
        mcp.Description("第二個數(shù)字"),
    ),
)

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
})

添加資源（Resources）

同樣地，我們也可以注冊一些靜態(tài)資源，比如README.md文件的內(nèi)容：

resource := mcp.NewResource(
    "docs://readme",
    "項目說明文檔",
    mcp.WithResourceDescription("項目的 README 文件"),
    mcp.WithMIMEType("text/markdown"),
)

s.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
})

啟動基于stdio傳輸類型的服務(wù)器

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

以上步驟完成后，我們就成功搭建了一個基礎(chǔ)的MCP服務(wù)器。

構(gòu)建MCP客戶端

接著，我們將展示如何使用mcp-go提供的client模塊構(gòu)建一個連接至上述MCP服務(wù)器的客戶端。

創(chuàng)建MCP客戶端

mcpClient, err := client.NewStdioMCPClient("./client/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("初始化成功，服務(wù)器信息: %s %s\n", initResult.ServerInfo.Name, initResult.ServerInfo.Version)

調(diào)用遠程工具

最后，我們可以通過構(gòu)造CallToolRequest來調(diào)用服務(wù)器上的工具，如下所示：

toolRequest := mcp.CallToolRequest{
    Request: mcp.Request{
        Method: "tools/call",
    },
}
toolRequest.Params.Name = "calculate"
toolRequest.Params.Arguments = map[string]any{
    "operation": "multiply", // 調(diào)用乘法
    "x":         2,
    "y":         3,
}

result, err := mcpClient.CallTool(ctx, toolRequest)
if err != nil {
    panic(err)
}
fmt.Println("調(diào)用工具結(jié)果:", result.Content[0].(mcp.TextContent).Text)

完整代碼示例

以下是完整的代碼示例，包括服務(wù)端和客戶端的實現(xiàn)：

服務(wù)端代碼：

package main

import (
    "context"
    "errors"
    "fmt"
    "os"

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

func main() {
    s := server.NewMCPServer("Server Demo", "1.0.0")

    // 添加工具
    calculatorTool := mcp.NewTool("calculate",
        mcp.WithDescription("執(zhí)行基本的算術(shù)運算"),
        mcp.WithString("operation",
            mcp.Required(),
            mcp.Description("要執(zhí)行的算術(shù)運算類型"),
            mcp.Enum("multiply", "divide"),
        ),
        mcp.WithNumber("x",
            mcp.Required(),
            mcp.Description("第一個數(shù)字"),
        ),
        mcp.WithNumber("y",
            mcp.Required(),
            mcp.Description("第二個數(shù)字"),
        ),
    )

    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
    })

    // 啟動基于 stdio 的服務(wù)器
    if err := server.ServeStdio(s); err != nil {
        fmt.Printf("Server error: %v\n", err)
    }
}

客戶端代碼：

package main

import (
    "context"
    "fmt"
    "time"

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

func main() {
    mcpClient, err := client.NewStdioMCPClient("./client/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("初始化成功，服務(wù)器信息: %s %s\n", initResult.ServerInfo.Name, initResult.ServerInfo.Version)

    // 調(diào)用工具
    toolRequest := mcp.CallToolRequest{
        Request: mcp.Request{
            Method: "tools/call",
        },
    }
    toolRequest.Params.Name = "calculate"
    toolRequest.Params.Arguments = map[string]any{
        "operation": "multiply",
        "x":         2,
        "y":         3,
    }

    result, err := mcpClient.CallTool(ctx, toolRequest)
    if err != nil {
        panic(err)
    }
    fmt.Println("調(diào)用工具結(jié)果:", result.Content[0].(mcp.TextContent).Text)
}

希望這篇文章能幫助你快速入門Go語言下的MCP開發(fā)！

到此這篇關(guān)于一文弄懂用Go實現(xiàn)MCP服務(wù)的示例代碼的文章就介紹到這了,更多相關(guān)Go MCP服務(wù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: