在Go语言中,可以使用sync.WaitGroup来实现大规模任务的并发优化。WaitGroup是一个计数器,它等待一组goroutine完成工作。
下面是使用sync.WaitGroup的并发优化方法的示例代码:
package mainimport ("fmt""sync")func main() {// 创建WaitGroupvar 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可以方便地管理大规模任务的并发执行,确保所有任务完成后再继续执行后续的代码。

