如何使用Go WaitGroup处理并发任务 - 问答

在 Go 中使用 sync.WaitGroup 来处理并发任务，可以确保在所有任务完成之前等待主程序的执行。下面是一个简单的例子：

package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
numTasks := 10
// 增加任务数
wg.Add(numTasks)
for i := 0; i < numTasks; i++ {
go func(taskNum int) {
defer wg.Done() // 每个任务执行完成后减少任务数
fmt.Printf("Task %d is running\n", taskNum)
// 执行具体任务
}(i)
}
// 等待所有任务完成
wg.Wait()
fmt.Println("All tasks are completed")
}

在上面的例子中，我们首先创建了一个 WaitGroup 对象 wg，然后使用 Add() 方法增加了需要等待的任务的数量。在每个任务的 goroutine 中，我们使用 defer wg.Done() 来在任务执行完毕后减少任务数。最后，我们使用 Wait() 方法来等待所有任务完成，然后输出相应的信息。

对于每个任务，你可以在 go func() 中执行具体的任务逻辑，例如进行数据处理、调用 API，或者执行其他操作。每个任务完成时，调用 wg.Done() 来减少任务数。

使用 sync.WaitGroup 的好处是可以确保在所有任务完成之前主程序不会退出。

0 赞

0 踩