使用channel控制協(xié)程數(shù)量

協(xié)程

goroutine 是輕量級線程，調(diào)度由 Go 運(yùn)行時進(jìn)行管理的。Go 語言的并發(fā)控制主要使用關(guān)鍵字 go 開啟協(xié)程 goroutine。Go 協(xié)程（Goroutine）之間通過信道（channel）進(jìn)行通信，簡單的說就是多個協(xié)程之間通信的管道。信道可以防止多個協(xié)程訪問共享內(nèi)存時發(fā)生資源爭搶的問題。語法格式：

// 普通函數(shù)創(chuàng)建 goroutinego 函數(shù)名(參數(shù)列表)
//匿名函數(shù)創(chuàng)建 goroutinego func(參數(shù)列表){    //函數(shù)體}(調(diào)用參數(shù)列表)

協(xié)程可以開啟多少個？是否有限制呢？

func testRoutine() {    var wg sync.WaitGroup    for i := 0; i < math.MaxInt32; i++ {        wg.Add(1)        go func(i int) {            defer wg.Done()            fmt.Printf("并發(fā)數(shù)量：%d/n", i)            time.Sleep(time.Second)        }(i)    }    wg.Wait()}

以上代碼開啟了 math.MaxInt32個協(xié)程的并發(fā)，執(zhí)行后可以看到結(jié)果直接 panic：“panic: too many concurrent operations on a single file or socket (max 1048575)”。整個并發(fā)操作超出了系統(tǒng)最大值。

控制協(xié)程數(shù)量

sync 同步機(jī)制

使用 sync.WaitGroup 啟動指定數(shù)量的協(xié)程 goroutine。

func testRoutine() {    var wg = sync.WaitGroup{}
    taskCount := 5 // 指定并發(fā)數(shù)量    for i := 0; i < taskCount; i++ {        wg.Add(1)        go func(i int) {            fmt.Println("go func ", i)            wg.Done()        }(i)    }    wg.Wait()}

如果 taskcount 設(shè)置的很大超出了限制的，則其還是沒有控制到并發(fā)數(shù)量。可以優(yōu)化下設(shè)計，類似池的設(shè)計思想，通過允許最大連接數(shù)控制量，當(dāng)超出了數(shù)量就需要等待釋放，有空閑的連接的時候才可以繼續(xù)執(zhí)行。

func testRoutine() {    task_chan := make(chan bool, 3) //100 為 channel長度    wg := sync.WaitGroup{}    defer close(task_chan)    for i := 0; i < math.MaxInt; i++ {        wg.Add(1)        fmt.Println("go func ", i)        task_chan <- true        go func() {                <-task_chan                defer wg.Done()        }()    }
    wg.Wait()}

• 創(chuàng)建緩沖區(qū)大小為 3 的 channel，在沒有被接收的情況下，至多發(fā)送 3 個消息則被阻塞。通過 channel 控制每次并發(fā)的數(shù)量。
• 開啟協(xié)程前，設(shè)置 task_chan <- true，若緩存區(qū)滿了則阻塞
• 協(xié)程任務(wù)執(zhí)行完成后就釋放緩沖區(qū)
• 等待所有的并發(fā)都處理結(jié)束后則函數(shù)結(jié)束。其實(shí)可以不使用 sync.WaitGroup。因使用 channel 控制并發(fā)處理的任務(wù)數(shù)量可以不用使用等待并發(fā)處理結(jié)束。