Golang并发编程：利用Go WaitGroup实现任务队列

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进行处理，从而实现并发执行。