利用channel阻塞的特性，另外在我实际业务中需要协程去处理数据，下面我使用了 context 来解决。
package tool
import (
"context"
"fmt"
"log"
"reflect"
"runtime"
)
var PoolLimit chan struct{}
var XC *xc
type xc struct{}
/*
InitXcPond main函数时需初始化
num 协程限制数量
*/
func InitXcPond(num int) {
PoolLimit = make(chan struct{}, num)
XC = &xc{}
}
// Go 开始协程 使用上下文来传递函数需要处理的变量
func (x *xc) Go(f func(c context.Context), ctx context.Context) {
log.Println(fmt.Sprintf("当前：%s 正在使用协程,总协程数为：%d", runtime.FuncForPC(reflect.ValueOf(f).Pointer()).Name(), runtime.NumGoroutine()))
f(ctx)
<-PoolLimit
return
}
使用:
package test
import (
"common/tool"
"context"
"fmt"
"testing"
)
func init() {
tool.InitXcPond(20)
}
func TestPool(t *testing.T) {
for i:=0;i<=100;i++ {
c := context.Background()
c = context.WithValue(c,"myKey",i)
tool.PoolLimit <- struct{}{}
go tool.XC.Go(myTest,c)
}
}
func myTest(c context.Context) {
fmt.Println("key",c.Value("myKey"))
}
结果:协程池中协程维持在20个，其他5个是没有进入协程池的协程
image.png
