sync.WaitGroup 实现 goroutine

追赶技术的脚步 · · 978 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。


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

func main() {

    var wg sync.WaitGroup
    wg.Add(3)
    s1 := 0
    s2 := 0

    go func() {
        defer wg.Done()
        js()
    }()

    go func() {
        defer wg.Done()
        s1 = a1(1, 2)
    }()
    go func() {
        defer wg.Done()
        //todo 在这里加睡眠时间,来测试执行结果
        time.Sleep(10*time.Second)
        s2 = a2(11, 2)
    }()
    wg.Wait()
    fmt.Println("s1=", s1)
    fmt.Println("s2=", s2)
    //打印时间
    fmt.Println(time.Now())
}

func js()  {
    //打印时间
    fmt.Println(time.Now())
}

func a1(a,b int)int {
    fmt.Println("111111111111")
    return a+b
}

func a2(a,b int)int {
    fmt.Println("222222222222")
    return a + b
}






这是执行结果


image.png

WaitGroup 会将main goroutine阻塞直到所有的goroutine运行结束,从而达到并发控制的目的。使用方法非常简单,真心佩服创造Golang的大师们!

type WaitGroup //相当于一个箱子,将main goroutine 保护到里面

func (*WaitGroup) Add //调用一次为箱子加一把锁(当然,你愿意也可以多把)

func (*WaitGroup) Done // 调用一次开一把锁(only one!)

func (*WaitGroup) Wait //箱子的盖子,没锁了自动打开

官网说明:一个WaitGroup锁等待一个goroutines合集结束。main goroutine里面调用Add方法设置需要等待的goroutines 数量,然后运行每一个goroutine,并且当其结束时调用Done方法。同时,main goroutine 被锁住直到所有的goroutines完成。

同时发现,channel和WaitGroup,都是阻塞式的。但是使用channel比WaitGroup会好一点,channel是先进先出的,但是WaitGroup是要等到所有的goroutine全部完成后,才返回结果来。而且WaitGroup启用select也是需要借助channel来帮忙的,所以,个人认为channel实现起来,是不错的选择


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:追赶技术的脚步

查看原文:sync.WaitGroup 实现 goroutine

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

978 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传