Golang并发编程：使用Go WaitGroup实现任务调度器
- 引言
在Golang中，实现并发编程可以大大提升程序的性能和效率。而任务调度器是并发编程中非常重要的一部分，它可以用于调度并发任务的执行顺序和同步任务的完成。本文将介绍如何使用Golang中的WaitGroup来实现一个简单的任务调度器，并给出具体的代码示例。 - WaitGroup简介
WaitGroup是Golang包sync中的一个重要类型，它提供了并发任务的同步和等待功能。WaitGroup有三个主要的方法：Add、Done和Wait。
- Add(n int)：添加 n 个并发任务到WaitGroup中。
- Done()：标记一个任务完成。
- Wait()：等待所有任务完成，并阻塞当前的goroutine。
WaitGroup内部维护了一个计数器，用来记录未完成的任务数量。当调用Add方法时，计数器加上指定的值；当调用Done方法时，计数器减1；当调用Wait方法时，如果计数器大于0，当前goroutine会被阻塞，直到计数器归零。
- 任务调度器实现
下面是一个使用WaitGroup实现任务调度器的示例代码：
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("Worker %d starting
", id)
time.Sleep(time.Second)
fmt.Printf("Worker %d done
", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 5; i++ {
wg.Add(1)
go worker(i, &wg)
}
wg.Wait()
fmt.Println("All workers done")
}在这个示例中，我们使用一个for循环启动5个worker goroutine，并通过调用wg.Add(1)来将任务数量添加到WaitGroup中。在worker函数中，使用defer关键字来确保在函数返回前调用wg.Done()，来减少计数器的值。最后，使用wg.Wait()来等待所有任务完成。
- 运行结果
运行这个示例代码，你会看到如下的输出结果：
Worker 1 starting Worker 2 starting Worker 3 starting Worker 4 starting Worker 5 starting Worker 3 done Worker 1 done Worker 2 done Worker 5 done Worker 4 done All workers done
可以看到，所有的worker goroutine被并发执行，但是输出的顺序并不一定是按照启动顺序来的，这是因为goroutine的调度是由Go运行时进行的。
立即学习“go语言免费学习笔记（深入）”；
- 总结
使用WaitGroup可以方便地实现并发任务的调度和同步。通过调用Add方法添加任务数量、调用Done方法标记任务完成、调用Wait方法来等待任务完成，可以实现任务的同步和顺序控制。在实际应用中，任务调度器可以用于处理并发任务、限制并发数量等场景，帮助我们更好地利用Golang的并发能力。
通过本文的介绍和示例代码，相信你对如何使用Golang的WaitGroup实现任务调度器有了更深的理解。希望本文对你学习和使用Golang的并发编程有所帮助！
