温馨提示×

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

小云
112
2023-10-08 11:34:28
栏目: 编程语言

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

0