golang40行代碼實現通用協程池

代碼倉庫

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}

需要實現的需求有兩個:

1. 限制最大協程數，本例為4
2. 等待所有協程完成，本例為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}

可以看到最大的區別是只需要關注商務邏輯即可，並發控制和等待都已經被協程池接管

寫在最後

希望本文能減輕你控制協程的負擔