zoukankan      html  css  js  c++  java
  • Golang使用WaitGroup的一点随想

    前言(为什么又要写一篇随想文)

    上次我写了一个channel的文章,我寻思,这Golang控制三大巨头,channel,waitgroup,context,我得尽快都安排上,最近工作太忙,压力过大,但是Update Blog还是不能够停下来,所以继续补上,学习还是不能停,那么来吧。

    WaitGroup的简单用法(等待组)

    你品一下人家这名字,等待组。等待什么,等待goroutine完成啊。有些时候,我们启动多个goroutine去执行任务,我举个例子

    listip := []string{"10.0.9.11","10.0.9.22","10.0.9.33"}
    
    for _, ip := range(listip) {
        //假设我们执行一个ping ip 的逻辑
        go PingIPWork(ip)
    }
    
    

    我这里执行了一个多ip去ping的逻辑,一般这种时候,你要是执行一波,人家肯定毛都不会返回给你,为什么?因为人家主线程直接就退出了,还是那句话,你又没告诉人家主线程要等这ip全部都ping 完,所以你必须要加个等待,等着Goroutine完成,这里我再举一个网上的例子

    package main
    
    import (
        "fmt"
    )
    
    func main() {
        go func() {
            fmt.Println("Goroutine 1")
        }()
    
        go func() {
            fmt.Println("Goroutine 2")
        }()
    
        //来个睡眠,等Goroutine结束
        time.Sleep(time.Second * 1)
    }
    

    看到了么,加了一个sleep,用sleep去等着Goroutine跑完,上面我举的那个例子也可以这么来

    listip := []string{"10.0.9.11","10.0.9.22","10.0.9.33"}
    
    for _, ip := range(listip) {
        //假设我们执行一个ping ip 的逻辑
        go PingIPWork(ip)
    }
    
    time.Sleep(time.Second * 1)
    

    加个sleep可以等待完成,但是万一啊,Goroutine有的跑的快,有的慢,你那sleep就一秒,要是有的Goroutine没跑完不就白瞎了吗,所以咱们需要一个机制,这个机制可以帮助咱们去管理Goroutine,让我们知道Goroutine这东西什么时候停,什么时候完成。所以,WaitGroup这个东西,就可以帮助我们解决这个问题,还是老样子,我举一个简单的例子来说明我的想法。

    package main
    
    import (
        "fmt"
    	"sync"
    )
    
    
    func PingIPWork(ip string) {
    	fmt.Println(ip)
    }
    
    
    func main() {
    
        //定义一个等待阿祖
    	var wg sync.WaitGroup
    
    
    	wg.Add(3) // 因为有3个Ip,咱们定义三个动作,所以来三个计数
    
    	listip := []string{"10.0.9.11","10.0.9.22","10.0.9.33"}
    
    	for _, ip := range(listip) {
    		//假设我们执行一个ping ip 的逻辑
    		go func(ip string) {
                //执行一个work
                PingIPWork(ip)
                //操作完成之后,done一个计数,也就是3-1
    			wg.Done()
    		}(ip)
     }
        //等待
    	wg.Wait() // 等待,直到计数为0
    }
    

    这里我举了一个简单的例子,其实wg的用法较为简单,在这个例子里面我们用到了

    wg.wait
    等待Goroutine结束之后退出主进程
    
    wg.Add
    添加Goroutine,其实你可以把它想成,可添加的最大Goroutine数
    
    wg.Done
    想象成销毁参数,当Goroutine结束之后调用,意思就是,你没了,我减1
    

    WaitGroup的其他注意事项

    将Wg作为参数进行传递的时候,需要使用指针

    有些时候,咱们不想写的这么麻烦,就寻思怎么才能简单一点,或者可变性稍微强一点,有些时候我们要把wg最为参数,在函数内部调用,我们该怎么写呢?

    package main
    
    import (
    	"fmt"
    	"sync"
    )
    
    
    func PingIPWork(ip string, wg *sync.WaitGroup) {
    	fmt.Println(ip)
    	wg.Done()
    }
    
    
    func main() {
    
    	var wg sync.WaitGroup
    
    
    	wg.Add(3) // 因为有两个动作,所以增加2个计数
    
    	listip := []string{"10.0.9.11","10.0.9.22","10.0.9.33"}
    
    	for _, ip := range(listip) {
    		//假设我们执行一个ping ip 的逻辑
    		go PingIPWork(ip, &wg)
    		}
    
    	wg.Wait() // 等待,直到计数为0
    }
    

    看到了么,如果你把Wg作为参数进行传递,你得要用指针的形式传值,否则就会死锁!!!!!!!!

    Wg.Add的数值不能为负

    wg.Add()的数值必须为正数,如果为负数,将会抛出异常。

    panic: sync: negative WaitGroup counter
    
    goroutine 1 [running]:
    sync.(*WaitGroup).Add(0xc042008230, 0xffffffffffffff9c)
        D:/Go/src/sync/waitgroup.go:75 +0x1d0
    main.main()
        D:/code/go/src/test-src/2-Package/sync/waitgroup/main.go:10 +0x54
    
  • 相关阅读:
    Ubuntu+XAMPP+Wordpress的安装与配置问题
    Html5最简单的游戏Demo——Canvas绘图的骰子
    Html5最简单的游戏Demo——Canvas绘图的弹弹球
    LinqToExcel: LINQ查询Excel电子表格
    Asp.Net Web API开发微信后台
    Json.Net使用JSON Schema验证JSON格式【实例】
    Json.Net使用JSON Schema验证JSON格式
    LeetCode-C#实现-哈希表(#349)
    大话设计模式-解释器模式
    大话设计模式-享元模式
  • 原文地址:https://www.cnblogs.com/Yemilice/p/12994816.html
Copyright © 2011-2022 走看看