diff options
| -rw-r--r-- | pkg/commands/command-registry.go | 10 | ||||
| -rw-r--r-- | pkg/commands/lpush.go | 26 | ||||
| -rw-r--r-- | pkg/commands/rpush.go | 26 |
3 files changed, 62 insertions, 0 deletions
diff --git a/pkg/commands/command-registry.go b/pkg/commands/command-registry.go index 7db4923..6e34ff7 100644 --- a/pkg/commands/command-registry.go +++ b/pkg/commands/command-registry.go @@ -75,6 +75,16 @@ var Registry = map[string]CommandDef{ ExtractKeys: SingleKey, Execute: Rename, }, + "RPUSH": { + MinArgs: 3, + ExtractKeys: SingleKey, + Execute: Rpush, + }, + "LPUSH": { + MinArgs: 3, + ExtractKeys: SingleKey, + Execute: Lpush, + }, "RPOP": { MinArgs: 2, ExtractKeys: SingleKey, diff --git a/pkg/commands/lpush.go b/pkg/commands/lpush.go new file mode 100644 index 0000000..dcd67ee --- /dev/null +++ b/pkg/commands/lpush.go @@ -0,0 +1,26 @@ +package commands + +import ( + "errors" + "redisClone/pkg/core" +) + +func Lpush(args []string, getShard func(k string) *core.Shard) interface{}{ + key := args[1] + newItems := args[2:] + s := getShard(key) + var list []string + existingItem, exists := s.Data[key] + if exists { + var ok bool + list, ok = existingItem.Value.([]string) + if !ok { + return errors.New("WRONGTYPE Operation against a key holding the wrong kind of value") + } + } else { + list = []string{} + } + list = append(newItems, list...) + s.Data[key] = core.Item{Value: list} + return len(list) +}
\ No newline at end of file diff --git a/pkg/commands/rpush.go b/pkg/commands/rpush.go new file mode 100644 index 0000000..ea7e691 --- /dev/null +++ b/pkg/commands/rpush.go @@ -0,0 +1,26 @@ +package commands + +import ( + "errors" + "redisClone/pkg/core" +) + +func Rpush(args []string, getShard func(k string) *core.Shard) interface{}{ + key := args[1] + newItems := args[2:] + s := getShard(key) + var list []string + existingItem, exists := s.Data[key] + if exists { + var ok bool + list, ok = existingItem.Value.([]string) + if !ok { + return errors.New("WRONGTYPE Operation against a key holding the wrong kind of value") + } + } else { + list = []string{} + } + list = append(list, newItems...) + s.Data[key] = core.Item{Value: list} + return len(list) +}
\ No newline at end of file |
