added hash support

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