go语言中控制并发数量的方法

因为go语言层面支持并发，所以面试中经常会问到并发的问题，比如说控制go并发数量的方式有哪些？下面是我个人整理的两个例子：

func waitGroup() {    count := 10    wg := sync.WaitGroup{}    for i := 0; i < count; i++ {        wg.Add(1)        go func(j int) {            fmt.Print(j)            wg.Done() // 也可使用 wg.Add(-1)        }(i)    }    wg.Wait()}

上面主要用到的是go中sync包下的waitGroup，这也是在工作中比较常见的实现方式，关键点就是把握好Add方法的位置，Wait方法则是等待所有的协程执行完毕

func channel() {    count := 10 // 最大支持并发    sum := 100 // 任务总数    c := make(chan struct{}, count) // 控制任务并发的chan    sc := make(chan struct{}, sum) // 控制任务总数的chan    defer close(c)    defer close(sc)    for i:=0; i<sum;i++{        c <- struct{}{} // 作用类似于waitgroup.Add(1)        go func(j int) {            fmt.Println(j)            <- c // 执行完毕，释放资源            sc <- struct {}{} // 记录到执行总数里        }(i)    }    for i:=sum; i>0;i-- {        <- sc    }}

上面的例子用到的是go中的channel，利用channel阻塞的特性和带缓冲的channel来实现控制并发数量，其中sc这个channel是可以去掉的，例子里用只是为了防止主程序退出之后，没有全部输出，正常工作中，程序一般都是阻塞式的，所以可以去掉。

更多相关知识请关注go语言教程栏目

更多相关文章

随机推荐