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

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

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

写了一个测试DEMO如下：

package main

import (
	"fmt"
	"runtime"
	"time"
)

var (
	chanNum   = 3 //启动的数量
	readChan  = make(chan int)        //操作信息的channel
	limitChan = 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 string
	fmt.Scanln(&input)
}

//工作池
func Queue(qid int, rchan chan int) {
	var dat int
	t := time.Tick(time.Second) //定时器，一秒
	for {
		select {
		case d := <-rchan:
			limitChan <- true //缓冲区满之后阻塞，后面的readChan将等待
			dat += d
			go 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