同时收听多个Go频道

时间:2019-03-07 05:26:16

标签: go channel goroutine

假设我要去接收频道。有什么办法可以让我一次听所有这些?例如:

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

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

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

1 个答案:

答案 0 :(得分:3)

您可以使用同步包创建一个等待组。 添加到waitGroup后,为每个通道启动一个goroutine。 最后,当通道完成时,它会减小waitGroup。调用方仅在等待组上等待。这是playground link

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()
}