Golang--channel+waitGroup控制并发量

channel+waitGroup控制并发量

前言

  • golang的goroutine非常轻量级,同时启动数万协程都没问题。
  • 如果不对并发量进行控制,比如同时产生数百万的协程,会压垮服务器
  • 通过控制channel缓冲区的大小,从而控制并发数
  • waitGroup保证子协程都运行完成后主协程才退出

示例

产生了100个协程,控制并发数为10,可以看到用时10098ms
在这里插入图片描述

package main

import (
	"fmt"
	"sync"
	"time"
)

// channel + WaitGroup实现控制并发数量
var maxConcurrentNum = 10

var chData = make(chan int, maxConcurrentNum)
var wg sync.WaitGroup
var taskNum = 100

func main() {
	start_time := time.Now().UnixMilli()
	for i := 0; i < taskNum; i++ {
		chData <- 1
		go doTask(i)
	}
	wg.Wait()
	close(chData)
	fmt.Println("Main over")
	end_time := time.Now().UnixMilli()
	fmt.Printf("Duration: %vms", end_time-start_time)
}
func doTask(taskID int) {
	wg.Add(1)
	time.Sleep(time.Millisecond * 1000)
	fmt.Println("Finish task ", taskID)
	<-chData
	wg.Done()
}



调整并发数为50时,用时2032ms
在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值