最佳实践:使用 waitgroup 管理 goroutine 并发何时使用 waitgroup:用于等待所有子例程完成后再继续执行。使用方法:wg.add(n):添加 goroutine 数量。wg.wait():阻塞调用程序,直到所有 goroutine 完成。wg.done():goroutine 完成时递减计数。优点:简单、可扩展、安全,防止数据争用。
Go 函数并发编程的最佳实践:何时使用 WaitGroup?
并发编程是 Go 语言的核心,它允许您编写可以充分利用多核 CPU 的高效程序。WaitGroup 是一个内置的并发原语,可帮助您执行以下操作:
- 等待一组 goroutine 完成任务
- 跟踪 goroutine 的数量
- 在所有 goroutine 完成之前阻止调用程序
何时使用 WaitGroup?
立即学习“”;
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) }
登录后复制
使用方法
- wg.Add(n):将 goroutine 的数量添加到 WaitGroup。
- wg.Wait():阻塞调用程序,直到 WaitGroup 的计数变为 0。
- wg.Done():当 goroutine 完成其任务时,它应该调用此方法以递减 WaitGroup 计数。
优点
使用 WaitGroup 的优点包括:
- 简单性:它提供了一种简单且有效的方式来同步 goroutine。
- 可扩展性:它可以很容易地扩展到支持大量并发任务。
- 安全性:它防止在所有 goroutine 完成任务之前继续执行,有助于避免数据争用。
以上就是Golang 函数并发编程的最佳实践:何时使用 WaitGroup?的详细内容,更多请关注php中文网其它相关文章!