基础教程：Go WaitGroup及其在Golang中的应用-Golang-PHP中文网

基础教程：Go WaitGroup及其在Golang中的应用

王林

发布： 2023-09-28 09:42:38

原创

1077 人浏览过

基础教程：Go WaitGroup及其在Golang中的应用

基础教程：Go WaitGroup及其在Golang中的应用，需要具体代码示例

前言：
在Golang（Go语言）中，编写并发程序是一项常见任务。Golang提供了丰富的并发原语和API，其中一个重要的工具是WaitGroup（等待组）。本文将为您介绍WaitGroup的概念和用法，并提供了一些具体的代码示例。

一、WaitGroup的概念
WaitGroup是Golang中的一种并发原语，它用于等待一组协程（goroutine）的完成。WaitGroup内部维护着一个计数器，初始值为0。当每个协程开始执行时，计数器会加1。当协程结束时，计数器会减1。主线程可以通过Wait()方法阻塞，直到计数器为0，即所有协程都执行完成。

二、WaitGroup的基本用法
使用WaitGroup需要依次进行如下步骤：

创建WaitGroup对象：首先，我们需要导入"sync"包，并创建一个WaitGroup对象。
```
import "sync"

var wg sync.WaitGroup
```
登录后复制
添加协程到WaitGroup：对于每个要执行的协程，我们需要增加计数器，并使用go关键字在新的协程中执行该任务。
```
wg.Add(1)
go func() {
 // 协程的具体逻辑
 // ...
 wg.Done() // 协程执行完成后减少计数器
}()
```
登录后复制
等待协程的完成：主线程可以使用Wait()方法来阻塞，直到计数器为0，即所有协程都已经执行完成。
```
wg.Wait()
```
登录后复制

示例代码：
下面是一个具体的代码示例，演示了WaitGroup在Golang中的应用。该示例使用WaitGroup来等待多个协程完成后再继续后续的逻辑。

package main
 
import (
    "fmt"
    "sync"
    "time"
)
 
func main() {
    var wg sync.WaitGroup
 
    for i := 0; i < 5; i++ {
        wg.Add(1)
 
        go func(num int) {
            defer wg.Done()
            time.Sleep(time.Second) // 模拟长时间的任务
 
            fmt.Printf("协程 %d 完成
", num)
        }(i)
    }
 
    fmt.Println("等待所有协程完成...")
    wg.Wait()
    fmt.Println("所有协程已完成！")
}

登录后复制

在上面的示例中，我们模拟了5个协程的执行，每个协程都等待1秒钟。通过WaitGroup，主线程会阻塞，直到所有协程执行完成。在输出中，我们可以看到协程完成的顺序并不固定，因为它们是并行执行的。

三、WaitGroup的注意事项
在使用WaitGroup时，需要遵循以下几点注意事项：