在golang编写服务程序过程中,如果每次来斗启动一个goroutine去处理任务,处理一个任务后就退出,这样势必会造成资源浪费。构建一个工作goroutine池来处理任务相对资源利用会少些,具体的情况需对比测试。
首先定义工作goroutine池,内部定义了两个变量,一个是任务队列,一个是需启动goroutine的数量
type WorkerPool struct {
tasks <-chan *string //任务队列长度
poolSize int //启动goroutine的数目
}
实现工作函数
func (p *WorkerPool) Run() {
var wg sync.WaitGroup
for i := 0; i < p.poolSize; i++ {
wg.Add(1)
go func() {
for task := range p.tasks {
fmt.Println("Consume Task", *task)
}
wg.Done()
}()
}
wg.Wait()
fmt.Println("WorkerPool | Pool exit.")
}
以下是测试代码
func main() {
taskNum := 100
projects := make(chan *string, taskNum)
//启动生产任务goroutine
go func() {
for i := 0; ; i++ {
s := "Project " + fmt.Sprintf("%d", i)
fmt.Println("Produce task", s)
projects <- &s
time.Sleep(5 * time.Millisecond)
}
}()
p := WorkerPool{
tasks: projects,
poolSize: 10,
}
p.Run()
}
有疑问加站长微信联系(非本文作者)