From f18210c30e67de4bede7c6080d85629f7086676c Mon Sep 17 00:00:00 2001 From: alex Date: Fri, 10 Jul 2026 20:28:47 +0200 Subject: separated commands into their own files TODO: refactor all commands to work like this --- pkg/commands/command-registry.go | 58 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 58 insertions(+) create mode 100644 pkg/commands/command-registry.go (limited to 'pkg/commands/command-registry.go') diff --git a/pkg/commands/command-registry.go b/pkg/commands/command-registry.go new file mode 100644 index 0000000..0193da1 --- /dev/null +++ b/pkg/commands/command-registry.go @@ -0,0 +1,58 @@ +package commands + +import ( + "errors" + "redisClone/pkg/core" +) +func SingleKey(args []string) []string { + return []string{args[1]} +} + +func AllSubsequentKeys(args []string) []string { + return args[1:] +} + +func NoKeys(args []string) []string { + return nil +} + +type CommandDef struct{ + MinArgs int + ExtractKeys func(args []string)[]string + Execute func(args []string, getShard func(k string) *core.Shard) interface{} +} + +func DispatchBaseCommand(db core.RedisDB ,commandName string, args []string) interface{} { + cmd, exists := Registry[commandName] + if !exists { + return nil + } + + if len(args) < cmd.MinArgs { + return errors.New("wrong number of arguments") + } + + keys := cmd.ExtractKeys(args) + + db.Lock(keys) + defer db.Unlock(keys) + return cmd.Execute(args, db.GetShard) +} + +var Registry = map[string]CommandDef{ + "GET": { + MinArgs: 2, + ExtractKeys: SingleKey, + Execute: Get, + }, + "SET": { + MinArgs: 3, + ExtractKeys: SingleKey, + Execute: Set, + }, + "DEL": { + MinArgs: 2, + ExtractKeys: AllSubsequentKeys, + Execute: Del, + }, +} \ No newline at end of file -- cgit v1.2.3