diff options
Diffstat (limited to 'handlers.go')
| -rw-r--r-- | handlers.go | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/handlers.go b/handlers.go index 244d332..a6ae6ef 100644 --- a/handlers.go +++ b/handlers.go @@ -396,4 +396,95 @@ func handleExpire(conn net.Conn, args []string){ item.ExpiresAt = &expiry db.data[key] = item conn.Write(serializeRESP(1)) +} + +func handleHset(conn net.Conn, args []string){ + + if len(args) < 4 || (len(args)-2)%2 != 0 { + err := errors.New("wrong number of arguments for 'HSET'") + conn.Write(serializeRESP(err)) + return + } + key := args[1] + db.mu.Lock() + defer db.mu.Unlock() + + item, exists := db.data[key] + + var hash map[string]string + + if exists { + var ok bool + hash, ok = item.Value.(map[string]string) + if !ok { + conn.Write(serializeRESP(errors.New("WRONGTYPE Operation against a key holding the wrong kind of value"))) + return + } + } else { + hash = make(map[string]string) + } + count := 0 + for i := 2; i < len(args); i += 2 { + hash[args[i]] = args[i+1] + count += 1 + } + db.data[key] = Item{Value: hash} + conn.Write(serializeRESP(count)) +} + +func handleHget(conn net.Conn, args []string) { + if len(args) != 3 { + conn.Write(serializeRESP(errors.New("ERR wrong number of arguments for 'HGET'"))) + return + } + + key := args[1] + field := args[2] + + db.mu.Lock() + defer db.mu.Unlock() + + item, exists := db.data[key] + if !exists { + conn.Write(serializeRESP(nil)) + return + } + + hash, ok := item.Value.(map[string]string) + if !ok { + conn.Write(serializeRESP(errors.New("WRONGTYPE Operation against a key holding the wrong kind of value"))) + return + } + + val, found := hash[field] + if !found { + conn.Write(serializeRESP(nil)) + return + } + + conn.Write(serializeRESP(val)) +} + +func handleHgetall(conn net.Conn, args []string){ + if len(args) != 2 { + conn.Write(serializeRESP(errors.New("ERR wrong number of arguments for 'HGETALL'"))) + return + } + key := args[1] + + db.mu.Lock() + defer db.mu.Unlock() + + item, exists := db.data[key] + if !exists { + conn.Write(serializeRESP(nil)) + return + } + + hash, ok := item.Value.(map[string]string) + if !ok { + conn.Write(serializeRESP(errors.New("WRONGTYPE Operation against a key holding the wrong kind of value"))) + return + } + conn.Write(serializeRESP(hash)) }
\ No newline at end of file |
