在 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<- Result），而应关闭原始 make(chan ...) 创建的双向通道；
- 若 worker 需从输入通道读取且该通道会被关闭，务必使用 for range in 或带 ok 的 <-in 判断，避免读取已关闭通道引发 panic；
- WaitGroup 不替代通道通信逻辑，而是与通道协同完成生命周期管理——这是 Go “通过通信共享内存”哲学的延伸实践。
综上，sync.WaitGroup + close(channel) 组合是 Go 社区广泛采纳的标准模式，兼具简洁性、可读性与可靠性，应作为并发任务收尾的首选方案。
