From b326d8cc8ce2932df651e8020324ca17969d090a Mon Sep 17 00:00:00 2001 From: zy Date: Thu, 16 Oct 2025 23:32:01 +0800 Subject: [PATCH] feat: add panic recovery middleware for MCP tools (#246) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * feat: 添加统一的 panic recovery 错误处理机制 实现了类似 Gin middleware 的 panic recovery 机制： - 添加 withPanicRecovery 泛型函数，捕获 handler 中的 panic - 包装所有 11 个 MCP 工具的 handler 函数 - 记录完整的错误日志和堆栈信息 - 向客户端返回友好的错误提示信息 - 保证程序在单个工具出错时不会崩溃 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude * update comments --------- Co-authored-by: Claude --- mcp_server.go | 80 ++++++++++++++++++++++++++++++++++++--------------- 1 file changed, 57 insertions(+), 23 deletions(-) diff --git a/mcp_server.go b/mcp_server.go index d5e31ee..7bc4898 100644 --- a/mcp_server.go +++ b/mcp_server.go @@ -3,6 +3,8 @@ package main import ( "context" "encoding/base64" + "fmt" + "runtime/debug" "github.com/modelcontextprotocol/go-sdk/mcp" "github.com/sirupsen/logrus" @@ -89,6 +91,38 @@ func InitMCPServer(appServer *AppServer) *mcp.Server { return server } +func withPanicRecovery[T any]( + toolName string, + handler func(context.Context, *mcp.CallToolRequest, T) (*mcp.CallToolResult, any, error), +) func(context.Context, *mcp.CallToolRequest, T) (*mcp.CallToolResult, any, error) { + + return func(ctx context.Context, req *mcp.CallToolRequest, args T) (result *mcp.CallToolResult, resp any, err error) { + defer func() { + if r := recover(); r != nil { + logrus.WithFields(logrus.Fields{ + "tool": toolName, + "panic": r, + }).Error("Tool handler panicked") + + logrus.Errorf("Stack trace:\n%s", debug.Stack()) + + result = &mcp.CallToolResult{ + Content: []mcp.Content{ + &mcp.TextContent{ + Text: fmt.Sprintf("工具 %s 执行时发生内部错误: %v\n\n请查看服务端日志获取详细信息。", toolName, r), + }, + }, + IsError: true, + } + resp = nil + err = nil + } + }() + + return handler(ctx, req, args) + } +} + // registerTools 注册所有 MCP 工具 func registerTools(server *mcp.Server, appServer *AppServer) { // 工具 1: 检查登录状态 @@ -97,10 +131,10 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "check_login_status", Description: "检查小红书登录状态", }, - func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) { + withPanicRecovery("check_login_status", func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) { result := appServer.handleCheckLoginStatus(ctx) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 2: 获取登录二维码 @@ -109,10 +143,10 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "get_login_qrcode", Description: "获取登录二维码（返回 Base64 图片和超时时间）", }, - func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) { + withPanicRecovery("get_login_qrcode", func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) { result := appServer.handleGetLoginQrcode(ctx) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 3: 发布内容 @@ -121,7 +155,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "publish_content", Description: "发布小红书图文内容", }, - func(ctx context.Context, req *mcp.CallToolRequest, args PublishContentArgs) (*mcp.CallToolResult, any, error) { + withPanicRecovery("publish_content", func(ctx context.Context, req *mcp.CallToolRequest, args PublishContentArgs) (*mcp.CallToolResult, any, error) { // 转换参数格式到现有的 handler argsMap := map[string]interface{}{ "title": args.Title, @@ -131,19 +165,19 @@ func registerTools(server *mcp.Server, appServer *AppServer) { } result := appServer.handlePublishContent(ctx, argsMap) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 4: 获取Feed列表 mcp.AddTool(server, &mcp.Tool{ Name: "list_feeds", - Description: "获取用户发布的内容列表", + Description: "获取首页 Feeds 列表", }, - func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) { + withPanicRecovery("list_feeds", func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) { result := appServer.handleListFeeds(ctx) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 5: 搜索内容 @@ -152,10 +186,10 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "search_feeds", Description: "搜索小红书内容（需要已登录）", }, - func(ctx context.Context, req *mcp.CallToolRequest, args SearchFeedsArgs) (*mcp.CallToolResult, any, error) { + withPanicRecovery("search_feeds", func(ctx context.Context, req *mcp.CallToolRequest, args SearchFeedsArgs) (*mcp.CallToolResult, any, error) { result := appServer.handleSearchFeeds(ctx, args) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 6: 获取Feed详情 @@ -164,14 +198,14 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "get_feed_detail", Description: "获取小红书笔记详情，返回笔记内容、图片、作者信息、互动数据（点赞/收藏/分享数）及评论列表", }, - func(ctx context.Context, req *mcp.CallToolRequest, args FeedDetailArgs) (*mcp.CallToolResult, any, error) { + withPanicRecovery("get_feed_detail", func(ctx context.Context, req *mcp.CallToolRequest, args FeedDetailArgs) (*mcp.CallToolResult, any, error) { argsMap := map[string]interface{}{ "feed_id": args.FeedID, "xsec_token": args.XsecToken, } result := appServer.handleGetFeedDetail(ctx, argsMap) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 7: 获取用户主页 @@ -180,14 +214,14 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "user_profile", Description: "获取指定的小红书用户主页，返回用户基本信息，关注、粉丝、获赞量及其笔记内容", }, - func(ctx context.Context, req *mcp.CallToolRequest, args UserProfileArgs) (*mcp.CallToolResult, any, error) { + withPanicRecovery("user_profile", func(ctx context.Context, req *mcp.CallToolRequest, args UserProfileArgs) (*mcp.CallToolResult, any, error) { argsMap := map[string]interface{}{ "user_id": args.UserID, "xsec_token": args.XsecToken, } result := appServer.handleUserProfile(ctx, argsMap) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 8: 发表评论 @@ -196,7 +230,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "post_comment_to_feed", Description: "发表评论到小红书笔记", }, - func(ctx context.Context, req *mcp.CallToolRequest, args PostCommentArgs) (*mcp.CallToolResult, any, error) { + withPanicRecovery("post_comment_to_feed", func(ctx context.Context, req *mcp.CallToolRequest, args PostCommentArgs) (*mcp.CallToolResult, any, error) { argsMap := map[string]interface{}{ "feed_id": args.FeedID, "xsec_token": args.XsecToken, @@ -204,7 +238,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { } result := appServer.handlePostComment(ctx, argsMap) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 9: 发布视频（仅本地文件） @@ -213,7 +247,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "publish_with_video", Description: "发布小红书视频内容（仅支持本地单个视频文件）", }, - func(ctx context.Context, req *mcp.CallToolRequest, args PublishVideoArgs) (*mcp.CallToolResult, any, error) { + withPanicRecovery("publish_with_video", func(ctx context.Context, req *mcp.CallToolRequest, args PublishVideoArgs) (*mcp.CallToolResult, any, error) { argsMap := map[string]interface{}{ "title": args.Title, "content": args.Content, @@ -222,7 +256,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { } result := appServer.handlePublishVideo(ctx, argsMap) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 10: 点赞笔记 @@ -231,7 +265,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "like_feed", Description: "为指定笔记点赞或取消点赞（如已点赞将跳过点赞，如未点赞将跳过取消点赞）", }, - func(ctx context.Context, req *mcp.CallToolRequest, args LikeFeedArgs) (*mcp.CallToolResult, any, error) { + withPanicRecovery("like_feed", func(ctx context.Context, req *mcp.CallToolRequest, args LikeFeedArgs) (*mcp.CallToolResult, any, error) { argsMap := map[string]interface{}{ "feed_id": args.FeedID, "xsec_token": args.XsecToken, @@ -239,7 +273,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { } result := appServer.handleLikeFeed(ctx, argsMap) return convertToMCPResult(result), nil, nil - }, + }), ) // 工具 11: 收藏笔记 @@ -248,7 +282,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { Name: "favorite_feed", Description: "收藏指定笔记或取消收藏（如已收藏将跳过收藏，如未收藏将跳过取消收藏）", }, - func(ctx context.Context, req *mcp.CallToolRequest, args FavoriteFeedArgs) (*mcp.CallToolResult, any, error) { + withPanicRecovery("favorite_feed", func(ctx context.Context, req *mcp.CallToolRequest, args FavoriteFeedArgs) (*mcp.CallToolResult, any, error) { argsMap := map[string]interface{}{ "feed_id": args.FeedID, "xsec_token": args.XsecToken, @@ -256,7 +290,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) { } result := appServer.handleFavoriteFeed(ctx, argsMap) return convertToMCPResult(result), nil, nil - }, + }), ) logrus.Infof("Registered %d MCP tools", 11)