温馨提示×

Golang并发编程:使用Go WaitGroup实现任务调度器

小云
82
2023-10-19 17:52:34
栏目: 编程语言

任务调度器是一种常见的并发编程模式,它用于同时执行多个任务并等待所有任务完成后再继续执行其他操作。在Golang中,可以使用sync.WaitGroup实现任务调度器。

sync.WaitGroup是Golang提供的一个并发原语,用于等待一组goroutine完成。它提供了三个方法:Add()Done()Wait()

下面是一个使用sync.WaitGroup实现任务调度器的示例代码:

package main
import (
"fmt"
"sync"
)
func main() {
// 创建一个WaitGroup对象
var wg sync.WaitGroup
// 设置任务的数量
numTasks := 10
wg.Add(numTasks)
// 并发执行任务
for i := 0; i < numTasks; i++ {
go func(taskId int) {
// 执行任务
fmt.Printf("Task %d is running\n", taskId)
// 标记任务完成
wg.Done()
}(i)
}
// 等待所有任务完成
wg.Wait()
// 所有任务完成后执行其他操作
fmt.Println("All tasks have been completed")
}

在上面的代码中,首先创建了一个sync.WaitGroup对象。然后,通过调用Add()方法设置了要执行的任务数量。接下来,通过循环并发执行任务,并在每个任务完成时调用Done()方法标记任务完成。最后,调用Wait()方法等待所有任务完成。

运行上述代码,将会输出类似以下内容:

Task 0 is running
Task 1 is running
Task 2 is running
Task 3 is running
Task 4 is running
Task 5 is running
Task 6 is running
Task 7 is running
Task 8 is running
Task 9 is running
All tasks have been completed

可以看到,所有任务完成后,程序继续执行其他操作。

使用sync.WaitGroup实现任务调度器可以方便地管理多个并发任务,并等待它们全部完成。这对于实现并行计算、批量处理等场景非常有用。

0