Go语言中通过WaitGroup控制并发

与Channel区别

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

创新互联建站-专业网站定制、快速模板网站建设、高性价比兴山网站开发、企业建站全套包干低至880元,成熟完善的模板库,直接使用。一站式兴山网站制作公司更省心,省钱,快速模板网站建设找我们，业务覆盖兴山地区。费用合理售后完善，10多年实体公司更值得信赖。

基本使用示例

我们将之前的示例加以改造，引入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控制并发
网页网址：http://www.shufengxianlan.com/qtweb/news12/553362.html

网站建设、网络推广公司-创新互联，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 创新互联