如何等待多个 goroutine 完成？

Question

package main

var fooRunning = false
var barRunning = false

func foo() {
    fooRunning = true
    defer func() { fooRunning = false }()
    if barRunning {
        // wait for bar() to finish
    }
    ...
}

func bar() {
    barRunning = true
    defer func() { barRunning = false }()
    if fooRunning {
        // wait for foo() to finish
    }
    ...
}

在我的情况下，如果我们运行go foo()，它应该等待bar() 完成，反之亦然。最好的方法是什么？请注意，它们也可以独立执行。

Answer 1

任何具体的设计都无法安全地满足您的要求。按照规定，您说foo 和bar 可以 在并发 goroutine 中运行，并且如果其中一个或两个都已启动，则另一个应等待它们都完成。但是，这太弱了。如果foo 启动然后结束，但bar 还没有开始运行，会发生什么？如果bar 根本不运行怎么办？或者如果bar 运行，但foo 从来没有运行呢？

您是否要求foo 和bar 都必须启动和完成才能使您的程序正确？如果是这样，我可以猜到你的意思是：你想要一个 barrier 等待他们都完成后再继续。

package main

import (
    "fmt"
    "sync"
    "time"
)

func foo() {
    fmt.Println("foo")
}

func bar() {
    fmt.Println("bar")
}

func within(wg *sync.WaitGroup, f func()) {
    wg.Add(1)
    go func() {
        defer wg.Done()
        f()
    }()
}

func main() {
    var wg sync.WaitGroup
    within(&wg, foo)
    within(&wg, bar)
    wg.Wait()
    fmt.Println("Both foo and bar completed.")
}

(That same example in the Playground)

注意这里，foo 和 bar 都不是相互感知的；为了协调这两个呼叫，只有他们的呼叫者是。

您最初的尝试可能会导致您走上使foo 和bar 各自关闭或接受sync.WaitGroup 作为参数的道路，每个函数在退出之前首先是adding itself to the group 和waiting on it。那就是疯狂。

如果foo 在bar 有机会将自身添加到WaitGroup 之前启动并完成，则foo 将在bar 之前退出，即使您可以声称它们已经同时运行，或者反过来在foo 之前运行bar 可以注册其活动状态。同样，由于这是您程序的一个未明确说明的方面，我建议您转而关注更高级别的障碍，而不是这两个函数的相互依赖。

Answer 2

您可以使用频道！正如我从生锈的过程中所记得的那样，这会给：

func foo() {
    c := make(chan int)
    go bar(c)
    <-c
}

在酒吧里

func bar(c chan int) {
    // do stuff here
    c <- 0
}