等待 n 个 goroutine 终止

发布于05月16日

我需要开始大量的goroutines并等待它们的终止.直观的方法似乎是使用一个通道来等待所有这些操作完成:

package main

type Object struct {
    //data
}

func (obj *Object) Update(channel chan int) {
    //update data
    channel <- 1
    return
}

func main() {

    channel := make(chan int, n)
    list := make([]Object, n, m)
    for {
        for _, object := range list {
            go object.Update(channel)
        }
        for i := 0; i < n; i++ {
            <-channel
        }
        //now everything has been updated. start again
    }
}

但问题是，对象的数量以及goroutine的数量可能会发生变化.是否可以更改通道的缓冲区大小？

有没有更优雅的方式来做这件事呢？

推荐答案

我用WaitGroup来解决这个问题.翻译您当前的代码，并使用一些日志(log)来清楚地说明正在发生的事情:

package main

import "sync"
import "fmt"
import "time"

type Object struct {
    //data
}

func (obj *Object) Update(wg *sync.WaitGroup) {
    //update data
    time.Sleep(time.Second)
    fmt.Println("Update done")
    wg.Done()
    return
}

func main() {
    var wg sync.WaitGroup
    list := make([]Object, 5)
    for {
        for _, object := range list {
            wg.Add(1)
            go object.Update(&wg)
        }
        //now everything has been updated. start again
        wg.Wait()
        fmt.Println("Group done")
    }
}