sync.WaitGroup-Go服务器开发

sync.WaitGroup 是 Go 语言标准库 sync 包中的一个并发原语，用于等待一组 Goroutine 完成执行。它通过计数器机制实现协程的同步，常用于主 Goroutine 阻塞等待所有子 Goroutine 完成任务后再继续执行。

核心方法

Add(delta int)
- 增加或减少 WaitGroup 的计数器值（通常为正数）。
- 在启动新 Goroutine 前调用，表示需要等待的任务数。
- delta 可以是负数（但需确保计数器不为负）。
Done()
- 将计数器减 1（等价于 Add(-1)）。
- 在 Goroutine 内部 defer 调用，确保任务完成后通知 WaitGroup。
Wait()
- 阻塞当前 Goroutine，直到计数器归零。
- 通常由主 Goroutine 调用。

使用示例

package main

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

func worker(id int, wg *sync.WaitGroup) {
	defer wg.Done() // 确保任务完成后计数器减1
	fmt.Printf("Worker %d starting\n", id)
	time.Sleep(time.Second) // 模拟耗时任务
	fmt.Printf("Worker %d done\n", id)
}

func main() {
	var wg sync.WaitGroup

	for i := 1; i <= 3; i++ {
		wg.Add(1) // 每个任务开始前增加计数器
		go worker(i, &wg)
	}

	wg.Wait() // 阻塞直到所有任务完成
	fmt.Println("All workers completed")
}

关键注意事项

指针传递
WaitGroup 必须通过指针传递给 Goroutine，否则会导致副本操作，引发死锁。
计数器的正确性
- Add() 必须在 Wait() 之前调用（通常在启动 Goroutine 前）。
- Done()次数必须与 Add()的总增量匹配，否则会 panic。
避免竞态条件
不要在多个 Goroutine中并发调用 Add()（建议在主 Goroutine中集中管理）。
不可重用
一个 WaitGroup在计数器归零后不能重复使用（除非重新初始化）。