如何将 WaitGroup 传递给顺序函数调用？

Question

我有一个可以在单独的 goroutine 中顺序或同时调用的函数。我想确保函数在主 goroutine 完成之前完全执行，所以我将 *sync.WaitGroup 参数传递给函数。现在，在某些地方要顺序调用该函数。我可以像这样将 nil waitGroup 传递给函数：func my_func(wg *sync.WaitGroup){   if wg != nil{   defer wg.Done()   }   // do the task}func main(){my_func(nil) // sequential callwg := sync.WaitGroup{}wg.Add(1)go my_func(&wg)  // concurrent callwg.Wait()}有没有更好的方法来实现这一目标？

Answer 1

你my_func()不应该知道/不应该关心它是如何执行的（无论是否在新的 goroutine 中）。所以就为了这个你不应该通过wg。不要强制并发或非并发使用你的 API，让你的包的用户决定他们希望如何调用它。

如果有人希望同时运行它，在一个新的 goroutine 中，wg可以这样处理my_func()：

wg.Add(1)

go func() {

    defer wg.Done()

    my_func()

}()

这也提供了在将在调用之前执行的函数调用之前/之后放置更多代码的可能性wg.Done()：

wg.Add(1)

go func() {

    defer wg.Done()

    // other code before

    my_func()

    // other code after

}()

还要注意，如果你在很多地方都有这个，你可以创建一个辅助函数来处理 goroutine 启动和等待组处理：

func launchMyFunc(wg *sync.WaitGroup) {

    go func() {

        defer wg.Done()

        my_func()

    }()

}

您还可以创建一个接受任意无参数不返回函数的助手：

func launchFunc(wg *sync.WaitGroup, f func()) {

    go func() {

        defer wg.Done()

        f()

    }()

}

使用上面的助手，你可以这样做：

wg.Add(1)

launchMyFunc(wg)

// or

wg.Add(1)

launchFunc(wg, my_func)