什么是WaitGroups?下面本篇文章就來帶大家了解一下go語(yǔ)言中WaitGroups,介紹一下使用WaitGroups的方法,希望對(duì)大家有所幫助!
【資料圖】
WaitGroups是同步你的goroutines的一種有效方式。想象一下,你和你的家人一起駕車旅行。你的父親在一個(gè)條形商場(chǎng)或快餐店停下來,買些食物和上廁所。你最好想等大家回來后再開車去地平線。WaitGroups幫助你做到這一點(diǎn)。
WaitGroups是通過調(diào)用標(biāo)準(zhǔn)庫(kù)中的sync包來定義的。
var wg sync.WaitGroup
那么,什么是WaitGroup呢?WaitGroup是一個(gè)結(jié)構(gòu),它包含了程序需要等待多少個(gè)goroutine的某些信息。它是一個(gè)包含你需要等待的goroutines數(shù)量的組。
WaitGroups有三個(gè)最重要的方法: Add, Done和 Wait。
讓我們來看看一段代碼:
package mainimport ( "fmt" "sync" "time")func main() { var wg sync.WaitGroup wg.Add(1) go func() { defer wg.Done() fmt.Println(time.Now(), "start") time.Sleep(time.Second) fmt.Println(time.Now(), "done") }() wg.Wait() fmt.Println(time.Now(), "exiting...")}2022-08-21 17:01:54.184744229 +0900 KST m=+0.000021800 start2022-08-21 17:01:55.184932851 +0900 KST m=+1.000210473 done2022-08-21 17:01:55.18507731 +0900 KST m=+1.000354912 exiting...
我們首先初始化一個(gè)WaitGroup wg的實(shí)例。然后我們?cè)?code>wg中添加1,因?yàn)槲覀円却粋€(gè)goroutine完成。然后我們運(yùn)行這個(gè)goroutine。在goroutine內(nèi)部,我們對(duì)wg.Done()進(jìn)行延遲調(diào)用,以確保我們遞減要等待的goroutine的數(shù)量。如果我們不這樣做,那么代碼將永遠(yuǎn)等待goroutine完成,并將導(dǎo)致死鎖。在goroutine調(diào)用之后,我們要確保阻斷代碼,直到WaitGroup為空。我們通過調(diào)用wg.Wait()來做到這一點(diǎn)。現(xiàn)在我們知道了如何使用WaitGroups,一個(gè)自然而然的想法將我們引向這個(gè)問題:為什么使用WaitGroups而不是通道?
根據(jù)我的經(jīng)驗(yàn),有幾個(gè)原因。
WaitGroups往往更直觀。當(dāng)你閱讀一段代碼時(shí),當(dāng)你看到一個(gè)WaitGroup時(shí),你會(huì)立即知道代碼在做什么。方法的名稱很明確,而且直奔主題。然而,對(duì)于通道來說,有時(shí)就不是那么清楚了。使用通道是很聰明的,但當(dāng)你閱讀一段復(fù)雜的代碼時(shí),理解起來會(huì)很麻煩。有的時(shí)候,你不需要使用通道。例如,讓我們看一下這段代碼:var wg sync.WaitGroup for i := 0; i < 5; i++ { wg.Add(1) go func() { defer wg.Done() fmt.Println(time.Now(), "start") time.Sleep(time.Second) fmt.Println(time.Now(), "done") }() } wg.Wait() fmt.Println(time.Now(), "exiting...")你可以看到,這個(gè)goroutine并沒有與其他goroutine進(jìn)行數(shù)據(jù)交流。如果你的goroutine是一次性的工作,你不需要知道結(jié)果,使用WaitGroup是可取的。現(xiàn)在看一下這段代碼:
ch := make(chan int) for i := 0; i < 5; i++ { go func() { randomInt := rand.Intn(10) ch <- randomInt }() } for i := 0; i < 5; i++ { fmt.Println(<-ch) }這里,goroutine正在向 channel發(fā)送數(shù)據(jù)。在這些情況下,我們不需要使用WaitGroup,因?yàn)檫@將是多余的。如果接收已經(jīng)做了足夠的阻塞,為什么還要等待goroutine完成?
WaitGroups是專門用來處理等待goroutines的。我覺得通道的主要目的是為了交流數(shù)據(jù)。你不能用WaitGroup來發(fā)送和接收數(shù)據(jù),但你可以用一個(gè)channel來同步你的goroutines。
最后,沒有正確的答案。我知道這可能很煩人,但這取決于你和你工作的團(tuán)隊(duì)。無(wú)論什么方法都是最好的,沒有答案是錯(cuò)誤的。我個(gè)人傾向于使用WaitGroups進(jìn)行同步,但你的情況可能有所不同。選擇對(duì)你來說最直觀的東西。
有時(shí),你可能需要將WaitGroup實(shí)例傳遞給goroutine??赡苡袔讉€(gè)WaitGroup來處理不同的goroutine,也可能是一種設(shè)計(jì)選擇。不管是什么原因,請(qǐng)確保傳遞指向WaitGroup的指針,像這樣:
var wg sync.WaitGroupfor i := 0; i < 5; i++ { wg.Add(1) go func(wg *sync.WaitGroup) { defer wg.Done() fmt.Println(time.Now(), "start") time.Sleep(time.Second) fmt.Println(time.Now(), "done") }(&wg)}wg.Wait()fmt.Println(time.Now(), "exiting...")原因是Go是一種值傳遞的語(yǔ)言。這意味著每當(dāng)你向一個(gè)函數(shù)傳遞一個(gè)參數(shù)時(shí),Go會(huì)復(fù)制一個(gè)參數(shù)并傳遞給它而不是原始對(duì)象。在這種情況下發(fā)生的是,整個(gè)WaitGroup對(duì)象將被復(fù)制,這意味著goroutine將處理一個(gè)完全不同的WaitGroup。wg.Done()不會(huì)從原始的wg中減去,而是減去它的一個(gè)副本,這個(gè)副本只存在于goroutine中。
通過使用WaitGroups,我們可以輕松同步goroutines,從而確保我們的代碼在正確的時(shí)間執(zhí)行。盡管通道也可以用于同步,但WaitGroups通常更直觀且更易于閱讀。在使用WaitGroup時(shí),請(qǐng)確保正確傳遞指向WaitGroup的指針,以防止出現(xiàn)副本問題。無(wú)論您選擇哪種方法,都應(yīng)該選擇最直觀和最適合您和您的團(tuán)隊(duì)的方法。
推薦學(xué)習(xí):Golang教程
以上就是go語(yǔ)言中什么是WaitGroups?怎么使用?的詳細(xì)內(nèi)容,更多請(qǐng)關(guān)注php中文網(wǎng)其它相關(guān)文章!
關(guān)鍵詞: