golang wg

Go语言(Golang)是由谷歌开发的一种静态类型、编译型的高级编程语言。它旨在提供一种简单、高效和可靠的开发方式，以满足当今大规模分布式系统的需求。Go语言内置了并发支持，具备快速编译和执行的特性，这使得它成为构建高性能软件的理想选择。

背景介绍

在Go语言的标准库中，有一个很重要的包叫做sync。这个包提供了许多实用的同步原语，其中的WaitGroup就是我们今天要介绍的主角。

什么是WaitGroup

WaitGroup是Go语言中用于等待一组goroutine完成的工具。它通过计数器的方式来实现这个功能。主线程调用Add方法来设定需要等待的goroutine数量，每个goroutine在执行完成后，调用Done方法来将计数器减1。最后，主线程通过调用Wait方法来等待计数器归零。

使用WaitGroup

下面是一个使用WaitGroup的例子：

package main

import (
    "fmt"
    "sync"
)

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done()

    fmt.Printf("Worker %d starting\n", id)

    // 模拟一些耗时的工作
    for i := 0; i < 5; i++ {
        fmt.Printf("Worker %d working...\n", id)
    }

    fmt.Printf("Worker %d finished\n", id)
}

func main() {
    var wg sync.WaitGroup

    // 设置需要等待的goroutine数量为2
    wg.Add(2)

    // 启动两个goroutine
    go worker(1, &wg)
    go worker(2, &wg)

    // 等待所有goroutine完成
    wg.Wait()

    fmt.Println("All workers finished")
}

在上面的例子中，我们定义了一个worker函数来模拟一些耗时的工作。主线程通过调用Add方法将计数器设置为2，表示需要等待两个goroutine执行完成。然后，我们启动了两个goroutine来执行worker函数，并将WaitGroup的指针传递给它们。每个goroutine执行完成后，调用Done方法将计数器减1。最后，主线程通过调用Wait方法来等待计数器归零，即等待所有goroutine完成。

使用WaitGroup可以有效地协调和控制多个goroutine的执行顺序。在实际开发中，我们常常会遇到需要等待多个goroutine完成后再执行某些操作的场景。比如并发爬虫，我们需要等待所有的爬取任务完成后再进行数据处理；或者在并发测试中，我们需要等待所有的测试任务完成后再进行结果统计。

总之，WaitGroup是Go语言中非常有用且方便的工具，它可以帮助我们解决并发编程中的协调和控制问题。通过使用Add、Done和Wait方法，我们可以轻松地实现多个goroutine的同步和等待操作。