Support Go 1.23 iterators for the ZADD command (#652)

* Support Go 1.23 iterators for the ZADD * refactor: change to [string, float64]
redis · Oct 25, 2024 · 25821d0 · 25821d0
1 parent 1928749
commit 25821d0
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 (
+	"iter"
+	"strconv"
+)
 
 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[string, float64]) ZaddScoreMember {
+	for member, score := range seq {
+		c.cs.s = append(c.cs.s, strconv.FormatFloat(score, 'f', -1, 64), 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[string]float64{"1": float64(1)})).Build()
 }
 
 func TestIter(t *testing.T) {