cover_image

Go并发可视化解释 - sync.WaitGroup

小技术君 小技术君
2023年10月20日 01:06

场景

Avito是一名校车司机，他帮助4个Gopher孩子上学。每天，Avito在他们的社区等待孩子们。他不知道孩子们需要多长时间，但他确切地知道有4个孩子他需要等待。

图片
1*aZnEggopv4Tsbyyj3e5JFg.png

当一个孩子准备好时，他/她会说：Done()，将计数器减1。Avito仍然被阻塞，因为计数器仍然大于0。他必须等到所有其他孩子准备好。

图片
1*qouGWmMAqY2CDrzz5widhQ.png

如果有两个孩子同时准备好，它们的同时准备会导致WaitGroup出现不一致吗？绝对不会。与sync包中的大多数其他组件一样，WaitGroup具有内置的同步机制，以处理并发。因此，计数器减少了准备好的孩子数量。

图片
1*057bX4zo_LCzEkdzGyDYpA.png

在最后一个孩子准备好后，Avito启动引擎，将他们送到学校。

图片
1*rjH8OR3t7QgUx-dO-Iszeg.png

就是这样！正如我所说，sync.WaitGroup很简单。

超时

如果一个孩子花费太多时间准备，他们会不会因此迟到？如果Avito在时间到达时不管怎样都开始行驶会更好吗？嗯，Golang倾向于保持一切尽可能简洁，因此与其他编程语言中的CountDownLatch（例如Java中的）不同，sync.WaitGroup默认情况下不支持超时。在这种情况下，选择语句可能会有所帮助。

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）。当我们希望子Goroutines等待主Goroutine时，WaitGroup也可以使用。想象一下孩子们正在进行体育课。Torcher - 体育老师，在学生中主持比赛。他向WaitGroup中Add(1)，并要求所有孩子在相同的WaitGroup上Wait()。

图片
1*btDQK4QKsu1HkEpfJDa2EA.png

当Torcher调用wg.Done()时，计数器变为0，允许所有孩子同时开始奔跑。

图片
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)}


go语言8
go2
继续滑动看下一个
小技术君
向上滑动看下一个