Question

我有一个昂贵的功能，我应用于切片的所有项目。我正在使用goroutines来处理这个问题，每个goroutine处理一个切片项目。

func Huge(lst []foo) {
  for _, item := range lst {
     go performSlow(item)
  }

  // How do I synchronize here ?
  return someValue(lst)
}

问题是，如评论中所示，在调用someValue函数之前，等待所有goroutine完成工作的首选方法是什么？将频道传递给performSlow并等到所有人都写完就可以了，但这似乎有点过分了：

func Huge(lst []foo) {
  ch := make(chan bool)

  for _, item := range lst {
     go performSlow(item, ch)  // performSlow does its job, then writes a dummy value to ch
  }

  for i := range lst {
     _ = <-ch
  }

  return someValue(lst)
}

是否有更好的（即更高效和/或更惯用）方式？

Answer 1

使用sync.WaitGroup（http://godoc.org/sync#WaitGroup）

func Huge(lst []foo) {
  var wg sync.WaitGroup
  for _, item := range lst {
     wg.Add(1)
     go func() {
         performSlow(item)
         wg.Done()
     }()
  }

  wg.Wait()
  return someValue(lst)
}

什么是同步goroutines的首选方式

1 个答案: