golang学习笔记之—WaitGroup
自己毕业工作过后,由于时间有限,一度中断了写学习笔记。
最近心血来潮,逛CSDN发现博客已经改版了,添加了对Markdown语法的支持。加之最近在学习google的go语言,因此想借写博客的机会学习Markdown语言,同时梳理所学的go语言知识。
本文主要讲解go标准库sync中的WaitGroup的用法。
WaitGroup的作用
WaitGroup用于goroutine的同步,当需要阻塞当前执行线程,等待一组goroutine执行完毕之后再继续执行当前线程时,就需要用到WaitGroup。
WaitGroup的定义
type@H_301_22@ WaitGroup struct@H_301_22@ {
state1 [12@H_301_22@]byte@H_301_22@
sema uint32@H_301_22@
}
WaitGroup的定义比较简单,由一个12字节额 state1 字段和一个32位的 sema 组成。
WaitGroup的API
1.Add(delta int)
Add的原型声明如下:
func@H_301_22@ (wg *WaitGroup) Add(delta int@H_301_22@)
Add函数接受一个int
型的参数delta,用于设置WaitGroup实例(wg)的计数器,当计数器变为0时,所有因为在该wg上调用Wait而阻塞的goroutine的都会被唤醒。
若调用Add导致计数器变为负数,会引起panic。
2.Done()
Done的原型声明如下:
func@H_301_22@ (wg *WaitGroup) Done()
Done函数的作用很简单,就是将wg的计数器减一。该函数等同于wg.Add(-1)
,从go的源码中可以看到Done()就是用wg.Add(-1)实现的。
// Done decrements the WaitGroup counter.@H_301_22@
func@H_301_22@ (wg *WaitGroup) Done() {
wg.Add(-1@H_301_22@)
}
3.Wait()
Wait的原型声明如下:
func@H_301_22@ (wg *WaitGroup) Wait()
Wait会阻塞调用该函数的goroutine,直到wg的计数器变为1。
WaitGroup的用法
package@H_301_22@ main
import@H_301_22@ (
"fmt"@H_301_22@
"sync"@H_301_22@
)
func@H_301_22@ main() {
var@H_301_22@ wg sync.WaitGroup
for@H_301_22@ i := 0@H_301_22@; i < 10@H_301_22@; i++ {
wg.Add(1@H_301_22@)
go@H_301_22@ func@H_301_22@(i int@H_301_22@){
fmt.Println("Hello world"@H_301_22@,i)
wg.Done()
}(i)
}
wg.Wait()
}
该程序启动了十个goroutine,并在main
函数中使用wg.Wait
等待这些goroutine执行结束。该程序的输出如下:
Hello world 9@H_301_22@
Hello world 2@H_301_22@
Hello world 3@H_301_22@
Hello world 4@H_301_22@
Hello world 5@H_301_22@
Hello world 6@H_301_22@
Hello world 7@H_301_22@
Hello world 0@H_301_22@
Hello world 8@H_301_22@
Hello world 1@H_301_22@