本文介绍一种基于 sync.waitgroup 和非阻塞通道发送的 go 工作池模式,用于处理可递归生成新任务的场景(如网页爬虫),避免死锁、竞态与过早退出,兼顾简洁性与生产可用性。
在构建异步任务处理系统时(例如 URL 抓取器),一个常见但棘手的需求是:任务本身可能动态产生新任务(即“递归式”任务调度),而工作协程需在无待处理任务时自动终止,且整个池必须整体活跃或整体休眠——不能出现部分协程空转、部分协程阻塞的僵局。
原始方案尝试通过 working 通道统计活跃 worker 数量,并依赖 absent 信号协调启停,但存在逻辑耦合高、状态管理脆弱、难以验证正确性等问题;更关键的是,它隐含了对 select 执行顺序的依赖(虽符合 Go 规范),却未解决核心矛盾:如何安全、无锁地判断“全局无任务且无人将生成新任务”?
✅ 推荐解法:WaitGroup + 非阻塞任务入队(fallback execution)
其核心思想是:用 sync.WaitGroup 原子跟踪“已提交但未完成”的总任务数,而非 worker 状态;当 worker 从通道取到任务后立即 wg.Add(1)(表示该任务及其潜在子任务将被计入总数),执行完毕调用 wg.Done()。入队函数 enqueue 采用非阻塞写入:若通道有空位则直接投递;否则立即在当前 goroutine 中执行该任务(并递归调用 enqueue 处理其子任务)。这彻底规避了“所有 worker 都在等任务,但新任务正试图入队却被阻塞”的死锁。
以下是精简可靠的实现: