<ol><li>使用channel实现Go协程池可控制并发、复用goroutine;2. 核心是带缓冲channel作任务队列,固定worker协程取任务执行;3. Pool结构含tasks chan和workers数;4. NewPool创建实例并初始化缓冲channel;5. Start启动worker监听任务;6. 通过tasks <- task提交任务;7. close(tasks)关闭后worker自动退出。</li></ol>

在Go语言中,使用channel实现协程池是一种常见且高效的方式,能够控制并发数量、复用goroutine并避免资源过度消耗。核心思路是通过一个带缓冲的channel作为任务队列,配合固定数量的worker协程从队列中取任务执行。
定义协程池结构
协程池通常包含任务队列、worker数量和控制信号等字段。任务可以用函数类型func()表示,通过channel传递。
示例结构:
type Task func()
<p>type Pool struct {
tasks chan Task
workers int
}
初始化协程池
创建Pool实例时,指定worker数量和任务队列的缓冲大小。每个worker启动一个goroutine,持续监听任务channel。
立即学习“go语言免费学习笔记(深入)”;
关键点:
- 任务channel需带缓冲,否则发送任务会阻塞
- 每个worker在独立goroutine中运行,循环读取任务执行
示例代码:
func NewPool(workers, queueSize int) *Pool {
return &Pool{
tasks: make(chan Task, queueSize),
workers: workers,
}
}
<p>func (p *Pool) Start() {
for i := 0; i < p.workers; i++ {
go func() {
for task := range p.tasks {
task()
}
}()
}
}
提交任务与关闭池
通过向tasks channel发送函数实现任务提交。当不再提交任务时,可关闭channel让worker自然退出。
使用方式:
- 调用pool.tasks <- task发送任务
- 所有任务发送完毕后,close(pool.tasks)关闭channel
- worker在range循环中自动退出
示例:
pool := NewPool(3, 10)
pool.Start()
<p>// 提交任务
for i := 0; i < 5; i++ {
pool.tasks <- func() {
fmt.Println("处理任务", i)
}
}</p><p>close(pool.tasks) // 关闭后worker会逐步退出
基本上就这些。这种方式简单可控,适合大多数需要限制并发的场景。关键是理解channel的阻塞机制和goroutine生命周期管理。










