feat: 增加 mcp Tools 自动测试工具

server/mcp/current_time.go

@@ -2,6 +2,8 @@ package mcpTool
 
 import (
 	"context"
+	"errors"
+	"fmt"
 	"github.com/mark3labs/mcp-go/mcp"
 	"time"
 )
@@ -16,13 +18,26 @@ type CurrentTime struct {
 // 获取当前系统时间
 func (t *CurrentTime) Handle(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
 	// 获取当前系统时间
-	currentTime := time.Now().Format("2006-01-02 15:04:05")
+
+	timezone, ok := request.Params.Arguments["timezone"].(string)
+
+	if !ok {
+		return nil, errors.New("参数错误：timezone 必须是字符串类型")
+	}
+	// 根据timezone参数加载对应时区
+	loc, err := loadTimeZone(timezone)
+	if err != nil {
+		return nil, err
+	}
+
+	// 获取当前时间并转换为指定时区
+	currentTime := time.Now().In(loc).Format("2006-01-02 15:04:05")
 	//返回
 	return &mcp.CallToolResult{
 		Content: []mcp.Content{
 			mcp.TextContent{
 				Type: "text",
-				Text: currentTime,
+				Text: fmt.Sprintf("%s 时区的当前时间是：%s", timezone, currentTime),
 			},
 		},
 	}, nil
@@ -37,3 +52,30 @@ func (t *CurrentTime) New() mcp.Tool {
 			mcp.Enum("UTC", "CST", "PST", "EST", "GMT", "CET", "JST", "MST", "IST", "AST", "HST"),
 		))
 }
+
+// 将简写时区转换为IANA标准时区
+func loadTimeZone(timezone string) (*time.Location, error) {
+	// 时区映射表
+	timezoneMap := map[string]string{
+		"UTC": "UTC",
+		"CST": "Asia/Shanghai", // 中国标准时间
+		"PST": "America/Los_Angeles",
+		"EST": "America/New_York",
+		"GMT": "GMT",
+		"CET": "Europe/Paris",
+		"JST": "Asia/Tokyo",
+		"MST": "America/Denver",
+		"IST": "Asia/Kolkata",
+		"AST": "Asia/Riyadh", // 阿拉伯标准时间
+		"HST": "Pacific/Honolulu",
+	}
+
+	// 获取标准时区名称
+	tzName, exists := timezoneMap[timezone]
+	if !exists {
+		return nil, errors.New("不支持的时区: " + timezone)
+	}
+
+	// 加载时区
+	return time.LoadLocation(tzName)
+}

server/mcp/get_nickname.go

@@ -31,15 +31,15 @@ func (t *GetNickname) Handle(ctx context.Context, request mcp.CallToolRequest) (
 	// 1. 参数验证
 	username, ok := request.Params.Arguments["username"].(string)
 	if !ok {
-		return nil, errors.New("参数错误：username必须是字符串类型")
+		return nil, errors.New("参数错误：username 必须是字符串类型")
 	}
 
 	if username == "" {
-		return nil, errors.New("参数错误：username不能为空")
+		return nil, errors.New("参数错误：username 不能为空")
 	}
 
 	// 2. 记录操作日志
-	global.GVA_LOG.Info("getNickname工具被调用")
+	global.GVA_LOG.Info("getNickname 工具被调用")
 
 	// 3. 优化查询，只选择需要的字段
 	var user struct {
@@ -54,7 +54,14 @@ func (t *GetNickname) Handle(ctx context.Context, request mcp.CallToolRequest) (
 	// 4. 优化错误处理
 	if err != nil {
 		if errors.Is(err, gorm.ErrRecordNotFound) {
-			return nil, errors.New("用户不存在")
+			return &mcp.CallToolResult{
+				Content: []mcp.Content{
+					mcp.TextContent{
+						Type: "text",
+						Text: fmt.Sprintf("用户 %s 不存在", username),
+					},
+				},
+			}, nil
 		}
 		global.GVA_LOG.Error("数据库查询错误")
 		return nil, errors.New("系统错误，请稍后再试")