Support Go 1.23 iterators for the ZADD

redis · Oct 25, 2024 · 76fe792 · 76fe792
1 parent 1aee272
commit 76fe792
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/internal/cmds/iter.go b/internal/cmds/iter.go
@@ -2,7 +2,10 @@
 
 package cmds
 
-import "iter"
+import (
+	"fmt"
+	"iter"
+)
 
 func (c HmsetFieldValue) FieldValueIter(seq iter.Seq2[string, string]) HmsetFieldValue {
 	for field, value := range seq {
@@ -24,3 +27,10 @@ func (c XaddFieldValue) FieldValueIter(seq iter.Seq2[string, string]) XaddFieldV
 	}
 	return c
 }
+
+func (c ZaddScoreMember) ScoreMemberIter(seq iter.Seq2[float64, string]) ZaddScoreMember {
+	for score, member := range seq {
+		c.cs.s = append(c.cs.s, fmt.Sprintf("%f", score), member)
+	}
+	return c
+}
diff --git a/internal/cmds/iter_test.go b/internal/cmds/iter_test.go
@@ -11,6 +11,7 @@ func iter0(s Builder) {
 	s.Hmset().Key("1").FieldValue().FieldValueIter(maps.All(map[string]string{"1": "1"})).Build()
 	s.Hset().Key("1").FieldValue().FieldValueIter(maps.All(map[string]string{"1": "1"})).Build()
 	s.Xadd().Key("1").Id("*").FieldValue().FieldValueIter(maps.All(map[string]string{"1": "1"})).Build()
+	s.Zadd().Key("1").ScoreMember().ScoreMemberIter(maps.All(map[float64]string{float64(1): "1"})).Build()
 }
 
 func TestIter(t *testing.T) {