GolangWaitGroup使用与控制技巧

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《Golang WaitGroup实战应用与控制技巧》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

WaitGroup通过Add、Done、Wait方法实现goroutine同步，主协程调用Add设置任务数，子协程完成时调用Done，主协程Wait阻塞直至所有任务结束，常用于并发请求、批量处理等场景。

在Go语言开发中，WaitGroup 是 sync 包提供的一个非常实用的同步原语，用于等待一组并发的 goroutine 执行完成。它特别适用于主协程需要等待多个子任务结束后再继续执行的场景。掌握 WaitGroup 的正确用法，是编写高效、稳定并发程序的基础。

WaitGroup 核心机制

WaitGroup 内部维护一个计数器，通过三个方法控制：

• Add(n)：增加计数器值，通常用于添加要等待的 goroutine 数量
• Done()：计数器减1，一般在 goroutine 结束时调用
• Wait()：阻塞当前协程，直到计数器归零

典型使用模式是：主协程调用 Add 设置任务数量，每个子协程执行完后调用 Done，主协程通过 Wait 阻塞等待所有任务完成。

基本使用示例

以下是一个简单的并行HTTP请求示例：

package main
import (
"fmt"
"net/http"
"sync"
)
func fetchURL(url string, wg *sync.WaitGroup) {
defer wg.Done()
resp, err := http.Get(url)
if err != nil {
    fmt.Printf("Error fetching %s: %v\n", url, err)
    return
}
defer resp.Body.Close()

fmt.Printf("Success: %s -> Status %d\n", url, resp.StatusCode)
}
func main() {
urls := []string{
"https://httpbin.org/get",
"https://httpbin.org/delay/1",
"https://httpbin.org/status/200",
}
var wg sync.WaitGroup

for _, url := range urls {
    wg.Add(1)
    go fetchURL(url, &wg)
}

wg.Wait()
fmt.Println("All requests completed.")
}

在这个例子中，每次启动一个 goroutine 前调用 Add(1)，并在函数末尾通过 defer 调用 Done() 确保计数器正确减少。

常见陷阱与最佳实践

使用 WaitGroup 时有几个关键点需要注意：

• 避免在 goroutine 内部调用 Add，否则可能造成竞争或漏加
• 确保每个 Add 都有对应的 Done，否则会永久阻塞
• 传递 WaitGroup 指针而非值，防止拷贝导致状态不一致
• 合理使用 defer Done()，确保即使发生 panic 也能正确计数

错误示例：在 goroutine 中 Add 会导致不可预测行为

// ❌ 错误做法
go func() {
    wg.Add(1) // 危险！可能在 Wait 后才执行
    defer wg.Done()
    // ...
}()

正确方式是在启动 goroutine 前完成 Add 操作。

进阶应用场景

WaitGroup 可与其他并发结构组合使用。例如配合 channel 实现超时控制：

func doWithTimeout() bool {
    var wg sync.WaitGroup
    done := make(chan bool, 1)
for i := 0; i < 5; i++ {
    wg.Add(1)
    go func(id int) {
        defer wg.Done()
        // 模拟耗时任务
        // time.Sleep(time.Second)
        fmt.Printf("Worker %d done\n", id)
    }(i)
}

go func() {
    wg.Wait()
    done <- true
}()

select {
case <-done:
    return true
case <-time.After(3 * time.Second):
    return false // 超时
}
}

这种模式将 WaitGroup 的等待逻辑放到单独的协程中，使主流程能支持超时退出。

基本上就这些。WaitGroup 虽然简单，但用好并不容易。关键是理解其计数机制，避免竞态，并结合实际场景灵活运用。在真实项目中，它常用于批量任务处理、服务启动关闭、数据预加载等需要协同等待的场合。

到这里，我们也就讲完了《GolangWaitGroup使用与控制技巧》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！