温馨提示×

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

小云
79
2023-10-12 09:46:11
栏目: 编程语言

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

0