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结束时传送一个标示给信道。
}

posted @ 2019-05-10 15:05  small_lei_it  阅读(653)  评论(0编辑  收藏  举报