首頁  >  文章  >  後端開發  >  go語言中控制併發數量的方法

go語言中控制併發數量的方法

尚
轉載
2020-06-17 17:48:473339瀏覽

go語言中控制併發數量的方法

因為go語言層面支持並發,所以面試中常會問到並發的問題,比如說控制go並發數量的方式有哪些?以下是我個人整理的兩個例子:

func waitGroup() {
    count := 10
    wg := sync.WaitGroup{}

    for i := 0; i < count; i++ {
        wg.Add(1)
        go func(j int) {
            fmt.Print(j)
            wg.Done() // 也可使用 wg.Add(-1)
        }(i)
    }

    wg.Wait()
}

上面主要用到的是go中sync包下的waitGroup,這也是在工作中比較常見的實現方式,關鍵點就是把握好Add方法的位置,Wait方法則是等待所有的協程執行完畢

func channel() {
    count := 10 // 最大支持并发
    sum := 100 // 任务总数

    c := make(chan struct{}, count) // 控制任务并发的chan
    sc := make(chan struct{}, sum) // 控制任务总数的chan
    defer close(c)
    defer close(sc)

    for i:=0; i<sum;i++{
        c <- struct{}{} // 作用类似于waitgroup.Add(1)
        go func(j int) {
            fmt.Println(j)
            <- c // 执行完毕,释放资源
            sc <- struct {}{} // 记录到执行总数里
        }(i)
    }

    for i:=sum; i>0;i-- {
        <- sc
    }
}

上面的例子用到的是go中的channel,利用channel阻塞的特性和帶緩衝的channel來實現控制並發數量,其中sc這個channel是可以去掉的,例子裡用只是為了防止主程式退出之後,沒有全部輸出,正常工作中,程式一般都是阻塞式的,所以可以去掉。

更多相關知識請關注go語言教學欄位

以上是go語言中控制併發數量的方法的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:segmentfault.com。如有侵權,請聯絡admin@php.cn刪除