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

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/miao0916/article/details/55045894
个人分类: 限制goroutine数量
想对作者说点什么? 我来说一句

Go Channel 详解

Go Channel 详解

shachao888 shachao888

2016-10-27 11:25:54

阅读数:3378

没有更多推荐了,返回首页

不良信息举报

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

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭