golang工作者goroutine池模型

qq_33204116 · · 2840 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

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

有疑问加站长微信联系(非本文作者)

本文来自:CSDN博客

感谢作者:qq_33204116

查看原文:golang工作者goroutine池模型

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

2840 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传