golang并发编程-05-同步-04-sync包的once、WaitGroup结构体的常用方法

1. 函数仅执行一次（onceDo）

1.1 语法

• 作用
  传入onceDo的函数，不管调用多少次onceDo，也只会执行一次（首次执行）。

• 语法

func (o *Once) Do(f func())

示例

var once sync.Once
once.Do(f)

f是只调用一次的函数

1.2 示例（空城计）

传说空城计只能用一次。

说明：
不管空城计调用多少次，司马懿被吓跑函数只在第一次发生

func main() {
	var result = false
	var once sync.Once
	f := func() {
		fmt.Println("【司马懿】吓跑了 ==》")
		result = true
	}

	for i := 1; i < 4; i++ {
		fmt.Printf("===== 空城计 第 %d 次 =====\n",i)
		once.Do(f)
		if result{
			fmt.Println("+++胜利+++")
		}else {
			fmt.Println("---失败---")
		}
		time.Sleep(time.Second)
		result = false

	}
}

输出

===== 空城计 第 1 次 =====
【司马懿】吓跑了 ==》
+++胜利+++
===== 空城计 第 2 次 =====
---失败---
===== 空城计 第 3 次 =====
---失败---

2. 等待组（WaitGroup）

前边我们已经多次使用，在收尾时总结一下

2.1 语法

• 定义等待组

var wg sync.WaitGroup

• 等待组计数器加

go.add(N)

• 计数器减

wg.Done()

• 等待组等待计数结束

wg.Wait()

2.2 示例1（七擒孟获）

等待组等待七次孟获被抓后，打印孟获头像了。

func main() {
	var wg sync.WaitGroup
	wg.Add(7)

	for i:=1;i<=7;i++{
		fmt.Printf("孟获被抓了%d次\n",i)
		time.Sleep(time.Second)
		wg.Done()
	}
    wg.Wait()
	fmt.Println("====== END 孟获投降了 ======")
}

输出

孟获被抓了1次
孟获被抓了2次
孟获被抓了3次
孟获被抓了4次
孟获被抓了5次
孟获被抓了6次
孟获被抓了7次
====== END 孟获投降了 ======

2.3 示例2（捕获量子孟获）

实际使用中，等待组成员通常是在不同协程中，因此我们将看到多协程在时空乱流中七次捕获“量子孟获”

func main() {
	var wg sync.WaitGroup
	wg.Add(7)

	for i:=1;i<=7;i++{
		tmp := i
		go func(tmp int) {
			fmt.Printf("孟获被抓了%d次\n",tmp)
			wg.Done()
		}(tmp)
	}
    wg.Wait()
	fmt.Println("====== END 孟获投降了 ======")
}

• 结果输出

七个协程在时空乱流中瞬间捕获了七次孟获，“量子孟获”投降了

孟获被抓了1次
孟获被抓了7次
孟获被抓了3次
孟获被抓了6次
孟获被抓了4次
孟获被抓了5次
孟获被抓了2次
====== END 孟获投降了 ======

---