golang使用之使用channel限制goroutine的数量

来源：互联网发布：华为校招知乎编辑：程序博客网时间：2024/05/24 04:50

golang虽然启动一个goroutine很廉价但并不是可以无限制的使用的.

大多数的是用channel来限制goroutine的数量

写了一个测试DEMO如下：

package mainimport ("fmt""runtime""time")var (chanNum   = 3 //启动的数量readChan  = make(chan int)        //操作信息的channellimitChan = make(chan bool, 1000) //限制goroutine数量的channel，此处限制1000个)//初始人方法 func init() {fmt.Println("init")for i := 0; i < chanNum; i++ {go Queue(i, readChan) //开启工作池}}func main() {fmt.Println("main")//启一个go方法 ， 无限制的往readChan里塞数据go func() {for {readChan <- 1}}()//监听到键盘事件后程序退出var input stringfmt.Scanln(&input)}//工作池func Queue(qid int, rchan chan int) {var dat intt := time.Tick(time.Second) //定时器，一秒for {select {case d := <-rchan:limitChan <- true //缓冲区满之后阻塞，后面的readChan将等待dat += dgo showNum(qid, dat) //每从channel接到一个数据就起一个goroutine,limitChan会限制goroutine的数量case <-t:showGoNum(qid) //定时器，每秒打印一次当前 goroutine数量}}}func showNum(qid, i int) {//处理成功后在限制goroutine的Channel缓冲区里取一个数据，limitChan就可以再写入//使用Defer 确保limitChan的一个缓冲区被释放defer func() {<-limitChan}()time.Sleep(time.Millisecond * 100) //模拟程序处理耗时//fmt.Println(qid, "===========", i)}//显示当前goroutine数量 func showGoNum(qid int) {fmt.Printf("%d====numGo:==%d\n", qid, runtime.NumGoroutine())}

执行结果如下：

go run channel_limit_goroutine.go

init
main
0====numGo:==1004
1====numGo:==1003
2====numGo:==1005
1====numGo:==1005
0====numGo:==1005
2====numGo:==1005
0====numGo:==1005
1====numGo:==1005
2====numGo:==1005

0 0