Go语言通过WaitGroup实现控制并发的示例详解

更新时间：2023年01月30日 10:51:37 作者：孙琦Ray

Channel能够很好的帮助我们控制并发，但是在开发习惯上与显示的表达不太相同，所以在Go语言中可以利用sync包中的WaitGroup实现并发控制，本文就来和大家详细聊聊WaitGroup如何实现控制并发

与Channel区别

Channel能够很好的帮助我们控制并发，但是在开发习惯上与显示的表达不太相同，所以在Go语言中可以利用sync包中的WaitGroup实现并发控制，更加直观。

基本使用示例

我们将之前的示例加以改造，引入sync.WaitGroup来实现并发控制。

首先我们在主函数中定义WaitGroup

var wg sync.WaitGroup

每执行一个任务，则调用Add()方法

wg.Add(1)

在主函数中我们利用Wait()方法等待并发结束

wg.Wait()

在调用的函数中，我们需要将WaitGroup以指针方式传入，否则将造成Deadlock

// 主函数内
go ready(5, &wg)

// 函数
func ready(s int, wg *sync.WaitGroup)

同时在函数执行完成后，调用wg.Done，我们使用defer实现

defer wg.Done()

完整代码

package main

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

func ready(s int, wg *sync.WaitGroup) {

    defer wg.Done()

    fmt.Printf("Run func in a goroutine and wait for %v\n", s)
    time.Sleep(time.Second * time.Duration(s))
    fmt.Printf("Run func in a goroutine and wait for %v end\n", s)
}

func main() {

    var wg sync.WaitGroup

    wg.Add(1)
    go ready(5, &wg)

    mainWaitSec := 2
    fmt.Printf("Run Main function and wait for %v\n", mainWaitSec)
    time.Sleep(time.Second * time.Duration(mainWaitSec))
    fmt.Printf("Run Main function and wait for %v done\n", mainWaitSec)

    wg.Wait()

}

特别提示

WaitGroup传入给函数时，需要以指针方式传递，否则会造成Deadlock

多任务示例

如果不想在函数中传递WaitGroup，也可以采用以下这种方式，通过并发匿名函数的方式，在主函数逻辑中对并发进行精准控制

    var wg sync.WaitGroup

    for i := 0; i < 5; i++ {
        wg.Add(1)

        waitSec := i + 1
        go func() {
            defer wg.Done()
            ready(waitSec)
        }()
    }

完整代码

package main

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

func ready(s int) {
    fmt.Printf("Run func in a goroutine and wait for %v\n", s)
    time.Sleep(time.Second * time.Duration(s))
    fmt.Printf("Run func in a goroutine and wait for %v end\n", s)
}

func main() {

    var wg sync.WaitGroup

    for i := 0; i < 5; i++ {
        wg.Add(1)

        waitSec := i + 1
        go func() {
            defer wg.Done()
            ready(waitSec)
        }()
    }

    mainWaitSec := 2
    fmt.Printf("Run Main function and wait for %v\n", mainWaitSec)
    time.Sleep(time.Second * time.Duration(mainWaitSec))
    fmt.Printf("Run Main function and wait for %v done\n", mainWaitSec)

    wg.Wait()

}

运行结果如下

Run Main function and wait for 2
Run func in a goroutine and wait for 2
Run func in a goroutine and wait for 4
Run func in a goroutine and wait for 5
Run func in a goroutine and wait for 1
Run func in a goroutine and wait for 3
Run func in a goroutine and wait for 1 end
Run Main function and wait for 2 done
Run func in a goroutine and wait for 2 end
Run func in a goroutine and wait for 3 end
Run func in a goroutine and wait for 4 end
Run func in a goroutine and wait for 5 end

到此这篇关于Go语言通过WaitGroup实现控制并发的示例详解的文章就介绍到这了,更多相关Go语言 WaitGroup控制并发内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Golang TCP粘包拆包问题的解决方法
这篇文章主要给大家介绍了Golang TCP粘包拆包问题的解决方法，文中通过示例代码介绍的非常详细，对大家学习或者使用Golang具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
2019-07-07
Go语言中的Slice学习总结
这篇文章主要介绍了Go语言中的Slice学习总结,本文讲解了Slice的定义、Slice的长度和容量、Slice是引用类型、Slice引用传递发生“意外”等内容,需要的朋友可以参考下
2014-11-11
解析Go语言编程中的struct结构
这篇文章主要介绍了Go语言编程中的struct结构,是Go语言入门学习中的基础知识,需要的朋友可以参考下
2015-10-10
Golang urfave/cli库简单应用示例详解
这篇文章主要为大家介绍了Golang urfave/cli库简单应用示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2023-09-09
Golang RSA生成密钥、加密、解密、签名与验签的实现
RSA 是最常用的非对称加密算法,本文主要介绍了Golang RSA生成密钥、加密、解密、签名与验签的实现,具有一定的参考价值,感兴趣的可以了解一下
2023-11-11
golang metrics各个指标含义讲解说明
这篇文章主要为大家介绍了golang metrics各个指标含义讲解说明，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-05-05
Go语言开发代码自测绝佳go fuzzing用法详解
这篇文章主要为大家介绍了Go语言开发代码自测绝佳go fuzzing用法详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-06-06
Go语言每天必学之switch语句
这篇文章主要为大家详细介绍了Go语言每天必学之switch语句的相关资料，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2017-12-12
go使用makefile脚本编译应用的方法小结
makefile可以看作是make工具的脚本文件，而make主要用来处理一系列命令。常用的比如用来编译和打包文件，在C/C++的编译打包中应用最广泛了，这篇文章主要介绍了go使用makefile脚本编译应用,需要的朋友可以参考下
2022-08-08
一文初探Go语言中的reflect反射包
这篇文章主要和大家分享一下Go语言中的reflect反射包，文中的示例代码讲解详细，对我们学习Go语言有一定的帮助，需要的小伙伴可以参考一下
2022-12-12