go - 使用sync.WaitGroup.wait时如何实现超时？

Question

这个问题在这里已经有了答案:

Timeout for WaitGroup.Wait() (10 个回答)

关闭 4 个月前。

我遇到了一种情况，我想跟踪一些 goroutine 以在特定点同步，例如当所有 url 都被获取时。然后，我们可以将它们全部放置并按特定顺序显示。

我认为这是 barrier 进来的。它在 go 和 sync.WaitGroup 中。但是，在实际情况下，我们不能确保所有的 fetch 操作都会在短时间内成功。所以，我想在 wait 获取操作时引入超时。

我是 Golang 的新手，谁能给我一些建议？

---

我要找的是这样的:

   wg := &sync.WaigGroup{}
   select {
   case <-wg.Wait():
   // All done!
   case <-time.After(500 * time.Millisecond):
   // Hit timeout.
   }

我知道 Wait 不支持 Channel。

Answer 1

如果您想要的只是整洁的选择，您可以通过生成一个调用方法并在完成后关闭/发送 channel 的例程来轻松地将阻塞函数转换为 channel 。

done := make(chan struct{})
go func() {
   wg.Wait()
   close(done)
}()

select {
case <-done:
// All done!
case <-time.After(500 * time.Millisecond):
// Hit timeout.
}