[TOC] ## 方法 ### 利用 channel 的緩存區 可以控制協程最多創建三個 ``` // main_chan.go func main() { var wg sync.WaitGroup ch := make(chan struct{}, 3) for i := 0; i < 10; i++ { ch <- struct{}{} wg.Add(1) go func(i int) { defer wg.Done() log.Println(i) time.Sleep(time.Second) <-ch }(i) } wg.Wait() } ``` * `make(chan struct{}, 3)`創建緩沖區大小為 3 的 channel，在沒有被接收的情況下，至多發送 3 個消息則被阻塞。 * 開啟協程前，調用`ch <- struct{}{}`，若緩存區滿，則阻塞。 * 協程任務結束，調用`<-ch`釋放緩沖區。 * `sync.WaitGroup`并不是必須的，例如 http 服務，每個請求天然是并發的，此時使用 channel 控制并發處理的任務數量，就不需要`sync.WaitGroup` ### 利用第三方庫 * [Jeffail/tunny](https://github.com/Jeffail/tunny) * [panjf2000/ants](https://github.com/panjf2000/ants) 以 tunny 為例 ``` package main import ( "log" "time" "github.com/Jeffail/tunny" ) func main() { pool := tunny.NewFunc(3, func(i interface{}) interface{} { log.Println(i) time.Sleep(time.Second) return nil }) defer pool.Close() for i := 0; i < 10; i++ { go pool.Process(i) } time.Sleep(time.Second * 4) } ``` * `tunny.NewFunc(3, f)`第一個參數是協程池的大小(poolSize)，第二個參數是協程運行的函數(worker)。 * `pool.Process(i)`將參數 i 傳遞給協程池定義好的 worker 處理。 * `pool.Close()`關閉協程池。