使用缓冲通道作为信号量时如何确保所有 Goroutine 完成？

使用WaitGroup等待缓冲通道为空

在并发编程中，通常需要控制并发执行的goroutines的数量。一种常见的方法是使用缓冲通道作为信号量。但是，请务必注意此方法的局限性。

考虑以下示例：

ints := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
 sem := make(chan struct{}, 2)
for _, i := range ints {
  sem<- struct{}{}
  go func(id int, sem chan struct{}) {
    // do something
    <-sem
  }(i, sem)
}

此代码使用大小为 2 的缓冲通道 (sem) 来限制并发数量goroutine 变为 2。但是，有一个问题：程序可能会在最后几个 goroutine 完成其任务之前结束。这是因为缓冲通道在处理过程中的任何时候都可能变空，即使程序正在调度更多的 goroutine。

要等待缓冲通道耗尽，我们不能依赖通道本身。相反，我们可以使用sync.WaitGroup来跟踪未完成的goroutines的数量：

sem := make(chan struct{}, 2)
var wg sync.WaitGroup
for _, i := range ints {
  wg.Add(1)
  sem<- struct{}{}
  go func(id int) {
    defer wg.Done()
    // do something
    <-sem
  }(i)
}
wg.Wait()

在这个修改后的代码中，我们在分派每个goroutine之前使用wg.Add(1)来增加等待组。当 goroutine 完成其任务时，defer wg.Done() 语句会递减等待组。最后，wg.Wait() 会阻塞，直到等待组达到零，以确保所有 goroutine 都已完成执行。这种方法允许我们等待通道耗尽并确保程序不会过早终止。

以上是使用缓冲通道作为信号量时如何确保所有 Goroutine 完成？的详细内容。更多信息请关注PHP中文网其他相关文章！