pkg/SugarDB
1
0
Fork 0
mirror of https://github.com/EchoVault/SugarDB.git synced 2025-10-06 08:27:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: added RENAME command

Browse Source
This commit is contained in:
Sahil
2024-06-25 18:23:54 +05:30
parent 67ea63193b
commit 38ca643d05
3 changed files with 74 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
internal/modules/generic/commands.go
View File

@@ -581,6 +581,35 @@ func handleDecrBy(params internal.HandlerFuncParams) ([]byte, error) {
return []byte(fmt.Sprintf(":%d\r\n", newValue)), nil
}
func handleRename(params internal.HandlerFuncParams) ([]byte, error) {
if len(params.Command) != 3 {
return nil, errors.New("wrong number of arguments for RENAME")
}
oldKey := params.Command[1]
newKey := params.Command[2]
// Get the current value for the old key
values := params.GetValues(params.Context, []string{oldKey})
oldValue, ok := values[oldKey]
if !ok || oldValue == nil {
return nil, errors.New("no such key")
}
// Set the new key with the old value
if err := params.SetValues(params.Context, map[string]interface{}{newKey: oldValue}); err != nil {
return nil, err
}
// Delete the old key
if err := params.DeleteKey(oldKey); err != nil {
return nil, err
}
return []byte("+OK\r\n"), nil
}
func Commands() []internal.Command {
return []internal.Command{
{
@@ -778,8 +807,8 @@ This operation is limited to 64 bit signed integers.`,
Command: "incrby",
Module: constants.GenericModule,
Categories: []string{constants.KeyspaceCategory, constants.WriteCategory, constants.FastCategory},
Description: `(INCRBY key increment)
Increments the number stored at key by increment. If the key does not exist, it is set to 0 before performing the operation.
Description: `(INCRBY key increment)
Increments the number stored at key by increment. If the key does not exist, it is set to 0 before performing the operation.
An error is returned if the key contains a value of the wrong type or contains a string that can not be represented as integer.`,
Sync: true,
KeyExtractionFunc: incrByKeyFunc,
@@ -789,13 +818,22 @@ An error is returned if the key contains a value of the wrong type or contains a
Command: "decrby",
Module: constants.GenericModule,
Categories: []string{constants.KeyspaceCategory, constants.WriteCategory, constants.FastCategory},
Description: `(DECRBY key decrement)
The DECRBY command reduces the value stored at the specified key by the specified decrement.
If the key does not exist, it is initialized with a value of 0 before performing the operation.
Description: `(DECRBY key decrement)
The DECRBY command reduces the value stored at the specified key by the specified decrement.
If the key does not exist, it is initialized with a value of 0 before performing the operation.
If the key's value is not of the correct type or cannot be represented as an integer, an error is returned.`,
Sync: true,
KeyExtractionFunc: decrByKeyFunc,
HandlerFunc: handleDecrBy,
},
{
Command: "rename",
Module: constants.GenericModule,
Categories: []string{constants.KeyspaceCategory, constants.WriteCategory, constants.FastCategory},
Description: `(RENAME key newkey) Renames key to newkey. If newkey already exists, it is overwritten. If key does not exist, an error is returned.`,
Sync: true,
KeyExtractionFunc: renameKeyFunc,
HandlerFunc: handleRename,
},
}
}