当所有 Goroutine 完成使用 WaitGroup 后，我们如何安全地关闭通道？

使用 WaitGroup 在 Goroutine 完成后关闭 Channel

使用通道和 Goroutine 时，在所有 Goroutine 完成后关闭通道是至关重要的。这可以防止通道上的进一步通信并允许优雅的资源管理。

提供的代码尝试实现此行为，但由于竞争条件而面临限制。为了解决这些挑战，我们可以利用sync.WaitGroup类型。

WaitGroup允许我们创建任务集合并等待它们完成后再继续执行。在关闭通道的上下文中，我们可以使用 WaitGroup 来确保仅当所有 goroutine 都使用完通道时才关闭通道。

使用 WaitGroup 的改进代码片段：

import (
    "sync"
)

// ...

var wg sync.WaitGroup
for i := 0; i <= 10; i++ {
    wg.Add(1)
    go func() {
        result := calculate()
        c <- result
        wg.Done()
    }()
}

// Close the channel when all goroutines are finished
go func() {
    wg.Wait()
    close(c)
}()

for result := range c {
    all_result = append(all_result, result...)
}

在这个更新的代码中，我们使用 WaitGroup 来跟踪未完成的 goroutine 的数量。每个执行任务的 goroutine 都会增加 WaitGroup 计数器，表明它已经启动。完成后，每个 goroutine 都会递减 WaitGroup 计数器，表明它已完成。一个单独的 goroutine 等待 WaitGroup，并在所有 goroutine 报告完成后关闭通道。

这种方法确保通道的同步且可靠的关闭，防止原始代码中遇到的问题。它还消除了睡眠呼叫和忙碌等待的需要，提供了更高效、更强大的解决方案。

以上是当所有 Goroutine 完成使用 WaitGroup 后，我们如何安全地关闭通道？的详细内容。更多信息请关注PHP中文网其他相关文章！