main loop now uses map to check which handler to use

1 files changed, 24 insertions, 24 deletions

diff --git a/main.go b/main.go
index 38ccd34..d225a52 100644
--- a/main.go
+++ b/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"bufio"
+	"errors"
 	"fmt"
 	"net"
 	"os"
@@ -11,6 +12,23 @@ import (
 
 var db RedisDB
 
+type handlerFunc func(net.Conn, []string)
+
+var commandRegistry = map[string]handlerFunc{
+    "GET":      handleGet,
+    "SET":      handleSet,
+    "DEL":      handleDel,
+    "EXISTS":   handleExists,
+    "INCR":     handleIncr,
+    "DECR":     handleDecr,
+    "PING":     handlePing,
+    "FLUSHALL": handleFlushall,
+    "RPUSH":    handleRpush,
+    "LRANGE":   handleLrange,
+    "LPOP":     handleLpop,
+    "RPOP":     handleRpop,
+}
+
 func main() {
 	db = RedisDB{data: make(map[string]any)}
 
@@ -52,29 +70,11 @@ func handleConnection(conn net.Conn) {
 
 		command := strings.ToUpper(args[0])
 
-		switch {
-		case command == "GET" && len(args) == 2:
-			handleGet(conn, args)
-		case command == "SET" && len(args) == 3:
-			handleSet(conn, args)
-		case command == "EXISTS" && len(args) == 2:
-			handleExists(conn, args)
-		case command == "DEL" && len(args) == 2:
-			handleDel(conn, args)
-		case command == "INCR" && len(args) == 2:
-			handleIncr(conn, args)
-		case command == "DECR" && len(args) == 2:
-			handleDecr(conn, args)
-		case command == "PING" && len(args) == 1:
-			handlePing(conn, args)
-		case command == "FLUSHALL" && len(args) == 1:
-			handleFlushall(conn, args)
-		case command == "RPUSH" && len(args) >= 3:
-            handleRpush(conn, args)
-		case command == "LRANGE" && len(args) == 4:
-            handleLrange(conn, args)
-		default:
-			conn.Write([]byte("-ERR unknown command or wrong arguments\r\n"))
-		}
+		if handler, ok := commandRegistry[command]; ok {
+            handler(conn, args)
+        } else {
+            conn.Write(serializeRESP(errors.New("unknown command")))
+        }
+		
 	}
 }
\ No newline at end of file