golang 同步等待全部协程执行完毕sync WaitGroup

golang的sync的包有一个功能WaitGroup

 

做用：

阻塞主线程的执行，直到全部的goroutine执行完成，说白了就是必须同步等待全部异步操做完成！！！

 

三个方法：

Add:添加或者减小等待goroutine的数量

Done:至关于Add(-1)

Wait:执行阻塞，直到全部的WaitGroup数量变成0

 

example：

package main

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

func main() {
	var wg sync.WaitGroup

	for i := 0; i < 5; i = i + 1 {
		wg.Add(1)
		go func(n int) {
			// defer wg.Done()
			defer wg.Add(-1)
			EchoNumber(n)
		}(i)
	}

	wg.Wait()
}

func EchoNumber(i int) {
	time.Sleep(3e9)
	fmt.Println(i)
}

　　程序很简单，只是将每次循环的数量过3秒钟输出。那么，这个程序若是不用WaitGroup，那么将看不见输出结果。由于goroutine还没执行完，主线程已经执行完毕。注释的defer wg.Done()和defer wg.Add(-1)做用同样。

这个很好，原来执行脚本，都是使用time.Sleep，用一个估计的时间等到子线程执行完。WaitGroup很好。虽然chanel也能实现，可是以为若是涉及不到子线程与主线程数据同步，这个感受不错。