golang waitgroup原理

Go语言是一门近年来越来越受欢迎的编程语言，它以其并发模型出名。并发模型是Go语言的一个核心特性，使得Go语言在处理高并发任务时非常高效。而在并发编程中，我们常常需要等待一组goroutine执行完毕后再继续执行下一步操作，这就需要用到WaitGroup（等待组）。

什么是WaitGroup?

WaitGroup是Go语言提供的一种并发同步机制。其作用是等待一组goroutine全部执行结束后，再继续执行主函数。它内部维护了一个计数器，可以通过Add方法来设置需要等待的goroutine数量，每个goroutine执行完毕后可以调用Done方法来减少计数器的值，主函数可以调用Wait方法来阻塞，直到计数器为零后再继续执行。这种机制将并发操作转化为了同步操作，方便编写并控制高并发程序。

WaitGroup的原理

WaitGroup的原理其实很简单。它内部维护了一个计数器，初始值为0。每调用一次Add方法，计数器加1；每调用一次Done方法，计数器减1；每调用一次Wait方法，如果计数器值大于0，则会被阻塞等待，直到计数器为0后再继续执行。

使用WaitGroup的示例

下面我们来看一个使用WaitGroup的示例：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup
    wg.Add(2)
    
    go func() {
        defer wg.Done()
        fmt.Println("goroutine 1 executing")
    }()
    
    go func() {
        defer wg.Done()
        fmt.Println("goroutine 2 executing")
    }()
    
    wg.Wait()
    
    fmt.Println("all goroutines finished executing")
}

在上述示例中，我们首先声明了一个WaitGroup变量wg。然后通过调用Add方法来设置需要等待的goroutine数量，这里设置为2。接着启动两个goroutine，并在每个goroutine结束时调用Done方法来减少计数器的值。最后调用Wait方法来阻塞主函数，直到计数器为0后再继续执行。这样我们就可以保证所有的goroutine都执行完毕后再继续执行主函数的后续代码。

总之，WaitGroup是Go语言提供的一种简单而有效的并发同步机制，它使得控制一组goroutine的执行顺序变得非常简单。通过对计数器进行设置和修改，我们可以实现对一组goroutine的并发操作进行控制。在实际应用中，我们可以结合WaitGroup和其他并发原语，如互斥锁等，来编写高效可靠的并发程序。