我正在从 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)
}
}
在你的第一次尝试中:
在你的第二次尝试中: