阅读量: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
可以方便地管理大规模任务的并发执行,确保所有任务完成后再继续执行后续的代码。