golang工作者goroutine池模型

qq_33204116 · · 2858 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

在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()
}