代碼倉庫
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行代碼實現通用協程池，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對VEVB武林網網站的支持！
新聞熱點
疑難解答
