Go语言的WaitGroup是用来等待一组goroutine执行完毕的工具,可以用来实现任务队列。下面是一个使用WaitGroup实现任务队列的示例代码:
package mainimport ("fmt""sync")func main() {// 创建一个WaitGroupvar 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进行处理,从而实现并发执行。