Go工作池通过固定goroutine复用任务,用channel分发与收集结果,WaitGroup协调生命周期;含jobs/results通道、worker数及初始化;worker循环读取jobs并发送结果;提交后关闭jobs再wg.Wait,最后关results。

用 Go 实现工作池(Worker Pool),核心是通过固定数量的 goroutine 复用执行任务,避免无节制创建 goroutine 导致内存耗尽或调度开销过大。关键在于用 channel 控制任务分发与结果收集,并用 WaitGroup 或关闭信号协调生命周期。
定义工作池结构和初始化
工作池通常包含:任务输入 channel、结果输出 channel、worker 数量、以及用于等待所有 worker 结束的 sync.WaitGroup。
- 任务 channel 一般用 buffered 或 unbuffered 均可,但 buffered 更利于突发任务缓冲(如
make(chan Job, 100)) - 每个 worker 是一个独立 goroutine,持续从任务 channel 中读取并处理,直到 channel 关闭
- 建议将 Job 定义为接口或结构体,携带所需参数和上下文,便于扩展
启动固定数量的 worker 并并发执行
在启动阶段,用 for 循环启动指定数量的 goroutine,每个 goroutine 运行同一个 worker 函数:
- worker 内部用
for job := range jobs持续接收任务(channel 关闭时自动退出循环) - 处理完任务后,把结果(或错误)发送到 results channel,保持非阻塞或带 buffer 避免卡住 worker
- 用
defer wg.Done()确保 worker 退出时通知 WaitGroup
提交任务与安全关闭工作池
任务提交只需向 jobs channel 发送 Job 实例;关闭则需两步:
立即学习“go语言免费学习笔记(深入)”;
- 所有任务提交完成后,关闭 jobs channel(
close(jobs)),通知 worker 不再有新任务 - 调用
wg.Wait()等待所有 worker 退出,之后可安全关闭 results channel(如需) - 注意:不要在发送任务时关闭 channel,也不要在 worker 运行中重复 close,会 panic
处理结果与错误的常用方式
results channel 可以是 chan Result 或 chan *Result,消费端按需处理:
- 如果需要按提交顺序获取结果,可用 map + sync.Mutex 缓存,配合任务 ID 关联
- 若只关心成功/失败统计,可另起 goroutine 收集并计数,避免阻塞主流程
- 对超时控制,可在提交任务时包装 context,worker 内部 select 判断 ctx.Done()
不复杂但容易忽略。










