golang+waitgroup原理

发布时间:2024-11-05 16:41:17

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和其他并发原语,如互斥锁等,来编写高效可靠的并发程序。

相关推荐