From 89b3019a3ab77a3e72720342fc0e5bea996fa4f2 Mon Sep 17 00:00:00 2001 From: alex Date: Sun, 12 Jul 2026 13:48:32 +0200 Subject: refactored all functions to use the command registry --- pkg/commands/lrem.go | 75 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 75 insertions(+) create mode 100644 pkg/commands/lrem.go (limited to 'pkg/commands/lrem.go') diff --git a/pkg/commands/lrem.go b/pkg/commands/lrem.go new file mode 100644 index 0000000..38be270 --- /dev/null +++ b/pkg/commands/lrem.go @@ -0,0 +1,75 @@ +package commands + +import ( + "errors" + "redisClone/pkg/core" + "strconv" +) + +func Lrem(args []string, getShard func(k string) *core.Shard) interface{} { + key := args[1] + count, err := strconv.Atoi(args[2]) + toBeRemoved := args[3] + if err != nil { + return errors.New("value is not an integer") + } + + s := getShard(key) + item, exists := s.Data[key] + if !exists { + return 0 + } + list, ok := item.Value.([]string) + if !ok { + return errors.New("WRONGTYPE Operation against a key holding the wrong kind of value") + } + + counter := 0 + absCount := count + if count < 0 { + absCount = -count + } + + var newList []string + if count == 0 { + newList = []string{} + for _, value := range list { + if value == toBeRemoved { + counter++ + } else { + newList = append(newList, value) + } + } + } else if count > 0 { + newList = []string{} + for _, value := range list { + if value == toBeRemoved && counter < absCount { + counter++ + } else { + newList = append(newList, value) + } + } + } else { // count < 0 + reversedNewList := []string{} + for i := len(list) - 1; i >= 0; i-- { + value := list[i] + if value == toBeRemoved && counter < absCount { + counter++ + } else { + reversedNewList = append(reversedNewList, value) + } + } + newList = make([]string, len(reversedNewList)) + for i, v := range reversedNewList { + newList[len(reversedNewList)-1-i] = v + } + } + + if len(newList) == 0 { + delete(s.Data, key) + } else { + item.Value = newList + s.Data[key] = item + } + return counter +} -- cgit v1.2.3