Golang 函数并发编程的最佳实践：何时使用 WaitGroup？

最佳实践：使用 waitgroup 管理 goroutine 并发何时使用 waitgroup：用于等待所有子例程完成后再继续执行。使用方法：wg.add(n)：添加 goroutine 数量。wg.wait()：阻塞调用程序，直到所有 goroutine 完成。wg.done()：goroutine 完成时递减计数。优点：简单、可扩展、安全，防止数据争用。

Go 函数并发编程的最佳实践：何时使用 WaitGroup？

并发编程是 Go 语言的核心，它允许您编写可以充分利用多核 CPU 的高效程序。WaitGroup 是一个内置的并发原语，可帮助您执行以下操作：

• 等待一组 goroutine 完成任务
• 跟踪 goroutine 的数量
• 在所有 goroutine 完成之前阻止调用程序

何时使用 WaitGroup？

立即学习“go语言免费学习笔记（深入）”；

WaitGroup 适用于需要等到所有子例程完成任务后才能继续执行的场景。典型的用例包括：

• 从多个来源收集数据
• 执行一系列异步任务
• 处理请求管道

实战案例

豆包AI编程

豆包推出的AI编程助手

483

查看详情

考虑这样一个场景：您有一个 HTTP 服务，该服务使用 goroutine 从多个 API 获取数据。您希望在所有数据收集完成之前阻止 HTTP 响应，以便您可以将完整的响应发送给客户端。

代码示例

使用WaitGroup实现此操作：

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    var wg sync.WaitGroup

    // 创建一个管道以接收每个 goroutine 的结果
    resultChan := make(chan string)

    // 创建 3 个 goroutine 从不同的 API 获取数据
    wg.Add(3)
    for i := 0; i < 3; i++ {
        go func(i int) {
            defer wg.Done() // 递减 WaitGroup 计数

            // 模拟 API 调用
            time.Sleep(time.Second)
            resultChan <- fmt.Sprintf("结果 %d", i)
        }(i)
    }

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

    // 从管道中收集结果
    var results []string
    for {
        select {
        case result, ok := <-resultChan:
            if !ok {
                break
            }
            results = append(results, result)
        }
    }

    // 将完整响应发送给客户端
    fmt.Println(results)
}

使用方法

• wg.Add(n)：将 goroutine 的数量添加到 WaitGroup。
• wg.Wait()：阻塞调用程序，直到 WaitGroup 的计数变为 0。
• wg.Done()：当 goroutine 完成其任务时，它应该调用此方法以递减 WaitGroup 计数。

优点

使用 WaitGroup 的优点包括：

• 简单性：它提供了一种简单且有效的方式来同步 goroutine。
• 可扩展性：它可以很容易地扩展到支持大量并发任务。
• 安全性：它防止在所有 goroutine 完成任务之前继续执行，有助于避免数据争用。

以上就是Golang 函数并发编程的最佳实践：何时使用 WaitGroup？的详细内容，更多请关注php中文网其它相关文章！