什么是WaitGroups

WaitGroups是同步你的goroutines的一種有效方式。想象一下，你和你的家人一起駕車(chē)旅行。你的父親在一個(gè)條形商場(chǎng)或快餐店停下來(lái)，買(mǎi)些食物和上廁所。你最好想等大家回來(lái)后再開(kāi)車(chē)去地平線(xiàn)。WaitGroups幫助你做到這一點(diǎn)。

WaitGroups是通過(guò)調(diào)用標(biāo)準(zhǔn)庫(kù)中的sync包來(lái)定義的。

var wg sync.WaitGroup

那么，什么是WaitGroup呢？WaitGroup是一個(gè)結(jié)構(gòu)，它包含了程序需要等待多少個(gè)goroutine的某些信息。它是一個(gè)包含你需要等待的goroutines數(shù)量的組。

WaitGroups有三個(gè)最重要的方法: Add, Done和 Wait。

• Add: 添加到你需要等待的goroutines的總量上。
• Done: 從你需要等待的goroutines總數(shù)中減去一個(gè)。
• Wait: 阻止代碼繼續(xù)進(jìn)行，直到?jīng)]有更多的goroutines需要等待。

如何使用WaitGroups

讓我們來(lái)看看一段代碼：

package main

import (
    "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 start
2022-08-21 17:01:55.184932851 +0900 KST m=+1.000210473 done
2022-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為空。我們通過(guò)調(diào)用wg.Wait()來(lái)做到這一點(diǎn)。

為什么使用WaitGroups而不是channel

現(xiàn)在我們知道了如何使用WaitGroups，一個(gè)自然而然的想法將我們引向這個(gè)問(wèn)題：為什么使用WaitGroups而不是通道？

根據(jù)我的經(jīng)驗(yàn)，有幾個(gè)原因。

• WaitGroups往往更直觀(guān)。當(dāng)你閱讀一段代碼時(shí)，當(dāng)你看到一個(gè)WaitGroup時(shí)，你會(huì)立即知道代碼在做什么。方法的名稱(chēng)很明確，而且直奔主題。然而，對(duì)于通道來(lái)說(shuō)，有時(shí)就不是那么清楚了。使用通道是很聰明的，但當(dāng)你閱讀一段復(fù)雜的代碼時(shí)，理解起來(lái)會(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并沒(méi)有與其他goroutine進(jìn)行數(shù)據(jù)交流。如果你的goroutine是一次性的工作，你不需要知道結(jié)果，使用WaitGroup是可取的?，F(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是專(zhuān)門(mén)用來(lái)處理等待goroutines的。我覺(jué)得通道的主要目的是為了交流數(shù)據(jù)。你不能用WaitGroup來(lái)發(fā)送和接收數(shù)據(jù)，但你可以用一個(gè)channel來(lái)同步你的goroutines。

最后，沒(méi)有正確的答案。我知道這可能很煩人，但這取決于你和你工作的團(tuán)隊(duì)。無(wú)論什么方法都是最好的，沒(méi)有答案是錯(cuò)誤的。我個(gè)人傾向于使用WaitGroups進(jìn)行同步，但你的情況可能有所不同。選擇對(duì)你來(lái)說(shuō)最直觀(guān)的東西。

需要注意的一件事

有時(shí)，你可能需要將WaitGroup實(shí)例傳遞給goroutine?？赡苡袔讉€(gè)WaitGroup來(lái)處理不同的goroutine，也可能是一種設(shè)計(jì)選擇。不管是什么原因，請(qǐng)確保傳遞指向WaitGroup的指針，像這樣：

var wg sync.WaitGroup

for 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中。

總結(jié)

通過(guò)使用WaitGroups，我們可以輕松同步goroutines，從而確保我們的代碼在正確的時(shí)間執(zhí)行。盡管通道也可以用于同步，但WaitGroups通常更直觀(guān)且更易于閱讀。在使用WaitGroup時(shí)，請(qǐng)確保正確傳遞指向WaitGroup的指針，以防止出現(xiàn)副本問(wèn)題。無(wú)論您選擇哪種方法，都應(yīng)該選擇最直觀(guān)和最適合您和您的團(tuán)隊(duì)的方法。

到此這篇關(guān)于一文帶你了解Golang中的WaitGroups的文章就介紹到這了,更多相關(guān)Golang WaitGroups內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論