incr.go•1.16 kB
package tools
import (
"context"
"fmt"
"github.com/dicedb/dicedb-go/wire"
"github.com/mark3labs/mcp-go/mcp"
"github.com/pottekkat/dicedb-mcp/internal/utils"
)
// NewIncrTool creates a new INCR tool for DiceDB
func NewIncrTool() mcp.Tool {
return mcp.NewTool("incr",
mcp.WithDescription("Increment the integer value of a key by one"),
utils.CommonURLParam(),
mcp.WithString("key",
mcp.Required(),
mcp.Description("The key whose value to increment"),
),
)
}
// HandleIncrTool handles the INCR tool request
func HandleIncrTool(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
key, ok := request.Params.Arguments["key"].(string)
if !ok || key == "" {
return nil, fmt.Errorf("missing or empty key parameter")
}
client, err := utils.GetClientFromRequest(request)
if err != nil {
return nil, err
}
resp := client.Fire(&wire.Command{
Cmd: "INCR",
Args: []string{key},
})
if resp.Err != "" {
return nil, fmt.Errorf("DiceDB error: %s", resp.Err)
}
value := utils.FormatDiceDBResponse(resp)
return mcp.NewToolResultText(fmt.Sprintf("Incremented key '%s', new value: %s", key, value)), nil
}