golang 40行代碼實現(xiàn)通用協(xié)程池
代碼倉庫
golang的協(xié)程管理
golang協(xié)程機制很方便的解決了并發(fā)編程的問題,但是協(xié)程并不是沒有開銷的,所以也需要適當(dāng)限制一下數(shù)量。
不使用協(xié)程池的代碼(示例代碼使用chan實現(xiàn),代碼略啰嗦)
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 }
需要實現(xiàn)的需求有兩個:
限制最大協(xié)程數(shù),本例為4
等待所有協(xié)程完成,本例為bytes切片長度
使用協(xié)程池的代碼
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 }
可以看到最大的區(qū)別是只需要關(guān)注業(yè)務(wù)邏輯即可,并發(fā)控制和等待都已經(jīng)被協(xié)程池接管
總結(jié)
以上所述是小編給大家介紹的golang 40行代碼實現(xiàn)通用協(xié)程池,希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回復(fù)大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持!
相關(guān)文章
go語言實現(xiàn)簡易比特幣系統(tǒng)錢包的原理解析
這篇文章主要介紹了go語言實現(xiàn)簡易比特幣系統(tǒng)錢包的原理解析,本文給大家介紹的非常詳細,對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2021-04-04

Go語言實現(xiàn)LRU算法的核心思想和實現(xiàn)過程