sync.WaitGroup在Golang中如何等待所有goroutine完成任务

sync.WaitGroup通过计数器协调goroutine完成。1. 初始化后，在启动goroutine前调用Add增加计数；2. 每个goroutine结束时调用Done减少计数，推荐使用defer wg.Done()确保执行；3. 主goroutine调用Wait阻塞，直到计数器归零。常见陷阱是Add调用时机错误，必须在go语句前执行，否则可能导致竞态条件或死锁。最佳实践包括：始终在启动goroutine前Add，使用defer wg.Done()保证通知，避免复制WaitGroup。

sync.WaitGroup

Add

Done

Wait

当我们在Golang中处理并发任务时，经常会遇到一个场景：启动多个goroutine去执行一些工作，然后主goroutine需要等待所有这些子goroutine都完成，才能继续执行后续逻辑，或者安全地退出程序。

sync.WaitGroup

它的核心机制其实很简单：

1. 初始化：声明一个

   var wg sync.WaitGroup

   登录后复制
   。
2. 计数器增加：在启动任何需要等待的goroutine之前，调用

   wg.Add(delta int)

   登录后复制
   。这个

   delta

   登录后复制
   通常是你即将启动的goroutine数量。每调用一次

   Add(1)

   登录后复制
   ，内部计数器就加一。
3. 任务完成通知：在每个goroutine内部，当其任务即将完成时，调用

   wg.Done()

   登录后复制
   。这会将内部计数器减一。一个非常好的实践是使用

   defer wg.Done()

   登录后复制
   ，这样即使goroutine内部发生panic，

   Done()

   登录后复制
   也能被执行，避免主goroutine永远等待。
4. 等待：在主goroutine中，调用

   wg.Wait()

   登录后复制
   。这个方法会阻塞当前goroutine，直到

   WaitGroup

   登录后复制
   的内部计数器变为零。

来看个例子，这能让你更直观地理解：

立即学习“go语言免费学习笔记（深入）”；

有道小P

有道小P，新一代AI全科学习助手，在学习中遇到任何问题都可以问我。

64

查看详情

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(id int, wg *sync.WaitGroup) {
    // 确保无论如何，这个goroutine完成时都会通知WaitGroup
    defer wg.Done() 
    fmt.Printf("Worker %d starting...\n", id)
    time.Sleep(time.Duration(id) * time.Second) // 模拟工作
    fmt.Printf("Worker %d finished.\n", id)
}

func main() {
    var wg sync.WaitGroup
    numWorkers := 3

    fmt.Println("Main: Launching workers...")
    for i := 1; i <= numWorkers; i++ {
        wg.Add(1) // 每启动一个worker，计数器加1
        go worker(i, &wg)
    }

    fmt.Println("Main: Waiting for all workers to complete...")
    wg.Wait() // 阻塞直到所有worker都调用了Done()

    fmt.Println("Main: All workers completed. Exiting.")
}

运行这段代码，你会看到主goroutine在启动所有

worker

worker

"Main: All workers completed. Exiting."

我个人在使用

WaitGroup

Add

go

go worker(i, &wg)

wg.Add(1)

worker

Add

Done

Wait

Golang

sync.WaitGroup

登录后复制

有哪些常见的陷阱和最佳实践？

在使用

sync.WaitGroup

以上就是sync.WaitGroup在Golang中如何等待所有goroutine完成任务的详细内容，更多请关注php中文网其它相关文章！