迹忆客 专注技术分享

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

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

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

package main

import (  
    "fmt"
    "sync"
    "time"
)

func process(i int, wg *sync.WaitGroup) {  
    fmt.Println("started Goroutine ", i)
    time.Sleep(2 * time.Second)
    fmt.Printf("Goroutine %d ended\n", i)
    wg.Done()
}

func main() {  
    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 将被解锁。

传递 wg 的指针很重要。如果没有传递指针，那么每个 Goroutine 都会有自己的 WaitGroup 副本，并且当它们完成执行时不会通知主协程（main）。

上面程序执行结果如下

每个人的输出可能都不同，因为 Goroutines 的执行顺序可能会有所不同:)。