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 孟获投降了 ======

猜你喜欢

转载自blog.csdn.net/xingzuo_1840/article/details/124576621