bool-applied

DiceDB · Oct 6, 2024 · dffb232 · dffb232
1 parent ec40c24
commit dffb232
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/pkg/util/blacklist.go b/pkg/util/blacklist.go
@@ -5,18 +5,17 @@ import (
 	"strings"
 )
 
-var blacklistedCommands = []string{
-	"FLUSHALL", "FLUSHDB", "DUMP", "ABORT", "AUTH", "CONFIG", "SAVE", "BGSAVE",
-	"BGREWRITEAOF", "RESTORE", "MULTI", "EXEC", "DISCARD", "QWATCH", "QUNWATCH",
-	"LATENCY", "CLIENT", "SLEEP", "PERSIST",
+var blocklistedCommands = map[string]bool{
+	"FLUSHALL": true, "FLUSHDB": true, "DUMP": true, "ABORT": true, "AUTH": true,
+	"CONFIG": true, "SAVE": true, "BGSAVE": true, "BGREWRITEAOF": true, "RESTORE": true,
+	"MULTI": true, "EXEC": true, "DISCARD": true, "QWATCH": true, "QUNWATCH": true,
+	"LATENCY": true, "CLIENT": true, "SLEEP": true, "PERSIST": true,
 }
 
 // BlockListedCommand checks if a command is blocklisted
 func BlockListedCommand(cmd string) error {
-	for _, blacklistedCmd := range blacklistedCommands {
-		if strings.ToUpper(cmd) == blacklistedCmd {
-			return errors.New("command is blocklisted")
-		}
+	if blocklistedCommands[strings.ToUpper(cmd)] {
+		return errors.New("command is blocklisted")
 	}
 	return nil
 }