17370845950

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 实例；关闭则需两步：

• 所有任务提交完成后，关闭 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()

不复杂但容易忽略。