golang协程池

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

type GoroutinePoll struct {
    Queue          chan func() error
    Total, Num     int
    Result         chan error
    FinishCallBack func() error
}

func (g *GoroutinePoll) Init(total int) {
    g.Total = total
    g.Queue = make(chan func() error, total)
    g.Result = make(chan error, total)
}
func (g *GoroutinePoll) SetFinishCallBack(f func() error) {
    g.FinishCallBack = f
}

func (g *GoroutinePoll) Add(f func() error) error {
    if len(g.Queue) == g.Total || g.Total == 0 {
        return errors.New(" add goroutine fail!")
    }

    g.Queue <- f
    g.Num++
    return nil
}
func (g *GoroutinePoll) Start() {
    for i := 0; i < g.Num; i++ {
        v, ok := <-g.Queue
        if !ok {
            break
        }

        go func() {
            err := v()
            g.Result <- err
        }()
    }

    for i := 0; i < g.Num; i++ {
        err, ok := <-g.Result
        if !ok {
            break
        }
        if err != nil {
            fmt.Println(err)
        }
    }

    if err := g.FinishCallBack(); err != nil {
        fmt.Println("执行回调函数失败!", err)
    }
}

func (g *GoroutinePoll) Close() {
    close(g.Queue)
    close(g.Result)
}
func main() {
    runtime.GOMAXPROCS(16)
    total := 10
    g := GoroutinePoll{}
    g.Init(total)
    finshCallBack := func() error {
        fmt.Println("callback !")
        return nil
    }
    g.SetFinishCallBack(finshCallBack)
    f1 := func() error {
        for i := 0; i < Max; i++ {

        }

        return nil
    }
    g.Add(f1)
    f2 := func() error {
        for i := 0; i < Max; i++ {

        }
        return nil
    }
    g.Add(f2)
    f3 := func() error {
        for i := 0; i < Max; i++ {

        }
        return nil
    }
    g.Add(f3)
    f4 := func() error {
        for i := 0; i < Max; i++ {

        }
        return nil
    }
    g.Add(f4)
    now := time.Now()
    g.Start()
    g.Close()
    fmt.Println(time.Since(now))
    now = time.Now()
    f1()
    f2()
    f3()
    f4()
    fmt.Println(time.Since(now))
}

callback !
3.863554ms
12.890933ms


有疑问加站长微信联系

本文来自:博客园

感谢作者:fwdqxl

查看原文:golang协程池

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

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