作者:feify_fei512_478 | 来源:互联网 | 2023-05-16 03:33
在golang编写服务程序过程中,如果每次来斗启动一个goroutine去处理任务,处理一个任务后就退出,这样势必会造成资源浪费。构建一个工作goroutine池来处理任务相对资源利用会少些,具
在golang编写服务程序过程中,如果每次来斗启动一个goroutine去处理任务,处理一个任务后就退出,这样势必会造成资源浪费。构建一个工作goroutine池来处理任务相对资源利用会少些,具体的情况需对比测试。
首先定义工作goroutine池,内部定义了两个变量,一个是任务队列,一个是需启动goroutine的数量
type WorkerPool struct {
tasks <-chan *string
poolSize int
}
实现工作函数
func (p *WorkerPool) Run() {
var wg sync.WaitGroup
for i := 0; 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)
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()
}