使用带缓冲通道控制并发数,通过信号量机制限制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("执行任务 %d\n", i) time.Sleep(1 * time.Second) }) } pool.Wait() fmt.Println("所有任务完成")}
优点与适用场景
这种方案的优势在于:
- 简单易懂:仅用通道和goroutine实现,无需复杂状态管理
- 天然并发安全:通道本身是Go的同步原语,避免锁的使用
- 资源可控:通过缓冲通道大小精确控制最大并发数
适用于Web请求处理、文件批量处理、爬虫任务等需要限制并发的场景。
基本上就这些,不需要额外依赖,也不容易出错。只要合理设置通道容量,就能在性能和资源之间取得平衡。










