Go中sync.WaitGroup使用详解

珍惜时间，勤奋学习！今天给大家带来《Go中sync.WaitGroup使用方法详解》，正文内容主要涉及到等等，如果你正在学习Golang，或者是对Golang有疑问，欢迎大家关注我！后面我会持续更新相关内容的，希望都能帮到正在学习的大家！

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在Golang中主要通过一个内部计数器来协调goroutine的完成。当需要等待一组goroutine时，我们会在启动它们之前调用Add方法增加计数，每个goroutine完成任务后调用Done方法递减计数。主goroutine则通过调用Wait方法阻塞，直到这个计数器归零，表示所有被追踪的goroutine都已完成。

当我们在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的内部计数器变为零。

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

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都打印出"finished"才继续执行"Main: All workers completed. Exiting."。这正是我们想要的效果。

我个人在使用WaitGroup时，最常犯的一个小错误，或者说容易忽略的细节，就是Add的时机。它必须在对应的go语句之前调用。如果你在go worker(i, &wg)之后才wg.Add(1)，那么在某些极端情况下，如果worker执行得特别快，可能在Add被调用之前就已经Done了，这会导致计数器不匹配，Wait可能会提前返回或者永远等待。所以，顺序很重要。

Golang sync.WaitGroup有哪些常见的陷阱和最佳实践？

在使用sync.WaitGroup时，虽然它看起来直观，但

以上就是《Go中sync.WaitGroup使用详解》的详细内容，更多关于的资料请关注golang学习网公众号！