
在 go 并发编程中,当多个 worker 协程向同一输出通道(output channel)发送结果时,主协程需可靠感知“所有 worker 已完成且无更多数据”,从而安全退出 `for range` 循环。`sync.waitgroup` 是最简洁、标准且符合 go idioms 的解决方案。
在典型的生产者-消费者并发模型中(如 Rob Pike 提倡的“扇入”模式),我们常启动多个 worker 协程从输入通道读取任务、处理后将结果写入共享的输出通道。此时,主协程通过 for range outchan 消费结果——但该循环仅在通道被显式关闭后才会自动退出;若无人关闭通道,循环将永久阻塞,导致程序无法正常终止。
因此,关键在于:何时、由谁来关闭输出通道? 答案是:由一个“协调协程”在确认所有 worker 全部完成工作后执行 close(outchan)。而 sync.WaitGroup 正是为此类场景设计的同步原语——它提供线程安全的计数器,支持 Add()、Done() 和阻塞式 Wait(),完美匹配“启动 N 个 worker → 等待全部结束 → 关闭通道”的控制流。
以下是推荐实现方式:
package main
import (
"fmt"
"sync"
"time"
)
type Result struct {
ID int
Value string
}
func worker(wg *sync.WaitGroup, in <-chan int, out chan<- Result) {
defer wg.Done() // 确保无论何种退出路径都调用 Done()
for task := range in {
// 模拟耗时处理
time.Sleep(100 * time.Millisecond)
result := Result{
ID: task,
Value: fmt.Sprintf("processed-%d", task),
}
out <- result
}
}
func main() {
const numWorkers = 3
// 输入通道(可由生产者协程填充)
in := make(chan int, 10)
// 输出通道(无缓冲,避免阻塞;也可根据需要设缓冲)
out := make(chan Result, 10)
var wg sync.WaitGroup
// 启动 worker 协程
for i := 0; i < numWorkers; i++ {
wg.Add(1)
go worker(&wg, in, out)
}
// 启动协调协程:等待所有 worker 完成后关闭输出通道
go func() {
wg.Wait()
close(out) // 关键:关闭后,for range 将自然退出
}()
// 主协程:发送任务到输入通道(模拟生产者)
go func() {
for i := 1; i <= 5; i++ {
in <- i
}
close(in) // 关闭输入通道,通知 worker 无新任务
}()
// 消费所有输出结果(自动在 out 关闭后退出)
for res := range out {
fmt.Printf("Received: %+v\n", res)
}
fmt.Println("All done.")
}✅ 关键要点说明:
- defer wg.Done() 应置于 worker 函数开头,确保即使发生 panic 或提前 return,计数仍能正确递减;
- 输出通道必须由单一方(此处为协调协程)关闭,且仅关闭一次——多次关闭会 panic;
- 不要关闭只读或只写通道变量(如 out chan
- 若 worker 需从输入通道读取且该通道会被关闭,务必使用 for range in 或带 ok 的
- WaitGroup 不替代通道通信逻辑,而是与通道协同完成生命周期管理——这是 Go “通过通信共享内存”哲学的延伸实践。
综上,sync.WaitGroup + close(channel) 组合是 Go 社区广泛采纳的标准模式,兼具简洁性、可读性与可靠性,应作为并发任务收尾的首选方案。










