Golang WaitGroup 踩坑
2026/9/12 0:22:27 网站建设 项目流程

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

  1. ‌未启动单独goroutine导致主线程阻塞‌
    ‌问题‌:若WaitGroup未在goroutine内调用,主线程可能因等待子任务完成而阻塞。
    ‌解决方案‌:确保WaitGroup操作在独立goroutine中执行:
varwg sync.WaitGroup wg.Add(1)gofunc(){deferwg.Done()// 子任务逻辑}()wg.Wait()// 主线程等待子任务完成
  1. ‌计数器操作顺序错误‌
    ‌问题‌:Add(n)应在goroutine启动前调用,否则可能导致Wait()提前返回。
    ‌解决方案‌:遵循Add() -> 启动goroutine -> Done()的顺序:
wg.Add(1)// 增加计数器gofunc(){deferwg.Done()// 任务结束时减少计数器// 任务逻辑}()
  1. ‌闭包变量捕获问题‌
    ‌问题‌:从循环启动goroutine时,若闭包捕获循环变量(如for循环中的i),可能导致所有goroutine共享同一变量值。
    ‌解决方案‌:通过参数传递循环变量值:
fori:=0;i<5;i++{wg.Add(1)gofunc(numint){// 传递参数deferwg.Done()fmt.Println("Goroutine",num)}(i)// 传递当前i值}
  1. ‌未正确传递指针导致计数器失效‌
    问题‌:将WaitGroup作为值传递给函数时,子函数操作的是副本,主函数计数器不变。
    解决方案‌:传递指针引用:
funcworker(wg*sync.WaitGroup){deferwg.Done()// 任务逻辑}wg.Add(1)worker(&wg)// 传递指针
  1. ‌未调用Done()导致死锁‌
    问题‌:若goroutine未调用Done(),计数器永远不会归零,Wait()将永久阻塞。
    解决方案‌:确保每个goroutine结束前调用Done():
gofunc(){deferwg.Done()// 确保Done()在return前执行// 任务逻辑}()
  1. ‌并发场景下的资源泄漏‌
    ‌问题‌:若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并发编程的基石,但需严格遵循计数器操作顺序和闭包变量传递规则,避免常见陷阱。

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询