Go 并发处理切片元素:使用 WaitGroup 实现安全的并行函数调用

本文讲解如何将顺序遍历切片并调用函数的操作改为真正的并发执行,避免闭包变量捕获错误和切片越界 panic,使用 sync.waitgroup 安全收集结果。

在 Go 中实现切片元素的并发处理时,核心目标是:对每个元素独立启动 goroutine 执行耗时操作(如 double(i)),不阻塞后续 goroutine 启动,并最终按原始顺序汇总结果。原代码中尝试用 []chan int 存储通道,但存在两个关键错误:

  1. 切片未初始化即索引访问:var chans []chan int 声明的是 nil 切片,长度为 0,直接通过 chans[i] = ... 赋值会触发 panic: index out of range;
  2. 闭包变量捕获陷阱:for counter, number := range arr { go func() { ... }() } 中,匿名函数内部引用的 counter 和 number 是循环变量的地址,所有 goroutine 共享同一份内存,导致最终读取到的是循环结束时的最后一个值(典型竞态问题)。

✅ 正确解法无需引入额外通道——直接使用预分配结果切片 + sync.WaitGroup 协调完成信号,简洁、高效且线程安全:

func parallel(arr []int) []int {
    outArr := make([]int, len(arr)) // 预分配,保证索引安全
    var wg sync.WaitGroup

    for i, num := range arr {
        wg.Add(1)
        // 显式传参,避免闭包捕获循环变量
        go func(index int, value int) {
            defer wg.Done()
            outArr[index] = double(value)
        }(i, num)
    }

    wg.Wait() // 主协程阻塞等待所有 goroutine 完成
    return outArr
}

? 关键要点说明:

  • outArr := make([]int, len(arr)) 确保切片有确定容量和长度,支持 outArr[i] 安全赋值;
  • go func(index, value int) { ... }(i, num) 通过函数参数“快照”当前循环变量值,彻底规避闭包陷阱;
  • defer wg.Done() 放在 goroutine 内部更符合习惯,确保即使函数中途 panic 也能释放计数;
  • wg.Wait() 是必要同步点——没有它,parallel 可能返回未填充完毕的切片(甚至全零值)。

⚠️ 注意事项:

  • 若需错误处理(如 double 可能失败),应扩展为返回 (int, error) 并配合 errgroup.Group;
  • 当切片极大(如百万级)时,无限制启动 goroutine 可能导致资源耗尽,建议结合 semaphore 或 worker pool 限流;
  • 本例中 double 有 fmt.Println,并发输出可能乱序(这是正常现象),若需严格顺序日志,应单独加锁或由主 goroutine 统一打印。

综上,Go 的并发编程强调明确的数据所有权与显式的同步机制。相比复杂通道编排,WaitGroup + 预分配切片 是处理“映射型并行任务”(map-reduce 中的 map 阶段)最直观、低开销且不易出错的模式。