开发者

Go语言开发实现查询IP信息的MCP服务器

开发者 https://www.devze.com 2025-04-26 10:40 出处:网络 作者: 陈明勇
目录前言mcp-ip-geo 服务器目录结构说明查询 IP 信息功能实现工具实现工具管理查询单个 IP 信息工具的实现服务器的创建与启动主程序入口实现从源码构建本地构建安装预编译版本MCP 集成可执行文件集成(本地运行)HTT
目录
  • 前言
  • mcp-ip-geo 服务器
    • 目录结构说明
  • 查询 IP 信息功能实现
    • 工具实现
      • 工具管理
      • 查询单个 IP 信息工具的实现
    • 服务器的创建与启动
      • 主程序入口实现
        • 从源码构建
          • 本地构建
          • 安装预编译版本
        • MCP 集成
          • 可执行文件集成(本地运行)
          • HTTP 接口集成(连接到已运行的服务)
        • 效果演示
          • 小结

            前言

            随着 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。

            目录结构说明

            ├─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 地理位置信息,具体实现如下:

            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 包中,用于管理工具,具体实现如下:

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

            代码解释:

            • TkPXelUMeooolFunc 类型定义

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

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

            • GetToolFuncList 函数

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

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

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

            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.NewIpApiServhttp://www.devze.comice()
            
                return singleIpParserTool, func(toolRequest *protocol.CallToolRequest) (*protocol.CallToolResult, error) {
                   var req ipRequest
                   if err := protocol.VerifyAndUnmarshal(toolRequest.RawArguments, &http://www.devze.com;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 包中,用于初始化服务并启动服务端,具体实现如下:

            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 := sjavascripterver.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 包中,作为程序启动入口,具体实现如下:

            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 参数值传递给服务启动函数

            从源码构建

            本地构建

            使用 Go 命令

            # 在类 Unix 系统(linux/MACOS)上
            go build -o mcp-ip-geo ./cmd/mcp-ip-geo
            
            # 在 Windows 上
            go build -o mcp-ip-geo.exe .\cmd\mcp-ip-geo
            

            使用 docker

            构建 Docker 镜像:

            docker build -t mcp-ip-geo-server .
            

            运行容器:

            docker run -d --name mcp-ip-geo-server -p 8000:8000 mcp-ip-geo-server
            

            安装预编译版本

            使用 Go 安装最新版本的服务:

            go install github.com/chenmingyong0423/mcp-ip-geo/cmd/mcp-ip-geo@latest
            

            MCP 集成

            你可以通过以下两种方式集成 mcp-ip-geo 服务:

            可执行文件集成(本地运行)

            {
              "mcpServers": {
                "mcp-ip-geo": {
                  "command": "/path/to/mcp-ip-geo"
                }
              }
            }
            

            HTTP 接口集成(连接到已运行的服务)

            {
              "mcpServers": {
                "mcp-ip-geo": {
                  "url": "http://host:port/sse"
                }
              }
            }
            

            效果演示

            Go语言开发实现查询IP信息的MCP服务器

            小结

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

            到此这篇关于Go语言开发实现查询IP信息的MCP服务器的文章就介绍到这了,更多相关Go MCP服务器内容请搜索编程客栈(www.devze.com)以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程客栈(www.devze.com)!

            0

            精彩评论

            暂无评论...
            验证码 换一张
            取 消

            关注公众号