代碼倉庫
goroutine-pool
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行代碼實現通用協程池，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對腳本之家網站的支持！
您可能感興趣的文章:- golang協程池模擬實現群發郵件功能
- golang協程池設計詳解
- GO實現協程池管理的方法
