任务调度器是一种常见的并发编程模式,它用于同时执行多个任务并等待所有任务完成后再继续执行其他操作。在Golang中,可以使用sync.WaitGroup实现任务调度器。
sync.WaitGroup是Golang提供的一个并发原语,用于等待一组goroutine完成。它提供了三个方法:Add()、Done()和Wait()。
下面是一个使用sync.WaitGroup实现任务调度器的示例代码:
package mainimport ("fmt""sync")func main() {// 创建一个WaitGroup对象var wg sync.WaitGroup// 设置任务的数量numTasks := 10wg.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 runningTask 1 is runningTask 2 is runningTask 3 is runningTask 4 is runningTask 5 is runningTask 6 is runningTask 7 is runningTask 8 is runningTask 9 is runningAll tasks have been completed可以看到,所有任务完成后,程序继续执行其他操作。
使用sync.WaitGroup实现任务调度器可以方便地管理多个并发任务,并等待它们全部完成。这对于实现并行计算、批量处理等场景非常有用。