首页 > Golang > Go教程

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

来源：脚本之家

时间：2023-02-25 10:15:32 500浏览收藏

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《Go语言通过WaitGroup实现控制并发的示例详解》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到语言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

完整代码

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

运行结果如下

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实现控制并发的示例详解》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于golang的内容请关注golang学习网公众号！

golang 语言WaitGroup

声明：本文转载于：脚本之家如有侵犯，请联系study_golang@163.com删除