阅读 213

Golang 中的 WaitGroup 是什么?

在 Golang 中可能存在执行不同goroutine可能导致意外行为的实例。在这种情况下,我们希望确保某些goroutines以预定义的方式工作,并且程序会等待所有从 main 函数启动的goroutines等待。为此,我们使用WaitGroups

WaitGroups使我们能够解决以上,因为他们的代码块,直到所提及的任何问题够程的范围内WaitGroup已成功执行。

一个WaitGroup有三个导出的方法是,我们利用的。这些是 -

  • Add(int) – 增加计数器。

  • Wait() – 阻止执行,直到内部计数器变为 0。

  • Done() – 将计数器减 1。

示例 1

让我们看一个例子,我们将使用一个gouroutine没有任何WaitGroups


package mainimport (
   "fmt")func sumOfNumbers() {
   fmt.Println("添加数字...")
}func main() {
   fmt.Println("First")

   go sumOfNumbers()

   fmt.Println("Second")
}

输出结果


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


FirstSecond


请注意,我们如何无法获得在goroutine中调用的函数的输出。这是因为main函数在执行goroutine之前就被终止了。

为了解决这个问题,我们使用WaitGroups,它用于阻塞程序,直到该WaitGroup 中的任何goroutine成功执行。

示例 2

现在,让我们在WaitGroups的帮助下解决上面的例子。考虑下面显示的代码。


package mainimport (
   "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 mainimport (
   "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


文章分类
百科问答
版权声明:本站是系统测试站点,无实际运营。本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 XXXXXXo@163.com 举报,一经查实,本站将立刻删除。
相关推荐