在 go 并发编程中，当多个 worker 协程向同一输出通道（output channel）发送结果时，主协程需可靠感知所有 worker 完成的时机，才能安全关闭通道并终止遍历——`sync.waitgroup` 是最简洁、标准且推荐的解决方案。
在基于通道的生产者-消费者模型中，一个常见痛点是：无法自然得知所有 worker 是否已完成工作并停止向输出通道写入数据。若主协程过早退出，可能丢失结果；若盲目 range 未关闭的通道，则会永久阻塞。此时，sync.WaitGroup 提供了轻量、线程安全的同步原语，完美解决“等待全部 goroutine 完成后关闭通道”的核心需求。
✅ 正确实践：WaitGroup + 通道关闭组合模式
关键原则是：
- 启动 worker 前调用 wg.Add(1)，为每个 worker 预注册；
- worker 内部在完成所有任务（包括向 outchan 发送最后结果）后调用 wg.Done()；
- 另启一个 goroutine 调用 wg.Wait()，随后 close(outchan) —— 这是唯一安全关闭通道的时机；
- 主逻辑使用 for range outchan，该循环会在通道关闭后自动退出，无需额外条件判断。
以下是完整可运行示例（以处理整数平方为例）：
package main
import (
"fmt"
"sync"
"time"
)
func worker(wg *sync.WaitGroup, in <-chan int, out chan<- int) {
defer wg.Done() // 确保无论何种退出路径都调用 Done()
for n := range in {
// 模拟耗时处理
time.Sleep(100 * time.Millisecond)
out <- n * n
}
}
func main() {
const N = 3
in := make(chan int, 10)
out := make(chan int, 10)
var wg sync.WaitGroup
// 启动 N 个 worker
for i := 0; i < N; i++ {
wg.Add(1)
go worker(&wg, in, out)
}
// 启动“关闭协程”：等待所有 worker 结束后关闭输出通道
go func() {
wg.Wait()
close(out) // 关键：仅在此处关闭，确保无 goroutine 再写入
}()
// 主协程：发送任务
go func() {
for i := 1; i <= 6; i++ {
in <- i
}
close(in) // 输入通道也可关闭（worker 会自然退出）
}()
// 安全消费所有输出（range 自动在 out 关闭后终止）
for result := range out {
fmt.Println("Result:", result)
}
fmt.Println("All done.")
}⚠️ 注意事项与最佳实践
- defer wg.Done() 是惯用写法：保证即使 worker 中发生 panic 或提前 return，计数器仍能正确减一；
- 切勿在 worker 内直接关闭 out 通道：多 goroutine 同时关闭同一通道会 panic；必须由单一协程（通常是 wg.Wait() 后的协程）执行；
- 通道缓冲区大小建议合理设置：如示例中 make(chan int, 10) 可缓解发送端阻塞，但非必需；无缓冲通道亦可正常工作；
- 避免 WaitGroup 误用：不要在 Add() 前调用 Wait()，也不要重复 Add() 同一实例而未配对 Done()，否则导致死锁或 panic；
- 替代方案对比：虽然 context.WithCancel 或额外 done 通道也能实现，但 WaitGroup 更直观、零开销、语义清晰，是 Go 官方文档和标准库广泛采用的方式。
通过 sync.WaitGroup 驱动通道关闭，你不仅能写出健壮的并发流水线，还能让代码逻辑清晰、易于测试与维护——这正是 Go 并发哲学中“通过通信共享内存”的典范实践。
