antlabs/easychan

对于chan使用场景，总结与收集好用的工具函数

easychan

各种chan操作的函数, 全部实现为泛型，go >= 1.18

read

// 无阻塞读
c := make(chan bool)
v, ok := easychan.AsyncRead(c)

// 阻塞带context读
ctx, cancel := context.WithCancel(context.TODO())
easychan.ReadContext(ctx, c)

write

// 无阻塞写
c := make(chan bool)
easychan.AsyncWrite(c)

// 阻塞带context读
ctx, cancel := context.WithCancel(context.TODO())

easychan.WriteContext(ctx, c, true)

Tee函数，一拖二(对原chan镜像两份数据流)

使用者只管往一个chan里面写数据，从easychan里面生成的两个chan可以得到同样的数据 伪代码如下

  // 这里支持任意类型chan
 	voice := make(chan []byte)
  var wg sync.WaitGroup
	out1, out2 := easychan.Tee(context.Background(), voice)

	wg.Add(2)
	defer wg.Wait()

	go func() {
    // 模拟生产者发音频
		for k := range [100]int{} {
			voice <- []byte{k}
		}

		close(voice)

	}()

	go func() {
		defer wg.Done()
		for {
			select {
			case d, ok := <-out1:
				if !ok {
					return
				}
				fmt.Printf("out1 :%d\n", d)
			}
		}
	}()

	go func() {
		defer wg.Done()
		for {
			select {
			case d, ok := <-out2:
				if !ok {
					return
				}
				fmt.Printf("out2 :%d\n", d)
			}
		}
	}()