Golang并发编程:使用Go WaitGroup实现任务调度器
任务调度器是一种常见的并发编程模式,它用于同时执行多个任务并等待所有任务完成后再继续执行其他操作。在Golang中,可以使用sync.WaitGroup
实现任务调度器。sync.WaitGroup
是Golang提供的一个并发原语,用于等待一组goroutine完成。它提供了三个方法:Add()
、Done()
和Wait()
。
下面是一个使用sync.WaitGroup
实现任务调度器的示例代码:go
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
实现任务调度器可以方便地管理多个并发任务,并等待它们全部完成。这对于实现并行计算、批量处理等场景非常有用。
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341