阅读量:0
Go语言的WaitGroup是用来等待一组goroutine执行完毕的工具，可以用来实现任务队列。下面是一个使用WaitGroup实现任务队列的示例代码：
package main import ( "fmt" "sync" ) func main() { // 创建一个WaitGroup var wg sync.WaitGroup // 创建一个带缓冲的任务队列 queue := make(chan string, 10) // 启动三个goroutine来处理任务 for i := 0; i < 3; i++ { // 增加WaitGroup的计数 wg.Add(1) // 启动一个goroutine来处理任务 go worker(i, queue, &wg) } // 向任务队列中添加任务 for i := 0; i < 10; i++ { queue <- fmt.Sprintf("Task %d", i) } // 关闭任务队列 close(queue) // 等待所有goroutine执行完毕 wg.Wait() } func worker(id int, queue chan string, wg *sync.WaitGroup) { // 在函数退出时，减少WaitGroup的计数 defer wg.Done() // 不断从任务队列中获取任务，并处理任务 for task := range queue { fmt.Printf("Worker %d processing task: %s\n", id, task) } } 上面的代码中，首先创建了一个WaitGroup和一个带缓冲的任务队列。然后启动了三个goroutine来处理任务。每个goroutine都会从任务队列中获取任务并处理。最后，通过调用Wait方法等待所有goroutine执行完毕。
在主函数中，通过循环向任务队列中添加任务，并在添加完毕后关闭任务队列。这样所有的goroutine就会在任务队列为空时退出，并通过调用Done方法减少WaitGroup的计数。
需要注意的是，在worker函数中，通过defer关键字来调用Done方法，以保证在函数退出时一定会调用Done方法。这样可以避免在处理任务过程中发生异常导致WaitGroup计数无法减少的问题。
通过上述代码，可以实现利用Go WaitGroup实现任务队列的功能。每个任务会被分配给可用的goroutine进行处理，从而实现并发执行。
