golang 40行代码实现通用协程池
程序员文章站
2022-07-02 12:00:14
代码仓库
golang的协程管理
golang协程机制很方便的解决了并发编程的问题,但是协程并不是没有开销的,所以也需要适当限制一下数量。
不使用协程池的代码(示...
代码仓库
golang的协程管理
golang协程机制很方便的解决了并发编程的问题,但是协程并不是没有开销的,所以也需要适当限制一下数量。
不使用协程池的代码(示例代码使用chan实现,代码略啰嗦)
func (p *converter) upload(bytes [][]byte) ([]string, error) { ch := make(chan struct{}, 4) wg := &sync.waitgroup{} wg.add(len(bytes)) ret := make([]string, len(bytes)) // 上传 for index, item := range bytes { ch <- struct{}{} go func(index int, imagedata []byte) { defer func() { wg.done() <-ch }() link, err := qiniu.uploadbinary(imagedata, fmt.sprintf("%d.png", time.now().unixnano())) if err != nil { log.println("上传图片失败", err.error()) return } ret[index] = link }(index, item) } wg.wait() return ret, nil }
需要实现的需求有两个:
限制最大协程数,本例为4
等待所有协程完成,本例为bytes切片长度
使用协程池的代码
func (p *converter) upload(bytes [][]byte) ([]string, error) { ret := make([]string, len(bytes)) pool := goroutine_pool.new(4, len(bytes)) for index, item := range bytes { index := index item := item pool.submit(func() { link, err := qiniu.uploadbinary(item, fmt.sprintf("%d.png", time.now().unixnano())) if err != nil { log.println("上传图片失败", err.error()) return } ret[index] = link }) } pool.wait() return ret, nil }
可以看到最大的区别是只需要关注业务逻辑即可,并发控制和等待都已经被协程池接管
总结
以上所述是小编给大家介绍的golang 40行代码实现通用协程池,希望对大家有所帮助
上一篇: MYSQL中的COLLATE是什么?