Golang Sync.WaitGroup 使用及原理

Golang Sync.WaitGroup 使用及原理

使用

func main() {
   
	var wg sync.WaitGroup
	for i := 0; i < 10; i++ {
   
		wg.Add(1)
		go func() {
   
			defer wg.Done()
			fmt.Println("Hello WaitGroup!")
		}()
	}
	wg.Wait()
}

实现

首先看 waitgroup 到底是什么数据结构

type WaitGroup struct {
   
	noCopy noCopy
	state1 [3]uint32
}

nocopy 避免这个结构体被复制的一个技巧，可以告诉go vet工具违反了复制使用的规则
state1 [3]uint32 字段中包含了 waitgroup 的所有状态信息, 根据标准库上自带的注释简单翻译是：state1 由 12 个字节组成，其中将8字节看作64位值，其中高32位存放的是 counter 计数器, 代表目前还未完成的 goroutine个数，低32位存放的是 waiter 计数器, 可以理解成下面这个结构体

type WaitGroup struct {
   
	// 代表目前尚未完成的个数
	// WaitGroup.Add(n) 将会导致 counter += n
	// WaitGroup.Done() 将导致 counter--
	counter uint32

	// 目前已调用 WaitGroup.Wait 的 goroutine 的个数
	waiter  uint32

	// 对应于 golang 中 runtime 内部的信号量