golang中如何阻塞等待所有goroutines都完成

有一天,一个人问了我此问题,回头仔细翻阅了一下资料,仔细的想了一下,这个问题的解决有两种方案。
方案一:
也是推荐方案,也是官方推荐方案,涉及到一个写并发经常关注的模块sync模块,利用里面的sync.WaitGroup去做
代码如下:
package main
import(
        "fmt"
        "sync"
        "time"
        "runtime"
)
var wg sync.WaitGroup //定义一个同步等待的组
func main() {
    maxProcs := runtime.NumCPU() //获取cpu个数
    runtime.GOMAXPROCS(maxProcs) //限制同时运行的goroutines数量
    for i:=0;i<10;i++{
            wg.Add(1)//为同步等待组增加一个成员
            go Printer(i)//并发一个goroutine
    }
    wg.Wait() //阻塞等待所有组内成员都执行完毕退栈
    fmt.Println("WE DONE!!!")
}
//定义一个Printer函数用于并发
func Printer(a int)(){
        time.Sleep(2000 * time.Millisecond)
        fmt.Printf("i am %d\n",a)
        defer wg.Done()
}

方案二:
思路也不绕路,利用的channel的阻塞机制,直接上代码了。
package main
import(
        "fmt"
        "time"
        "runtime"
)
var num=14 //定义一工并发多少数量
var cnum chan int
func main(){
   maxProcs := runtime.NumCPU()// 获取cpu个数
    runtime.GOMAXPROCS(maxProcs)//限制同时运行的goroutines数量
    cnum=make(chan int,num) //make一个chan,缓存为num
    for i:=0;i<num;i++{
            go Printer(i)
    }
// 下面这个for循环的意义就是利用信道的阻塞,一直从信道里取数据,直到取得跟并发数一样的个数的数据,则视为所有goroutines完成。
    for i:=0;i<num;i++{
            <-cnum
    }
    fmt.Println("WE DONE!!!")
}



func Printer(a int)(){
        time.Sleep(2000 * time.Millisecond)
        fmt.Printf("i am %d\n",a)
        cnum <- 1 //goroutine结束时传送一个标示给信道。
}

转载于:https://www.cnblogs.com/smallleiit/p/10844511.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值