php小编柚子在此为大家介绍一种在sync.WaitGroup goroutine中写入chan的方法。在并发编程中，sync.WaitGroup是一种非常有用的同步机制，它可以等待一组goroutine的执行完成。然而，有时我们需要在goroutine执行完毕后，将结果写入到一个chan中，以供其他goroutine消费。本文将详细介绍如何在sync.WaitGroup goroutine中实现这一功能，让我们一起来看看吧！
问题内容
我正在从 API 端点获取项目列表。然后，对于每个项目，我都会发出另一个 API 请求以获取有关单个项目的数据。
我无法同时对每个项目发出第二个 API 请求，因为我的 API 令牌有速率限制，如果我同时发出太多请求，我会受到限制。
但是，初始 API 响应数据可以分为多个页面，这使我能够同时处理数据页面。
经过一些研究，下面的代码完全符合我的要求：
func main() {
// pretend paginated results from initial API request
page1 := []int{1, 2, 3}
page2 := []int{4, 5, 6}
page3 := []int{7, 8, 9}
pages := [][]int{page1, page2, page3}
results := make(chan string)
var wg sync.WaitGroup
for i := range pages {
wg.Add(1)
go func(i int) {
defer wg.Done()
for j := range pages[i] {
// simulate making additional API request and building the report
time.Sleep(500 * time.Millisecond)
result := fmt.Sprintf("Finished creating report for %d", pages[i][j])
results <- result
}
}(i)
}
go func() {
wg.Wait()
close(results)
}()
for result := range results {
fmt.Println(result)
}
}
我想了解为什么它能发挥作用：
go func() {
wg.Wait()
close(results)
}()
我的第一次尝试没有成功——我想我可以在 wg.Wait() 之后遍历通道，并且我会在结果写入 results 通道时读取结果。
func main() {
// pretend paginated results from initial API request
page1 := []int{1, 2, 3}
page2 := []int{4, 5, 6}
page3 := []int{7, 8, 9}
pages := [][]int{page1, page2, page3}
results := make(chan string)
var wg sync.WaitGroup
for i := range pages {
wg.Add(1)
go func(i int) {
defer wg.Done()
for j := range pages[i] {
// simulate making additional API request and building the report
time.Sleep(500 * time.Millisecond)
result := fmt.Sprintf("Finished creating report for %d", pages[i][j])
results <- result
}
}(i)
}
// does not work
wg.Wait()
close(results)
for result := range results {
fmt.Println(result)
}
}
解决方法
在您的第一次尝试中：
- 主 goroutine 使 3 个 goroutine 将值放入结果通道中。
- 主协程等待所有协程完成。
- 其中一个 goroutine 将一个值放入结果通道中并填充通道（通道大小为 1 个字符串）。
- 现在所有三个 goroutine 都无法再将值放入结果通道并进入睡眠状态，直到结果通道被释放。
- 所有 goroutine 都处于睡眠状态。你陷入了僵局。
在第二次尝试中：
- 主 goroutine 包含 4 个 goroutine。
- 3 个 goroutine 将值放入结果通道中。
- 其他 goroutine（我将其称为第 4 个）等待这 3 个 goroutine 结束。
- 同时主协程等待结果通道中的值（for 循环）
- 在这种情况下，如果其中一个 goroutine 在结果通道中放入一个值，则会阻塞其余的三个 goroutine；主 Goroutine 将值从结果通道中取出，从而解除对其他 Goroutine 的阻塞。
- 因此，所有 3 个 goroutine 都放入各自的值并结束
- 然后第四个 goroutine 关闭通道
- 主 Goroutine 结束其 for 循环。
