Go 并发控制--WaitGroup的使用

开发过程中，经常遇到task之间的同步问题。例如，多个子task并发完成一部分任务，主task等待他们最后结束。

在Go语言，实现同步的一种方式就是WaitGroup。

Example

package main 

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



func main() {

	var wg sync.WaitGroup


	wg.Add(3)
	go func(n int){
		fmt.Println("n:", n)
		t := time.Duration(n)*time.Second
		time.Sleep(t)

		wg.Done()
	}(1)


	go func(n int){
		fmt.Println("n:", n)
		t := time.Duration(n)*time.Second
		time.Sleep(t)
		time.Sleep(t)

		wg.Done()
	}(2)

	go func(n int){
		fmt.Println("n:", n)
		t := time.Duration(n)*time.Second
		time.Sleep(t)

		wg.Done()
	}(3)

	wg.Wait()

	fmt.Println("main exit...")
}

output:

n: 3

n: 1

n: 2

main exit…

注意

WaitGroup变量定义后，是不允许被拷贝的，即不允许作为函数参数或者赋值给其他变量。