迹忆客 专注技术分享

当前位置:主页 > 学无止境 > 编程语言 >

Go WaitGroup 用法详解

作者:迹忆客 最近更新:2021/11/29 浏览次数:

在下一篇我们来介绍Go 工作池。 要了解工作池,我们首先需要了解 WaitGroup,因为它将在工作池的实现中使用。

WaitGroup 用于等待一组 Goroutine 完成执行。 控制被阻塞,直到所有 Goroutines 执行完毕。 假设我们有 3 个并发的从主 Goroutine 衍生的 Goroutine。 主 Goroutines 需要等待其他 3 个 Goroutines 完成才能终止。 这可以使用 WaitGroup 来完成。

下面我们通过一个示例来看一下WaitGroup的用法

package main

import (  
    "fmt"
    "sync"
    "time"
)

func process(i int, wg *sync.WaitGroup) {  
    fmt.Println("started Goroutine ", i)
    time.Sleep(2 * time.Second)
    fmt.Printf("Goroutine %d ended\n", i)
    wg.Done()
}

func main() {  
    no := 3
    var wg sync.WaitGroup
    for i := 0; i < no; i++ {
        wg.Add(1)
        go process(i, &wg)
    }
    wg.Wait()
    fmt.Println("所有 routines 完成了执行!")
}

运行示例

WaitGroup 是一个结构体类型,我们创建了一个 WaitGroup 类型的零值变量。 WaitGroup 的工作方式是使用计数器。 当我们在 WaitGroup 上调用 Add() 并向它传递一个 int 时,WaitGroup 的计数器会根据传递给 Add 的值递增。 递减计数器的方法是在 WaitGroup 上调用 Done() 方法。 Wait() 方法会阻塞调用它的 Goroutine,直到计数器变为零。

在上面的程序中,我们在迭代 3 次的 for 循环中调用 wg.Add(1)。所以计数器现在变成了 3。 for 循环还产生了 3 个 process Goroutines。然后调用 wg.Wait() 使主 Goroutine 等待直到计数器变为零。 计数器通过在process 协程中调用 wg.Done 递减。 一旦所有 3 个生成的 Goroutine 执行完毕,即 wg.Done() 被调用 3 次,计数器将变为 0,主 Goroutine 将被解锁。

传递 wg 的指针很重要。如果没有传递指针,那么每个 Goroutine 都会有自己的 WaitGroup 副本,并且当它们完成执行时不会通知主协程(main)

上面程序执行结果如下

Go wait group

每个人的输出可能都不同,因为 Goroutines 的执行顺序可能会有所不同:)。

除非注明转载,本站文章均为原创或翻译,欢迎转载,转载请以链接形式注明出处

本文地址:

迹忆客

专注技术分享,项目实战分享!

技术宅 乐于分享 7年编程经验
社交账号
  • https://www.github.com/onmpw
  • qq:1244347461

热门文章

教程更新

热门标签

Go