MCP 实战:用 Go 语言开发一个查询 IP 信息的 MCP 服务器

该文章已被 Model Context Protocol(MCP) 中文教程讲解 收录,欢迎 star 收藏。

📝 前言

随着 MCP 的快速普及和广泛应用,MCP 服务器也层出不穷。大多数开发者使用的 MCP 服务器开发库是官方提供的 typescript-sdk,而作为 Go 开发者,我们也可以借助优秀的第三方库去开发 MCP 服务器,例如 ThinkInAIXYZ/go-mcp

本文将详细介绍如何在 Go 语言中使用 go-mcp 库来开发一个查询 IP 信息的 MCP 服务器。

准备好了吗?准备一杯你最喜欢的咖啡或茶,随着本文一探究竟吧。

🌐 mcp-ip-geo 服务器

mcp-ip-geo 是一个用于查询 IP 信息的 MCP 服务器,项目已开源,仓库地址:mcp-ip-geo

📁 目录结构说明

csharp 复制代码
├─cmd
│  └─mcp-ip-geo
└─internal
    ├─domain
    ├─server
    ├─service
    └─tools
  • cmd/mcp-ip-geo:应用的启动入口目录,包含如 main.go 启动文件。
  • internal/domain:定义项目中的核心数据结构,例如 IP API 的响应体等。
  • internal/serverMCP 服务器的核心逻辑实现。
  • internal/service:对接第三方服务的实现,如调用 IP 解析 API
  • internal/toolsMCP 工具的具体实现,支持灵活扩展和注册。

🔍 查询 IP 信息功能实现

代码位于 service 包中,通过 ip-api.com 提供的接口获取 IP 地理位置信息,具体实现如下:

go 复制代码
package service

import (
    "context"
    "fmt"
    "github.com/chenmingyong0423/mcp-ip-geo/internal/domain"
    "net/http"
    "time"

    httpchain "github.com/chenmingyong0423/go-http-chain"
)

func NewIpApiService() *IpApiService {
    return &IpApiService{
       host: "http://ip-api.com",
       client: httpchain.NewWithClient(&http.Client{
          Timeout: time.Second * 10,
       }),
    }
}

type IIpApiService interface {
    GetLocation(ctx context.Context, ip string) (*domain.IpApiResponse, error)
}

var _ IIpApiService = (*IpApiService)(nil)

type IpApiService struct {
    host   string
    client *httpchain.Client
}

func (s *IpApiService) GetLocation(ctx context.Context, ip string) (*domain.IpApiResponse, error) {
    var resp domain.IpApiResponse
    err := s.client.Get(fmt.Sprintf("%s/json/%s", s.host, ip)).DoAndParse(ctx, &resp)
    if err != nil {
       return nil, err
    }
    return &resp, nil
}

代码解释:

  • 服务初始化(NewIpApiService

    • 创建一个新的 IpApiService 实例。
    • 设置了 API 地址为 http://ip-api.com
    • 使用 httpchain 封装的 HTTP 客户端,设置请求超时时间为 10 秒。
  • 接口定义(IIpApiService

    • 定义了服务对外暴露的功能:GetLocation 方法,用于获取 IP 地理位置信息。
    • 使用接口有助于后续做依赖注入、mock 测试等。
    • var _ IIpApiService = (*IpApiService)(nil) 这行代码用于编译时检查,确保 IpApiService 实现了 IIpApiService 接口。
  • 结构体定义(IpApiService

    • 包含两个字段:
    • hostAPI 的基础地址。
    • client:封装的 HTTP 客户端,类型为 *httpchain.Client
  • 核心方法实现(GetLocation

    • 根据传入的 IP 构造请求地址:http://ip-api.com/json/{ip}
    • 使用 httpchain 库发起 GET 请求,并将结果解析到 domain.IpApiResponse 结构体中。

🧰 工具实现

🔧 工具管理

代码位于 tools 包中,用于管理工具,具体实现如下:

go 复制代码
package tools

import (
    "github.com/ThinkInAIXYZ/go-mcp/protocol"
    "github.com/ThinkInAIXYZ/go-mcp/server"
)

type ToolFunc func() (tool *protocol.Tool, toolHandler server.ToolHandlerFunc)

func GetToolFuncList() []ToolFunc {
    return []ToolFunc{
       SingleIpParser,
    }
}

代码解释:

  • ToolFunc 类型定义

    • 定义了一个函数类型 ToolFunc,返回两个值:

      • *protocol.Tool:工具的元信息;
      • server.ToolHandlerFunc:该工具的处理逻辑函数。
    • 用这种方式可以将 工具的定义工具的执行逻辑 一并管理,后续在定义工具时都可以通过实现该函数签名进行表示。

  • GetToolFuncList 函数

    • 返回一个 ToolFunc 列表。
    • 当前只注册了一个工具:SingleIpParser,但这种结构易于扩展,后续只需往列表中添加新的工具函数即可。
    • 通过集中注册,应用在初始化时可以统一加载所有工具。

🌍 查询单个 IP 信息工具的实现

代码位于 tools 包中,用于查询单个 IP 信息,具体实现如下:

go 复制代码
package tools

import (
    "context"
    "encoding/json"
    "github.com/ThinkInAIXYZ/go-mcp/protocol"
    "github.com/ThinkInAIXYZ/go-mcp/server"
    "github.com/chenmingyong0423/mcp-ip-geo/internal/service"
)

var singleIpParserTool *protocol.Tool

type ipRequest struct {
    Ip string `json:"ip"`
}

func init() {
    var err error
    singleIpParserTool, err = protocol.NewTool("ip-details", "a tool that provides IP geolocation information", ipRequest{})
    if err != nil {
       panic(err)
    }
}

func SingleIpParser() (*protocol.Tool, server.ToolHandlerFunc) {
    ipApiService := service.NewIpApiService()

    return singleIpParserTool, func(toolRequest *protocol.CallToolRequest) (*protocol.CallToolResult, error) {
       var req ipRequest
       if err := protocol.VerifyAndUnmarshal(toolRequest.RawArguments, &req); err != nil {
          return nil, err
       }
       resp, err := ipApiService.GetLocation(context.Background(), req.Ip)
       if err != nil {
          return nil, err
       }

       marshal, err := json.Marshal(resp)
       if err != nil {
          return nil, err
       }

       return &protocol.CallToolResult{
          Content: []protocol.Content{
             protocol.TextContent{
                Type: "text",
                Text: string(marshal),
             },
          },
       }, nil
    }
}

代码解释:

  • 全局变量声明

    • singleIpParserTool:存储工具元信息的协议工具对象
    • ipRequest:定义工具输入参数结构体,包含 ip 字符串字段
  • 初始化函数(init

    • 在包加载时通过 protocol.NewTool 创建工具元信息
    • 指定工具标识符 ip-details,描述信息和输入参数结构体 ipRequest{}
    • 错误处理采用 panic,确保工具元信息必须正确初始化
  • 工具注册函数(SingleIpParser

    • 创建 IpApiService 服务实例用于 IP 定位查询

    • 返回两个值:

      • 预定义的 singleIpParserTool 元信息对象
      • 工具处理函数
  • 工具处理函数

    • 参数验证与解析

      • 调用 protocol.VerifyAndUnmarshal 验证请求参数有效性
      • 将原始参数反序列化到 ipRequest 结构体
    • 服务调用

      • 使用 ipApiService.GetLocation 获取 IP 地理位置信息
    • 结果处理

      • 将服务响应结果序列化为 JSON 字符串并包装为 protocol.CallToolResult 结构体返回

🚀 服务器的创建与启动

代码位于 server 包中,用于初始化服务并启动服务端,具体实现如下:

go 复制代码
package server

import (
    "github.com/ThinkInAIXYZ/go-mcp/server"
    "github.com/ThinkInAIXYZ/go-mcp/transport"
    "github.com/chenmingyong0423/mcp-ip-geo/internal/tools"
)

func Run(address string) error {
    var err error

    var ts transport.ServerTransport
    if address == "" {
       ts = transport.NewStdioServerTransport()
    } else {
       ts, err = transport.NewSSEServerTransport(address)
       if err != nil {
          return err
       }
    }

    s, err := server.NewServer(ts)
    if err != nil {
       return err
    }

    toolFuncList := tools.GetToolFuncList()
    for _, tool := range toolFuncList {
       s.RegisterTool(tool())
    }

    return s.Run()
}

代码解释:

  • 传输层初始化

    • 根据 address 参数判断运行模式:

      • 空地址模式 :使用 NewStdioServerTransport 创建标准输入输出传输,适用于命令行工具等场景。
      • 指定地址模式 :使用 NewSSEServerTransport 创建 SSE (Server-Sent Events) 传输,适用于 HTTP 长连接服务。
  • 服务实例化

    • 使用 server.NewServer 方法创建服务实例,注入配置好的传输层对象 ts
  • 工具注册

    • 调用 tools.GetToolFuncList 获取所有预定义的工具函数列表。

    • 遍历工具列表,通过 s.RegisterTool(tool()) 注册每个工具:

      • tool() 执行后返回元信息 *protocol.Tool 和处理函数 ToolHandlerFunc
  • 服务启动

    • 调用 s.Run() 启动服务,开始监听请求。

🧩 主程序入口实现

代码位于 main 包中,作为程序启动入口,具体实现如下:

go 复制代码
package main

import (
    "flag"
    "github.com/chenmingyong0423/mcp-ip-geo/internal/server"
)

func main() {
    addr := flag.String("address", "", "The host and port to run the sse server")
    flag.Parse()

    if err := server.Run(*addr); err != nil {
       panic(err)
    }
}

代码解释:

  • 命令行参数解析

    • 定义 address 参数:

      • 参数名称:-address
      • 默认值:空字符串
      • 描述:指定 SSE 服务运行的地址和端口
    • 调用 flag.Parse() 解析命令行参数

  • 服务启动

    • 调用 server.Run(*addr) 启动服务
    • 将解析后的 address 参数值传递给服务启动函数

✅ 小结

本文将详细介绍 mcp-ip-geo ------ 一个用于查询 IP 信息的 MCP 服务器的实现细节。该服务器目前支持两种数据传输方式:stdioSSE(Server-Sent Events)。未来还计划支持 Streamable HTTP 传输方式,并持续扩展更多实用的工具(tools)模块。


你好,我是陈明勇,一名热爱技术、乐于分享的开发者,同时也是开源爱好者。

我专注于分享 Go 语言相关的技术知识,同时也会深入探讨 AI 领域的前沿技术。

成功的路上并不拥挤,有没有兴趣结个伴?

Go 开源库代表作go-mongoxgo-optioner

相关推荐
云天徽上4 分钟前
【数据可视化-27】全球网络安全威胁数据可视化分析(2015-2024)
人工智能·安全·web安全·机器学习·信息可视化·数据分析
ONEYAC唯样11 分钟前
“在中国,为中国” 英飞凌汽车业务正式发布中国本土化战略
大数据·人工智能
mozun202016 分钟前
产业观察:哈工大机器人公司2025.4.22
大数据·人工智能·机器人·创业创新·哈尔滨·名校
-一杯为品-19 分钟前
【深度学习】#9 现代循环神经网络
人工智能·rnn·深度学习
硅谷秋水21 分钟前
ORION:通过视觉-语言指令动作生成的一个整体端到端自动驾驶框架
人工智能·深度学习·机器学习·计算机视觉·语言模型·自动驾驶
Java中文社群43 分钟前
最火向量数据库Milvus安装使用一条龙!
java·人工智能·后端
豆芽8191 小时前
强化学习(Reinforcement Learning, RL)和深度学习(Deep Learning, DL)
人工智能·深度学习·机器学习·强化学习
山北雨夜漫步1 小时前
机器学习 Day14 XGboost(极端梯度提升树)算法
人工智能·算法·机器学习
JAVA百练成神1 小时前
深度理解spring——BeanFactory的实现
java·后端·spring
LangGPT提示词社区1 小时前
MCP:AI世界的万能连接器,专家都在关注的下一代标准
mcp