Golang 中的 WaitGroup 是什么?
在Golang中可能存在执行不同goroutine可能导致意外行为的实例。在这种情况下,我们希望确保某些goroutines以预定义的方式工作,并且程序会等待所有从main函数启动的goroutines等待。为此,我们使用WaitGroups。
WaitGroups使我们能够解决以上,因为他们的代码块,直到所提及的任何问题够程的范围内WaitGroup已成功执行。
一个WaitGroup有三个导出的方法是,我们利用的。这些是-
Add(int)–增加计数器。
Wait()–阻止执行,直到内部计数器变为0。
Done()–将计数器减1。
示例1
让我们看一个例子,我们将使用一个gouroutine没有任何WaitGroups。
package main import ( "fmt" ) func sumOfNumbers() { fmt.Println("添加数字...") } func main() { fmt.Println("First") go sumOfNumbers() fmt.Println("Second") }输出结果
如果我们使用gorunmain.go命令运行上面的代码,那么我们将得到以下输出-
First Second
请注意,我们如何无法获得在goroutine中调用的函数的输出。这是因为main函数在执行goroutine之前就被终止了。
为了解决这个问题,我们使用WaitGroups,它用于阻塞程序,直到该WaitGroup中的任何goroutine成功执行。
示例2
现在,让我们在WaitGroups的帮助下解决上面的例子。考虑下面显示的代码。
package main import ( "fmt" "sync" ) func sumOfNumbers(wg *sync.WaitGroup) { fmt.Println("添加数字...") wg.Done() } func main() { fmt.Println("First") var wg sync.WaitGroup wg.Add(1) go sumOfNumbers(&wg) wg.Wait() fmt.Println("Second") }输出结果
如果我们运行上面的代码,它将产生以下输出-
First 添加数字... Second
示例3:匿名函数中的等待组
如果我们有一个在单独的goroutine上运行的匿名函数,我们也可以实现与上面相同的功能。
考虑下面显示的代码。
package main import ( "fmt" "sync" ) func main() { fmt.Println("First") var wg sync.WaitGroup wg.Add(1) go func() { fmt.Println("添加数字...") wg.Done() }() wg.Wait() fmt.Println("Second") }输出结果
如果我们使用gorunmain.go命令运行上面的代码,那么我们将得到以下输出-
First 添加数字... Second