<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生命周期管理。


