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