Return keys from commands

command.go

@@ -119,6 +119,35 @@ func cmdString(cmd Cmder, val interface{}) string {
 	return util.BytesToString(b)
 }
 
+// A wrapper around extract keys, just to make it easier to test.
+func GetKeys(cmd Cmder) []string {
+	return extractKeys(cmd)
+}
+
+func extractKeys(cmd Cmder) []string {
+	firstKeyPos := cmdFirstKeyPos(cmd)
+	if firstKeyPos == -1 {
+		return nil
+	}
+
+	args := cmd.Args()
+	keys := []string{}
+
+	switch cmd.Name() {
+	case "mget":
+		for i := int(firstKeyPos); i < len(args); i++ {
+			keys = append(keys, cmd.stringArg(i))
+		}
+
+	default:
+		if int(firstKeyPos) < len(args) {
+			keys = append(keys, cmd.stringArg(firstKeyPos))
+		}
+	}
+
+	return keys
+}
+
 //------------------------------------------------------------------------------
 
 type baseCmd struct {

commands_test.go

@@ -7279,6 +7279,29 @@ var _ = Describe("Commands", func() {
 	})
 })
 
+var _ = Describe("Get Keys from Command", func() {
+	var client *redis.Client
+
+	BeforeEach(func() {
+		client = redis.NewClient(redisOptions())
+		Expect(client.FlushDB(ctx).Err()).NotTo(HaveOccurred())
+	})
+
+	AfterEach(func() {
+		Expect(client.Close()).NotTo(HaveOccurred())
+	})
+
+	It("returns the keys extracted from the command", func() {
+		cmd := redis.NewCmd(ctx, "SET", "key1", "value1")
+		keys := redis.GetKeys(cmd)
+		Expect(keys).To(Equal([]string{"key1"}))
+
+		cmd = redis.NewCmd(ctx, "MGET", "key1", "key2", "key3")
+		keys = redis.GetKeys(cmd)
+		Expect(keys).To(Equal([]string{"key1", "key2", "key3"}))
+	})
+})
+
 type numberStruct struct {
 	Number int
 }

go.mod

@@ -10,6 +10,6 @@ require (
 )
 
 retract (
-	v9.5.3 // This version was accidentally released. Please use version 9.6.0 instead.
 	v9.5.4 // This version was accidentally released. Please use version 9.6.0 instead.
+	v9.5.3 // This version was accidentally released. Please use version 9.6.0 instead.
 )