快捷導(dǎo)航

golang基礎(chǔ)之waitgroup用法以及使用要點(diǎn)

更新時(shí)間：2023年01月07日 14:21:19 作者：北漂燕郊楊哥

WaitGroup是Golang并發(fā)的兩種方式之一,一個(gè)是Channel,另一個(gè)是WaitGroup,下面這篇文章主要給大家介紹了關(guān)于golang基礎(chǔ)之waitgroup用法以及使用要點(diǎn)的相關(guān)資料,需要的朋友可以參考下

一、前言

waitgroup在golang中，用于線程同步，指等待一個(gè)組，等待一個(gè)系列執(zhí)行完成后，才會向下執(zhí)行,可以解決一個(gè) 進(jìn)程goroutine 等待多個(gè)該進(jìn)程啟動(dòng)的子線程goroutine 都正常運(yùn)行完成的場景，這個(gè)比較常見的場景就是例如后端 main processer 啟動(dòng)了多個(gè)消費(fèi)者worker干活，還有爬蟲并發(fā)爬取數(shù)據(jù)，多線程下載等等,為了保證主進(jìn)程在所有的子線程完成后再退出，這時(shí)就要用上waitgroup

二、waitgroup使用示例

我們這里模擬一個(gè) worker 的例子

package main

import (
    "fmt"
    "sync"
    "time"
)
func worker(idx int, out chan struct{}, wg *sync.WaitGroup) {
	 defer wg.Done()
        time.Sleep(1 * time.Second)
        fmt.Println(time.Now())
        fmt.Println(idx )
        <-out 
}
func main() {
    wg := new(sync.WaitGroup)
    in := make(chan struct{}, 20)
    for i := 0; i < 200; i++ {
        in <- struct{}{}
        wg.Add(1)
        go worker(i, in, wg)
    }
    wg.Wait()
}

在這段代碼中，main最后一行是

wg.Wait()

這行代碼保證有所的200個(gè)子線程全部都執(zhí)行完成后才會退出main函數(shù)，如果沒有最后一行wg.Wait()，可能會出現(xiàn)for循環(huán)遍歷完程序就直接退出了，有可能只有不確定個(gè)幾個(gè)子線程執(zhí)行完成，其它線程由于主程序main退出后就直接退出了

從這個(gè)例子我們也可以看到 waitgroup通常配合來限制并發(fā)線程個(gè)數(shù)和確保所有的線程都最終都執(zhí)行完成
這段代碼中 ws 有三個(gè)緩沖，所以并發(fā)的數(shù)量是20，超過20個(gè)就要等待執(zhí)行完成釋放所占用通道后才能再開新的線程

上面的代碼，會在執(zhí)行到wg.Wait()后等待，直到所有的200線程全部執(zhí)行完后才會繼續(xù)往下執(zhí)行

可以説這段代碼非常精妙，示例代碼執(zhí)行結(jié)果如下:

同時(shí)我們也可以測試一下把最后一行

//wg.Wait()

注釋掉，我們看一下程序會怎么執(zhí)行

注釋后，我們看到程序完成時(shí)，只有171個(gè)線程完成運(yùn)行，剩下的20幾個(gè)線程異常結(jié)束了，看不到任何返回結(jié)果

三、waitgroup使用注意事項(xiàng)

同時(shí)我們在使用waitgroup時(shí)也要注意一些坑：

1、 Add一個(gè)負(fù)數(shù)

如果計(jì)數(shù)器的值小于0會直接panic

2、 Add在Wait之后調(diào)用

比如一些子協(xié)程開頭調(diào)用Add結(jié)束調(diào)用Wait，這些 Wait無法阻塞子協(xié)程。正確做法是在開啟子協(xié)程之前先Add特定的值。

3、未置為0就重用

WaitGroup可以完成一次編排任務(wù)，計(jì)數(shù)值降為0后可以繼續(xù)被其他任務(wù)所用，但是不要在還沒使用完的時(shí)候就用于其他任務(wù)，這樣由于帶著計(jì)數(shù)值，很可能出問題。

4、復(fù)制waitgroup

WaitGroup有nocopy字段，不能被復(fù)制。也意味著WaitGroup不能作為函數(shù)的參數(shù)

四、waitgroup使用總結(jié)

WaitGroup是Golang應(yīng)用開發(fā)過程中經(jīng)常使用的并發(fā)控制技術(shù)，學(xué)習(xí)golang是我們必須要掌握和理解的機(jī)制之一，建議有時(shí)間了大家再進(jìn)一步的研究一下WaitGroup的底層實(shí)現(xiàn)邏輯。