diff options
Diffstat (limited to 'handlers.go')
| -rw-r--r-- | handlers.go | 27 |
1 files changed, 15 insertions, 12 deletions
diff --git a/handlers.go b/handlers.go index 1a04db9..980e805 100644 --- a/handlers.go +++ b/handlers.go @@ -74,28 +74,31 @@ func handleSet(conn net.Conn, args []string) { conn.Write(serializeRESP(result)) } -/* + func handleDel(conn net.Conn, args []string) { - db.mu.Lock() - defer db.mu.Unlock() if len(args) < 2 { err := errors.New("wrong number of arguments for 'DEL'") conn.Write(serializeRESP(err)) return } - - deletedKeys := 0 - for i := 1; i<len(args);i++{ - _, exists := db.data[args[i]] - if exists { - delete(db.data, args[i]) - deletedKeys += 1 - } - } + keys := args[:1] + deletedKeys := db.ExecuteMulti(keys, func(shards []*Shard) interface{}{ + count := 0 + for _, key := range keys{ + shard := db.getShard(key) + _, exists := shard.data[key] + if exists { + delete(shard.data, key) + count += 1 + } + } + return count + }) conn.Write(serializeRESP(deletedKeys)) } +/* func handleIncr(conn net.Conn, args []string) { db.mu.Lock() defer db.mu.Unlock() |
