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()
}

上面主要用到的是gosync包下的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是可以去掉的,例子里用只是为了防止主程序退出之后,没有全部输出,正常工作中,程序一般都是阻塞式的,所以可以去掉

还有其他实现方式,比如利用原子包,互斥锁等,大家可以多去试试

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值