Go WaitGroup 用法详解

Z时代
2024-01-10
分类：综合

在下一篇我们来介绍Go 工作池。要了解工作池，我们首先需要了解 WaitGroup，因为它将在工作池的实现中使用。

WaitGroup 用于等待一组 Goroutine 完成执行。控制被阻塞，直到所有 Goroutines 执行完毕。假设我们有 3 个并发的从主 Goroutine 衍生的 Goroutine。主 Goroutines 需要等待其他 3 个 Goroutines 完成才能终止。这可以使用 WaitGroup 来完成。

下面我们通过一个示例来看一下WaitGroup的用法


package main
import (  
"fmt"
"sync"
"time"
)
funcprocess(i int, wg *sync.WaitGroup) {  
    fmt.Println("started Goroutine ", i)
    time.Sleep(2 * time.Second)
    fmt.Printf("Goroutine %d ended\n", i)
    wg.Done()
}
funcmain() {  
    no := 3
var wg sync.WaitGroup
for i := 0; i < no; i++ {
        wg.Add(1)
go process(i, &wg)
    }
    wg.Wait()
    fmt.Println("所有 routines 完成了执行!")
}

运行示例

WaitGroup 是一个结构体类型，我们创建了一个 WaitGroup 类型的零值变量。 WaitGroup 的工作方式是使用计数器。当我们在 WaitGroup 上调用 Add() 并向它传递一个 int 时，WaitGroup 的计数器会根据传递给 Add 的值递增。递减计数器的方法是在 WaitGroup 上调用 Done() 方法。 Wait() 方法会阻塞调用它的 Goroutine，直到计数器变为零。

在上面的程序中，我们在迭代 3 次的 for 循环中调用 wg.Add(1)。所以计数器现在变成了 3。 for 循环还产生了 3 个 process Goroutines。然后调用 wg.Wait() 使主 Goroutine 等待直到计数器变为零。计数器通过在process 协程中调用 wg.Done 递减。一旦所有 3 个生成的 Goroutine 执行完毕，即 wg.Done() 被调用 3 次，计数器将变为 0，主 Goroutine 将被解锁。