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

php中文网 2024-10-15 11:09:59

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

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

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

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

何时使用 WaitGroup？

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

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

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

实战案例

考虑这样一个场景：您有一个 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)
}

使用方法