文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

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

2023-10-08 11:41

关注

Go语言的WaitGroup是用来等待一组goroutine执行完毕的工具,可以用来实现任务队列。下面是一个使用WaitGroup实现任务队列的示例代码:
```go
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进行处理,从而实现并发执行。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯