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