如何在 Go 中使用WaitGroup同步 Goroutine？

偷偷努力，悄无声息地变强，然后惊艳所有人！哈哈，小伙伴们又来学习啦~今天我将给大家介绍《如何在 Go 中使用WaitGroup同步 Goroutine？》，这篇文章主要会讲到等等知识点，不知道大家对其都有多少了解，下面我们就一起来看一吧！当然，非常希望大家能多多评论，给出合理的建议，我们一起学习，一起进步！

如何在 Go 中使用 WaitGroup 同步 Goroutine？

什么是 WaitGroup？

WaitGroup 是 Go 中的一个内置类型，用于协调并发操作，它可以用来确保一组 goroutine 在执行完毕之前不会继续执行。

如何使用 WaitGroup

使用 WaitGroup 的步骤如下：

1. 创建 WaitGroup

var wg sync.WaitGroup

2. 在 goroutine 中使用 Add()

当一个新的 goroutine 开始执行时，使用 WaitGroup.Add(1) 来增加计数器。

wg.Add(1)
go func() {
  // goroutine 代码
  wg.Done()
}()

3. 在 goroutine 中使用 Done()

当 goroutine 执行完毕时，使用 WaitGroup.Done() 来减少计数器。

func() {
  // goroutine 代码
  wg.Done()
}

4. 等待 Goroutine 完成

使用 WaitGroup.Wait() 来阻塞当前 goroutine，直到所有关联的 goroutine 完成执行。

wg.Wait()

实战案例

以下是一个示例，演示了如何使用 WaitGroup 同步三个并发 goroutine：

package main

import (
  "fmt"
  "sync"
)

func main() {
  var wg sync.WaitGroup

  // 创建三个并发 goroutine
  for i := 0; i < 3; i++ {
    wg.Add(1)
    go func(i int) {
      defer wg.Done()
      fmt.Printf("Goroutine %d complete\n", i)
    }(i)
  }

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

  // 输出：Goroutine 0 complete
  // 输出：Goroutine 1 complete
  // 输出：Goroutine 2 complete

  fmt.Println("All goroutines completed")
}

本篇关于《如何在 Go 中使用WaitGroup同步 Goroutine？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！