因为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
是可以去掉的,例子里用只是为了防止主程序退出之后,没有全部输出,正常工作中,程序一般都是阻塞式的,所以可以去掉
还有其他实现方式,比如利用原子包,互斥锁等,大家可以多去试试
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。