golang waitgroup

在Golang中，我们经常需要处理同时运行的多个goroutine，并且需要等待它们全部执行完成后再进行下一步操作。为了解决这个问题，Golang提供了一个非常有用的同步机制——WaitGroup。

什么是WaitGroup

WaitGroup是一个用于等待一组goroutine完成的计数器。它主要有三个方法：Add、Done和Wait。

1. Add方法用来增加计数器的值，就像是给WaitGroup添加了一组goroutine。

2. Done方法用来减少计数器的值，就像是通知WaitGroup某个goroutine已经执行完毕。

3. Wait方法用来阻塞当前goroutine，直到计数器的值变为0。

使用WaitGroup的场景

WaitGroup在以下场景中非常有用：

1. 同时启动多个goroutine进行并行计算，并等待所有goroutine都执行完成后才进行后续操作。

2. 结合管道（Channel）和WaitGroup，实现多个goroutine协同工作的模型。

3. 控制并发访问资源，等待所有goroutine访问完毕后再关闭资源。

WaitGroup的基本用法

下面是一个使用WaitGroup的简单示例：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func(num int) {
            defer wg.Done()
            fmt.Println("goroutine", num)
        }(i)
    }
    wg.Wait()
    fmt.Println("All goroutines finished.")
}

在上面的示例中，我们使用了一个循环启动了5个goroutine，并使用WaitGroup来等待它们全部执行完毕。在每个goroutine中，我们使用了defer语句来在函数结束时调用Done方法，表示当前goroutine执行完毕。

注意事项

在使用WaitGroup时，需要注意以下几点：

1. 如果在调用Wait方法之前，计数器的值已经为0，调用Wait方法会立即返回，不会发生阻塞。

2. Add方法可以在任意位置调用，可以在多个goroutine中并发使用。

3. Done方法必须在计数器值减少之前调用，否则会引发panic。

4. 可以使用WaitGroup的传址方式使用，也可以直接声明一个WaitGroup类型的变量。

通过使用WaitGroup，我们可以很方便地控制多个goroutine的并发执行。它使得我们能够更好地组织和管理并发代码，提高程序的性能和可维护性。