feat: add Repeate function to generate a chan of repeate values

concurrency/channel.go

@@ -14,14 +14,14 @@ func NewChannel() *Channel {
 	return &Channel{}
 }
 
-// Generate a data of type any chan
+// Generate a data of type any chan, put param `values` into the chan
-func (c *Channel) Generate(done <-chan any, datas ...any) <-chan any {
+func (c *Channel) Generate(done <-chan any, values ...any) <-chan any {
 	dataStream := make(chan any)
 
 	go func() {
 		defer close(dataStream)
 
-		for _, v := range datas {
+		for _, v := range values {
 			select {
 			case <-done:
 				return
@@ -32,3 +32,23 @@ func (c *Channel) Generate(done <-chan any, datas ...any) <-chan any {
 
 	return dataStream
 }
+
+// Repeat return a data of type any chan, put param `values` into the chan repeatly,
+// until close the `done` chan
+func (c *Channel) Repeat(done <-chan any, values ...any) <-chan any {
+	dataStream := make(chan any)
+
+	go func() {
+		defer close(dataStream)
+		for {
+			for _, v := range values {
+				select {
+				case <-done:
+					return
+				case dataStream <- v:
+				}
+			}
+		}
+	}()
+	return dataStream
+}

concurrency/channel_test.go

@@ -7,7 +7,7 @@ import (
 )
 
 func TestGenerate(t *testing.T) {
-	assert := internal.NewAssert(t, "TestToChar")
+	assert := internal.NewAssert(t, "TestGenerate")
 
 	done := make(chan any)
 	defer close(done)
@@ -22,3 +22,20 @@ func TestGenerate(t *testing.T) {
 	assert.Equal(2, <-intStream)
 	assert.Equal(3, <-intStream)
 }
+
+func TestRepeat(t *testing.T) {
+	// assert := internal.NewAssert(t, "TestRepeat")
+
+	done := make(chan any)
+	defer close(done)
+
+	c := NewChannel()
+	intStream := c.Repeat(done, 1, 2)
+
+	for v := range intStream {
+		t.Log(v)
+	}
+	// assert.Equal(1, <-intStream)
+	// assert.Equal(2, <-intStream)
+	// assert.Equal(3, <-intStream)
+}