关于go语言中的WaitGroup

OGeek|极客世界-中国程序员成长平台 › 门户 › 编程› Go›Go教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

如果你刚接触Go语言并且想用它构建高并发，高性能的应用，弄明白WaitGroups是怎么回事很重要。

在本教程中，我们将掌握以下内容：

WaitGroups的用途
一个WaitGroups的简单示例

学习完本教程，你将会对WaitGroups有个全面的了解并且可以在你的高并发go应用中使用它。

理解WaitGroups

接下来我们就来看一下WaitGroups是什么以及它能为我们解决什么问题

当你在程序中使用go语言的协程的时候，在协程执行完成之前，你需要阻塞程序的执行。

请看以下代码示例：

 1 package main
 2 
 3 import (
 4     "fmt"
 5 )
 6 
 7 func main() {
 8     fmt.Println("Hello World")
 9     go myFunc()
10     fmt.Println("Finished Execution")
11 }
12 
13 func myFunc() {
14     fmt.Println("Inside my goroutine")
15 }

程序输出如下：

Hello World
Finished Execution

我们发现，程序并没有输出"Inside my goroutine"，原因是主程序在协程执行之前就已经退出了。

如何解决？使用WaitGroups

WaitGroups可以通过阻塞主函数来帮我们解决上面的问题。

请看以下代码：

 1 package main
 2 
 3 import (
 4     "sync"
 5     "fmt"
 6 )
 7 
 8 func main() {
 9     fmt.Println("Hello World")
10     var waitgroup sync.WaitGroup
11     waitgroup.Add(1)
12     go myFunc(&waitgroup)
13     waitgroup.Wait()
14 
15     fmt.Println("Finished Execution")
16 }
17 
18 func myFunc(waitgroup *sync.WaitGroup) {
19     fmt.Println("Inside my goroutine")
20     waitgroup.Done()
21 }

上面代码输出结果为：

Hello World
Inside my goroutine
Finished Execution

达到了我们想要的结果。

接下来对上面的代码做一些解释：

我们在开启协程之前，先调用了WaitGroup的Add(1)方法，是要设置主函数需要等待完成的协程数为1，Wait()方法是等待协程的完成。我们在协程中调用的WaitGroup的Done()方法，意思是当前协程执行完成（Done()做的工作其实就是把需要等待的协程个数减1），当需要等待的协程数为0时，则不需要再等待，继续执行以下的代码。

改造成匿名函数

我们可以把上面的例子改造成匿名函数的，代码如下：

package main

import (
    "sync"
    "fmt"
)

func main() {
    fmt.Println("Hello World")
    var waitgroup sync.WaitGroup
    waitgroup.Add(1)
    go func(){
        fmt.Println("Inside my goroutine")
        waitgroup.Done()
    }()
    waitgroup.Wait()

    fmt.Println("Finished Execution")
}