membank/aio-mcp

aio-mcp

aio-system 的自定义 MCP 服务器（Go）。作为 AI Agent 的「手」，以 MCP 协议对外暴露工具；后续的 SQLite / 文件 / 打印 / 检测等自定义工具都往这里加。

基于官方 SDK：github.com/modelcontextprotocol/go-sdk（v1.7.0）。

目录结构

aio-mcp/
├── main.go                  # 入口：选择 stdio / http 传输
└── internal/
    ├── server/
    │   └── server.go        # 服务器装配（创建实例 + 注册工具）
    └── tools/
        ├── registry.go      # 工具注册总表 RegisterAll
        ├── ping.go          # 示例工具：健康检查（无参数）
        └── echo.go          # 示例工具：参数传递（有参数）

运行

go build -o aio-mcp.exe .        # 构建

# stdio 模式（默认，供本地 Agent 拉起，如 Reasonix / Claude）
aio-mcp.exe

# http 模式（远程客户端访问）
aio-mcp.exe -transport http -addr localhost:8000

依赖拉取走内网代理，若需直连官方源：go env -w GOPROXY=https://proxy.golang.org,direct。

新增一个工具

1. 在 internal/tools/ 新建文件（如 weather.go），用参数 struct + 泛型 mcp.AddTool 注册：

   package tools
   
   import (
       "context"
       "github.com/modelcontextprotocol/go-sdk/mcp"
   )
   
   type weatherArgs struct {
       City string `json:"city" jsonschema:"城市名，必填"`
       Days int    `json:"days,omitempty" jsonschema:"预报天数，默认 1"`
   }
   
   func registerWeather(s *mcp.Server) {
       mcp.AddTool(s, &mcp.Tool{
           Name:        "weather",
           Description: "查询某城市未来几天的天气。",
       }, func(_ context.Context, _ *mcp.CallToolRequest, args weatherArgs) (*mcp.CallToolResult, any, error) {
           // ...业务逻辑...
           return &mcp.CallToolResult{
               Content: []mcp.Content{&mcp.TextContent{Text: "晴，25℃"}},
           }, nil, nil
       })
   }
   

2. 在 internal/tools/registry.go 的 RegisterAll 中追加一行：registerWeather(s)。

工具需要共享状态（SQLite 连接等）时，参考 registry.go 顶部注释，把 RegisterAll 演进为带 deps 参数。

冒烟测试

stdio 模式用换行分隔的 JSON-RPC 直接喂协议消息：

printf '%s\n' \
'{"jsonrpc":"2.0","id":1,"method":"initialize","params":{"protocolVersion":"2025-06-18","capabilities":{},"clientInfo":{"name":"test","version":"0.0.1"}}}' \
'{"jsonrpc":"2.0","method":"notifications/initialized"}' \
'{"jsonrpc":"2.0","id":2,"method":"tools/list"}' \
'{"jsonrpc":"2.0","id":3,"method":"tools/call","params":{"name":"echo","arguments":{"message":"你好","times":2}}}' \
| aio-mcp.exe

http 模式：

curl -X POST http://localhost:8000/mcp -H "Content-Type: application/json" \
  -d '{"jsonrpc":"2.0","id":1,"method":"initialize","params":{"protocolVersion":"2025-06-18","capabilities":{},"clientInfo":{"name":"curl","version":"0.0.1"}}}'

接入 Agent（以 Reasonix / Claude 为例）

在客户端 MCP 配置中注册，指向构建产物：

{ "mcpServers": { "aio-mcp": { "command": "D:\\workspace\\membank\\aio-mcp\\aio-mcp.exe" } } }

注意事项

• stdio 模式下 stdout 是协议通道：日志已固定输出到 stderr，业务代码请勿往 stdout 打印。
• 工具名只允许 a-z A-Z 0-9 _ - .。