80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
|
package mcpTool
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/flipped-aurora/gin-vue-admin/server/global"
|
|||
|
|
"github.com/flipped-aurora/gin-vue-admin/server/model/system"
|
|||
|
|
"github.com/mark3labs/mcp-go/mcp"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
RegisterTool(&GetNickname{})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type GetNickname struct{}
|
|||
|
|
|
|||
|
|
// 根据用户username获取nickname
|
|||
|
|
func (t *GetNickname) New() mcp.Tool {
|
|||
|
|
return mcp.NewTool("getNickname",
|
|||
|
|
mcp.WithDescription("根据用户username获取nickname"),
|
|||
|
|
mcp.WithString("username",
|
|||
|
|
mcp.Required(),
|
|||
|
|
mcp.Description("用户的username"),
|
|||
|
|
))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Handle 处理获取昵称的请求
|
|||
|
|
func (t *GetNickname) Handle(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
|||
|
|
// 1. 参数验证
|
|||
|
|
username, ok := request.GetArguments()["username"].(string)
|
|||
|
|
if !ok {
|
|||
|
|
return nil, errors.New("参数错误：username 必须是字符串类型")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if username == "" {
|
|||
|
|
return nil, errors.New("参数错误：username 不能为空")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 记录操作日志
|
|||
|
|
global.GVA_LOG.Info("getNickname 工具被调用")
|
|||
|
|
|
|||
|
|
// 3. 优化查询，只选择需要的字段
|
|||
|
|
var user struct {
|
|||
|
|
NickName string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err := global.GVA_DB.Model(&system.SysUser{}).
|
|||
|
|
Select("nick_name").
|
|||
|
|
Where("username = ?", username).
|
|||
|
|
First(&user).Error
|
|||
|
|
|
|||
|
|
// 4. 优化错误处理
|
|||
|
|
if err != nil {
|
|||
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|||
|
|
return &mcp.CallToolResult{
|
|||
|
|
Content: []mcp.Content{
|
|||
|
|
mcp.TextContent{
|
|||
|
|
Type: "text",
|
|||
|
|
Text: fmt.Sprintf("用户 %s 不存在", username),
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
global.GVA_LOG.Error("数据库查询错误")
|
|||
|
|
return nil, errors.New("系统错误，请稍后再试")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构造回复信息
|
|||
|
|
return &mcp.CallToolResult{
|
|||
|
|
Content: []mcp.Content{
|
|||
|
|
mcp.TextContent{
|
|||
|
|
Type: "text",
|
|||
|
|
Text: fmt.Sprintf("用户 %s 的昵称是 %s", username, user.NickName),
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
}, nil
|
|||
|
|
}
|