golang 40行代碼實(shí)現(xiàn)通用協(xié)程池
代碼倉庫
golang的協(xié)程管理
golang協(xié)程機(jī)制很方便的解決了并發(fā)編程的問題,但是協(xié)程并不是沒有開銷的,所以也需要適當(dāng)限制一下數(shù)量。
不使用協(xié)程池的代碼(示例代碼使用chan實(shí)現(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 }
需要實(shí)現(xiàn)的需求有兩個(gè):
限制最大協(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行代碼實(shí)現(xiàn)通用協(xié)程池,希望對(duì)大家有所幫助,如果大家有任何疑問請(qǐng)給我留言,小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持!
相關(guān)文章
go語言實(shí)現(xiàn)簡(jiǎn)易比特幣系統(tǒng)錢包的原理解析
這篇文章主要介紹了go語言實(shí)現(xiàn)簡(jiǎn)易比特幣系統(tǒng)錢包的原理解析,本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2021-04-04簡(jiǎn)單高效!Go語言封裝二級(jí)認(rèn)證功能實(shí)現(xiàn)
本文將介紹如何使用Go語言封裝二級(jí)認(rèn)證功能,實(shí)現(xiàn)簡(jiǎn)單高效的用戶認(rèn)證流程,二級(jí)認(rèn)證是一種安全措施,要求用戶在登錄后進(jìn)行額外的身份驗(yàn)證,以提高賬戶安全性,2023-10-10

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

jenkins構(gòu)建go及java項(xiàng)目的方法

GO使用阿里云,解決go get下載項(xiàng)目慢或無法下載的情況