Golang WaitGroup 踩坑

在Go语言中，sync.WaitGroup（简称WaitGroup）是用于多goroutine同步的核心机制，但其使用需谨慎，否则可能导致程序卡顿、死锁或数据竞争等问题。以下是关键踩坑点及解决方案：

1. ‌未启动单独goroutine导致主线程阻塞‌
   ‌问题‌：若WaitGroup未在goroutine内调用，主线程可能因等待子任务完成而阻塞。
   ‌解决方案‌：确保WaitGroup操作在独立goroutine中执行：

varwg sync.WaitGroup wg.Add(1)gofunc(){deferwg.Done()// 子任务逻辑}()wg.Wait()// 主线程等待子任务完成

2. ‌计数器操作顺序错误‌
   ‌问题‌：Add(n)应在goroutine启动前调用，否则可能导致Wait()提前返回。
   ‌解决方案‌：遵循Add() -> 启动goroutine -> Done()的顺序：

wg.Add(1)// 增加计数器gofunc(){deferwg.Done()// 任务结束时减少计数器// 任务逻辑}()

3. ‌闭包变量捕获问题‌
   ‌问题‌：从循环启动goroutine时，若闭包捕获循环变量（如for循环中的i），可能导致所有goroutine共享同一变量值。
   ‌解决方案‌：通过参数传递循环变量值：

fori:=0;i<5;i++{wg.Add(1)gofunc(numint){// 传递参数deferwg.Done()fmt.Println("Goroutine",num)}(i)// 传递当前i值}

4. ‌未正确传递指针导致计数器失效‌
   问题‌：将WaitGroup作为值传递给函数时，子函数操作的是副本，主函数计数器不变。
   解决方案‌：传递指针引用：

funcworker(wg*sync.WaitGroup){deferwg.Done()// 任务逻辑}wg.Add(1)worker(&wg)// 传递指针

5. ‌未调用Done()导致死锁‌
   问题‌：若goroutine未调用Done()，计数器永远不会归零，Wait()将永久阻塞。
   解决方案‌：确保每个goroutine结束前调用Done()：

gofunc(){deferwg.Done()// 确保Done()在return前执行// 任务逻辑}()

6. ‌并发场景下的资源泄漏‌
   ‌问题‌：若WaitGroup未正确管理goroutine生命周期，可能导致资源泄漏。
   ‌解决方案‌：结合context实现超时控制：

ctx,cancel:=context.WithTimeout(context.Background(),1*time.Second)defercancel()gofunc(ctx context.Context){select{case<-ctx.Done():return// 超时退出case<-time.After(2*time.Second):// 任务逻辑}}(ctx)

示例代码

varwg sync.WaitGroup wg.Add(1)gofunc(){deferwg.Done()time.Sleep(1*time.Second)fmt.Println("Task completed")}()wg.Wait()// 等待任务完成

WaitGroup是Go并发编程的基石，但需严格遵循计数器操作顺序和闭包变量传递规则，避免常见陷阱。