sync.WaitGroup 是 Go 语言中的一个并发原语,用于等待一组 Goroutine 完成执行。它通过计数器实现,可以动态地添加和减少计数器的值。
下面是使用 WaitGroup 的基本示例:
package main
import (
"fmt"
"sync"
)
func worker(id int, wg *sync.WaitGroup) {
fmt.Printf("Worker %d starting\n", id)
defer wg.Done() // 减少计数器的值
// 模拟耗时任务
for i := 0; i < 5; i++ {
fmt.Printf("Worker %d working...\n", id)
}
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
// 添加要等待的 Goroutine 数量
wg.Add(2)
go worker(1, &wg)
go worker(2, &wg)
// 等待所有 Goroutine 完成
wg.Wait()
fmt.Println("All workers done")
}
在上面的例子中,我们创建了两个 Goroutine(worker),它们会模拟执行一些工作。通过 WaitGroup 控制计数器的增加和减少,确保主函数在所有工作完成后才退出。
内容由零声教学AI助手提供,问题来源于学员提问




