温馨提示×

并发任务调度:使用Go WaitGroup构建任务调度引擎

小云
105
2023-10-08 12:59:56
栏目: 编程语言

并发任务调度是一种将多个任务并行执行的技术,可以提高程序的执行效率。Go语言中提供了一个很方便的并发控制工具——WaitGroup,可以用来构建任务调度引擎。

WaitGroup是一个计数器,用来等待一组任务的结束。当我们启动一个任务时,可以调用WaitGroup的Add方法来增加计数器的值,当任务结束时,可以调用Done方法来减少计数器的值。主线程可以通过调用Wait方法来阻塞,直到计数器的值变为0,即所有任务都结束。

下面是一个简单的示例,演示如何使用WaitGroup构建任务调度引擎:

package main
import (
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
// 启动10个任务
for i := 0; i < 10; i++ {
wg.Add(1)
go func(id int) {
defer wg.Done()
fmt.Printf("Task %d starting\n", id)
time.Sleep(1 * time.Second) // 模拟任务执行时间
fmt.Printf("Task %d done\n", id)
}(i)
}
// 等待所有任务完成
wg.Wait()
fmt.Println("All tasks done")
}

在上面的示例中,我们创建了一个WaitGroup变量wg。然后,我们启动了10个任务,每个任务都会调用wg.Add(1)来增加计数器的值,表示有一个任务正在执行。任务的实现是一个匿名函数,它会打印任务的开始和结束信息,并模拟任务执行时间。在任务结束时,我们调用wg.Done()来减少计数器的值,表示一个任务已经完成。

最后,我们调用wg.Wait()来阻塞主线程,直到计数器的值变为0,即所有任务都完成。在所有任务完成后,我们打印出"All tasks done"。

通过使用WaitGroup,我们可以方便地构建任务调度引擎,实现并发任务的调度和控制。在实际应用中,可以根据实际需求对任务进行更复杂的控制,例如设置最大并发数、任务优先级等。

0