使用带缓冲通道控制并发数,通过信号量机制限制goroutine数量,实现轻量级协程池,适用于需控制并发的任务场景。

Go语言中的协程(goroutine)轻量且高效,但无限制地创建协程可能导致资源耗尽。为控制并发数量,常使用协程池配合带缓冲的通道来实现任务调度。这种方案简单、稳定,适合处理大量短期任务。
基本思路:使用缓冲通道控制并发数
协程池的核心是限制同时运行的goroutine数量。通过一个带缓冲的通道作为信号量,每启动一个任务前先向通道写入一个值,任务结束再读出,从而实现并发控制。
这种方式不维护复杂的池结构,而是用通道天然的同步机制管理并发,代码简洁且线程安全。
实现步骤
以下是基于带缓冲通道的协程池实现方法:
立即学习“go语言免费学习笔记(深入)”;
定义任务类型:使用函数类型表示任务,如
type Task func()
创建信号量通道:容量为最大并发数,如
sem := make(chan struct{}, 5)
表示最多5个并发任务提交任务时获取信号量:在执行任务前尝试向通道写入,若通道满则阻塞等待任务执行完成后释放信号量:在goroutine末尾从通道读取,释放并发槽位
示例代码:
package mainimport ("fmt""sync""time")
type Task func()
type Pool struct {sem chan struct{}wg sync.WaitGroup}
func NewPool(maxConcurrency int) *Pool {return &Pool{sem: make(chan struct{}, maxConcurrency),}}
func (p *Pool) Submit(task Task) {p.wg.Add(1)go func() {defer p.wg.Done()p.sem <- struct{}{} // 获取执行权defer func() { <-p.sem }() // 释放执行权
task() // 执行任务}()
}
func (p *Pool) Wait() {p.wg.Wait()}
func main() {pool := NewPool(3) // 最大3个并发
for i := 0; i < 10; i++ { i := i pool.Submit(func() { fmt.Printf("执行任务 %dn", i) time.Sleep(1 * time.Second) })}pool.Wait()fmt.Println("所有任务完成")
}
优点与适用场景
这种方案的优势在于:
简单易懂:仅用通道和goroutine实现,无需复杂状态管理天然并发安全:通道本身是Go的同步原语,避免锁的使用资源可控:通过缓冲通道大小精确控制最大并发数
适用于Web请求处理、文件批量处理、爬虫任务等需要限制并发的场景。
基本上就这些,不需要额外依赖,也不容易出错。只要合理设置通道容量,就能在性能和资源之间取得平衡。
以上就是Golang协程池如何构建 带缓冲通道方案的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1399203.html
微信扫一扫
支付宝扫一扫