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")
}
输出结果

如果我们使用go run main.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")
}
输出结果

如果我们使用go run main.go命令运行上面的代码,那么我们将得到以下输出 -

First
添加数字...
Second