新闻资讯

质量为本、客户为根、勇于拼搏、务实创新

< 返回新闻资讯列表

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

发布时间:2023-10-19 17:52:34

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

任务调度器是一种常见的并发编程模式,它用于同时履行多个任务并等待所有任务完成后再继续履行其他操作。在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
", 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实现任务调度器可以方便地管理多个并发任务,并等待它们全部完成。这对实现并行计算、批量处理等场景非常有用。