通道和等待组进入死锁

Question

我在整理 go 例程并让它们与主要 go 例程上的频道进行通信时遇到问题。为简化起见，我的代码如下所示：

func main() {
    channel := make(chan string)
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go performTest(channel, &wg, i)
    }

    wg.Wait()
    close(channel)

    for line := range channel {
        fmt.Print(line)
    }
}

func performTest(channel chan string, wg *sync.WaitGroup, i int) {
     defer wg.Done()
     // perform some work here
     result := fmt.sprintf("Pretend result %d", i)
     channel <- result
}

这似乎进入了某种僵局，但我不明白为什么。它卡在wg.Wait() 上，尽管我希望它会在所有 goroutine 都在等待组上调用 Done 后继续。我在这里想念什么？我想等待 goroutines，然后遍历通道中的所有结果。

Answer 1

您可以等待组并在单独的 go 例程中关闭频道。如果通道关闭，您在通道上的范围将在收到最后一个发送的值后结束。

如果您只是等待，则不会从频道收到任何内容。由于通道没有缓冲，performTest goroutines 将无法发送。对于无缓冲通道，发送操作将阻塞，直到它被接收。因此，延迟的wg.Done 调用将永远不会发生，您的程序将陷入僵局。因为Done 仅在执行永久阻塞发送之后才被调用。

func main() {
    channel := make(chan string)
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go performTest(channel, &wg, i)
    }
    
    // this is the trick
    go func() {
        wg.Wait()
        close(channel)
    }()

    for line := range channel {
        fmt.Print(line)
    }
}

func performTest(channel chan string, wg *sync.WaitGroup, i int) {
    defer wg.Done()
    // perform some work here
    result := fmt.Sprintf("Pretend result %d\n", i)
    channel <- result
}

https://play.golang.com/p/5pACJzwL4Hi