同时收听多个 go 频道

Question

假设我有一个 go 接收通道。有没有一种方法可以让我同时收听所有内容？例如：

channels := make([]<-chan int, 0, N)
// fill the slice with channels
for _, channel := range channels {
  <-channel
}

这是我能做到的最接近的事情。但是，此实现取决于切片元素的顺序。

为了清楚起见，我不需要知道 go 通道的值。我只需要知道他们都完成了。

Answer 1

您可以使用同步包来创建等待组。添加到 waitGroup 后，为每个通道启动一个 goroutine。最后，当通道完成时，它会递减 waitGroup。呼叫者只是在等待组中等待。这是游乐场链接。

package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    wg       := sync.WaitGroup{}
    numChans := 10
    chans    := make([]chan interface{}, numChans)

    for i:=0;i<numChans;i++ {
      chans[i] = make(chan interface{})
      wg.Add(1) 
      go func(c chan interface{}) {
        defer wg.Done()     
        <-c 
      }(chans[i])
    }

    //simulate closing channels
    go func(){
       time.Sleep(time.Second)
       fmt.Println("closing")
        for _, ch := range chans{
            close(ch)
        }       
    }()

    fmt.Println("waiting")
    wg.Wait()
}