aboutsummaryrefslogtreecommitdiff
path: root/handlers.go
diff options
context:
space:
mode:
Diffstat (limited to 'handlers.go')
-rw-r--r--handlers.go91
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