Go语言学习之WaitGroup使用介绍

内容介绍

WaitGroup 对象内部有一个计数器，最初从0开始，它有三个方法：Add(), Done(), Wait() 用来控制计数器的数量。
Add(n) 把计数器设置为n
Done() 每次把计数器-1
Wait() 会阻塞代码的运行，直到计数器地值减为0。

场景说明

主线程等待所有线程执行结束再继续执行

反例

主线程为了等待goroutine都运行完毕，不得不在程序的末尾使用time.Sleep() 来睡眠一段时间，等待其他线程充分运行。对于简单的代码，100个for循环可以在1秒之内运行完毕，time.Sleep() 也可以达到想要的效果。

    import (
        "fmt"
        "time"
    )

    func main(){
        for i := 0; i < 100 ; i++{
            go fmt.Println(i)
        }
        time.Sleep(time.Second)
    }

正例

这里首先把 wg 计数设置为100， 每个for循环运行完毕都把计数器减一，主函数中使用 Wait() 一直阻塞，直到wg为零——也就是所有的100个for循环都运行完毕。相对于使用主线程睡眠来说 WaitGroup 更合理。

    func main() {
        wg := sync.WaitGroup{}
        wg.Add(100)
        for i := 0; i < 100; i++ {
            go func(i int) {
                fmt.Println(i)
                wg.Done()
            }(i)
        }
        wg.Wait()
    }

WaitGroup对象不是一个引用类型

WaitGroup对象不是一个引用类型，在通过函数传值的时候需要使用地址

    func main() {
        wg := sync.WaitGroup{}
        wg.Add(100)
        for i := 0; i < 100; i++ {
            go f(i, &wg)
        }
        wg.Wait()
    }

    // 一定要通过指针传值，不然进程会进入死锁状态
    func f(i int, wg *sync.WaitGroup) { 
        fmt.Println(i)
        wg.Done()
    }

Go 并发

https://www.runoob.com/go/go-concurrent.html

Golang sync.WaitGroup的用法

https://studygolang.com/articles/12972