我有一个昂贵的函数,我将它应用于切片的所有项。我使用goroutine来处理这个问题,每个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)
}
有没有更好的(即更有效和/或更地道的)方法来做这件事?
使用同步。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)
}