Infinite parameter for the Repeat iterator

Author: mowshon

repeat.go

@@ -5,10 +5,22 @@ func Repeat[T any](value T, n int) Iter[T] {
 	// Initialisation of a new channel.
 	iter := Instance[T](int64(n), false)
 
+	// If the length is below zero, then
+	// the iterator will run forever.
+	if n < 0 {
+		iter.SetInfinite(true)
+	}
+
 	go func() {
 		defer iterRecover()
 		defer iter.Close()
 
+		if iter.IsInfinite() {
+			for {
+				iter.Chan() <- value
+			}
+		}
+
 		for step := 0; step < n; step++ {
 			iter.Chan() <- value
 		}

repeat_test.go

@@ -12,3 +12,17 @@ func TestRepeat(t *testing.T) {
 		assert.Exactly(t, []string{"A", "A", "A"}, slice)
 	}
 }
+
+func TestRepeatInfinite(t *testing.T) {
+	repeat := Repeat("A", -1)
+	assert.Exactly(t, true, repeat.IsInfinite())
+
+	var data []string
+	for i := 0; i <= 3; i++ {
+		if next, err := repeat.Next(); assert.Nil(t, err) {
+			data = append(data, next)
+		}
+	}
+
+	assert.Exactly(t, []string{"A", "A", "A", "A"}, data)
+}