大规模任务处理:使用Go WaitGroup的并发优化方法

avatar
作者
筋斗云
阅读量:1

在Go语言中,可以使用sync.WaitGroup来实现大规模任务的并发优化。WaitGroup是一个计数器,它等待一组goroutine完成工作。

下面是使用sync.WaitGroup的并发优化方法的示例代码:

package main import ( "fmt" "sync" ) func main() { // 创建WaitGroup var wg sync.WaitGroup // 设置要处理的任务数量 taskCount := 10 // 增加WaitGroup的计数器 wg.Add(taskCount) // 启动goroutine执行任务 for i := 0; i < taskCount; i++ { go func(taskID int) { // 执行任务的逻辑 fmt.Printf("Task %d started\n", taskID) // 模拟任务的耗时操作 for j := 0; j < 100000000; j++ { } fmt.Printf("Task %d finished\n", taskID) // 减少WaitGroup的计数器 wg.Done() }(i) } // 等待所有任务完成 wg.Wait() fmt.Println("All tasks finished") } 

在上述示例代码中,我们首先创建了一个WaitGroup,并设置要处理的任务数量为10。然后,我们使用wg.Add(taskCount)增加WaitGroup的计数器。

在启动每个任务的goroutine之前,我们使用匿名函数传递任务ID,并在该goroutine中执行任务的逻辑。在任务完成后,我们使用wg.Done()减少WaitGroup的计数器。

最后,我们使用wg.Wait()等待所有任务完成。一旦所有任务完成,程序会继续执行,并输出"All tasks finished"。

使用sync.WaitGroup可以方便地管理大规模任务的并发执行,确保所有任务完成后再继续执行后续的代码。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!