場(chǎng)景

Avito是一名校車(chē)司機(jī)，他幫助4個(gè)Gopher孩子上學(xué)。每天，Avito在他們的社區(qū)等待孩子們。他不知道孩子們需要多長(zhǎng)時(shí)間，但他確切地知道有4個(gè)孩子他需要等待。

1*aZnEggopv4Tsbyyj3e5JFg.png

當(dāng)一個(gè)孩子準(zhǔn)備好時(shí)，他/她會(huì)說(shuō)：Done()，將計(jì)數(shù)器減1。Avito仍然被阻塞，因?yàn)橛?jì)數(shù)器仍然大于0。他必須等到所有其他孩子準(zhǔn)備好。

1*qouGWmMAqY2CDrzz5widhQ.png

如果有兩個(gè)孩子同時(shí)準(zhǔn)備好，它們的同時(shí)準(zhǔn)備會(huì)導(dǎo)致WaitGroup出現(xiàn)不一致嗎？絕對(duì)不會(huì)。與sync包中的大多數(shù)其他組件一樣，WaitGroup具有內(nèi)置的同步機(jī)制，以處理并發(fā)。因此，計(jì)數(shù)器減少了準(zhǔn)備好的孩子數(shù)量。

1*057bX4zo_LCzEkdzGyDYpA.png

在最后一個(gè)孩子準(zhǔn)備好后，Avito啟動(dòng)引擎，將他們送到學(xué)校。

1*rjH8OR3t7QgUx-dO-Iszeg.png

就是這樣！正如我所說(shuō)，sync.WaitGroup很簡(jiǎn)單。

超時(shí)

如果一個(gè)孩子花費(fèi)太多時(shí)間準(zhǔn)備，他們會(huì)不會(huì)因此遲到？如果Avito在時(shí)間到達(dá)時(shí)不管怎樣都開(kāi)始行駛會(huì)更好嗎？嗯，Golang傾向于保持一切盡可能簡(jiǎn)潔，因此與其他編程語(yǔ)言中的CountDownLatch（例如Java中的）不同，sync.WaitGroup默認(rèn)情況下不支持超時(shí)。在這種情況下，選擇語(yǔ)句可能會(huì)有所幫助。

func main() {
    wg := sync.WaitGroup{}
    wg.Add(1)

    go func() {
        defer wg.Done()
        time.Sleep(5 * time.Second)
    }()

    done := make(chan bool)

    go func() {
        wg.Wait()
        close(done)
    }()

    select {
    case <-done:
        log.Println("All done")
    case <-time.After(1 * time.Second):
        log.Println("Hit timeout")
    }
}

孩子等待

在上面的示例中，Avito（主Goroutine）等待孩子（子Goroutines）。當(dāng)我們希望子Goroutines等待主Goroutine時(shí)，WaitGroup也可以使用。想象一下孩子們正在進(jìn)行體育課。Torcher - 體育老師，在學(xué)生中主持比賽。他向WaitGroup中Add(1)，并要求所有孩子在相同的WaitGroup上Wait()。

1*btDQK4QKsu1HkEpfJDa2EA.png

當(dāng)Torcher調(diào)用wg.Done()時(shí)，計(jì)數(shù)器變?yōu)?，允許所有孩子同時(shí)開(kāi)始奔跑。

1*VkV3VlRTx5jxXhauBH0_Dg.png

展示你的代碼！

package main

import (
    "log"
    "sync"
    "time"
)

func main() {
    kids := []string{"Partier", "Stringer", "Candier", "Swimmer"}
    wg := sync.WaitGroup{}
    wg.Add(len(kids))

    for _, kid := range kids {
        go func(name string) {
            defer wg.Done()
            prepare(name)
        }(kid)
    }

    log.Printf("Avito: I'm waiting for %d kids\n", len(kids))
    wg.Wait()
    log.Println("Avito: The kids are all ready, go!")
}

func prepare(name string) {
    log.Printf("%v: I'm preparing for school\n", name)
    time.Sleep(2 * time.Second)
    log.Printf("%v: I'm ready\n", name)
}