迹忆客 专注技术分享

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

Go WaitGroup 用法详解

作者:迹忆客 最近更新:2023/01/08 浏览次数:

在下一篇我们来介绍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 的执行顺序可能会有所不同:)。

转载请发邮件至 1244347461@qq.com 进行申请,经作者同意之后,转载请以链接形式注明出处

本文地址:

相关文章

使用 C 语言中的 goto 语句

发布时间:2023/05/07 浏览次数:79 分类:C语言

本文介绍了如何在 C 语言中使用 goto 语句。使用 goto 语句在 C 语言中实现循环 goto 关键字是 C 语言的一部分,它提供了一个做无条件跳转的结构。

Django 中的 Slug

发布时间:2023/05/04 浏览次数:173 分类:Python

本篇文章旨在定义一个 slug 以及我们如何使用 slug 字段在 Python 中使用 Django 获得独特的帖子。

Django ALLOWED_HOSTS 介绍

发布时间:2023/05/04 浏览次数:181 分类:Python

本文展示了如何创建您的 Django 网站,为公开发布做好准备,如何设置 ALLOWED_HOSTS 以及如何在使用 Django 进行 Web 部署期间修复预期的主要问题。

Django 中的 Select_related 方法

发布时间:2023/05/04 浏览次数:129 分类:Python

本文介绍了什么是查询集,如何处理这些查询以及我们如何利用 select_related() 方法来过滤 Django 中相关模型的查询。

在 Django 中上传媒体文件

发布时间:2023/05/04 浏览次数:198 分类:Python

在本文中,我们简要介绍了媒体文件以及如何在 Django 项目中操作媒体文件。

Django 返回 JSON

发布时间:2023/05/04 浏览次数:106 分类:Python

在与我们的讨论中,我们简要介绍了 JSON 格式,并讨论了如何借助 Django 中的 JsonResponse 类将数据返回为 JSON 格式。

在 Django 中创建对象

发布时间:2023/05/04 浏览次数:59 分类:Python

本文的目的是解释什么是模型以及如何使用 create() 方法创建对象,并了解如何在 Django 中使用 save() 方法。

扫一扫阅读全部技术教程

社交账号
  • https://www.github.com/onmpw
  • qq:1244347461

最新推荐

教程更新

热门标签

扫码一下
查看教程更方便