Golang协程同步：WaitGroup使用全解析

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《Golang协程同步：WaitGroup使用详解》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

首先初始化WaitGroup，再通过Add增加计数，每个goroutine执行完调用Done，主线程调用Wait阻塞直至所有任务完成。

在Go语言中，sync.WaitGroup 是一种常用的协程同步机制，用于等待一组并发的goroutine执行完成。它特别适用于主线程需要等待多个子任务结束的场景，比如批量发起网络请求、并行处理数据等。使用 WaitGroup 可以避免程序提前退出，确保所有协程都执行完毕。

WaitGroup 核心方法

WaitGroup 提供了三个主要方法：

• Add(delta int)：增加计数器的值，通常用于添加要等待的goroutine数量。
• Done()：将计数器减1，一般在goroutine结束时调用，相当于 Add(-1)。
• Wait()：阻塞当前协程，直到计数器变为0。

基本使用示例

下面是一个简单的例子，展示如何使用 WaitGroup 等待多个goroutine完成：

package main

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

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 任务完成，计数器减1
    fmt.Printf("Worker %d starting\n", id)
    time.Sleep(time.Second)
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    var wg sync.WaitGroup

    for i := 1; i 

输出结果：

使用注意事项

虽然 WaitGroup 使用简单，但有几个关键点需要注意，避免常见错误：

• 避免Add负数或超出范围：在调用 Add 时传入负数会引发 panic，必须确保 Add 的值合理。
• 确保 Done 被调用：每个 Add 对应一次 Done，否则 Wait 会一直阻塞，导致死锁。
• 传递指针：WaitGroup 应以指针形式传递给goroutine，否则值拷贝会导致各个协程操作的是不同的实例。
• Add 应在 go 之前调用：如果在goroutine内部才 Add，可能主线程的 Wait 已经开始，导致遗漏。

实际应用场景

WaitGroup 常用于并行任务处理。例如，从多个API获取数据：

func fetchAllData() {
    var wg sync.WaitGroup
    urls := []string{"http://a.com", "http://b.com", "http://c.com"}

    for _, url := range urls {
        wg.Add(1)
        go func(u string) {
            defer wg.Done()
            resp, err := http.Get(u)
            if err != nil {
                fmt.Printf("Error: %s\n", err)
                return
            }
            defer resp.Body.Close()
            fmt.Printf("Fetched %s\n", u)
        }(url)
    }

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

这段代码并发请求多个URL，并等待全部完成后再继续。

基本上就这些。WaitGroup 是Go中最基础也最实用的同步工具之一，掌握它的正确用法对编写可靠的并发程序至关重要。不复杂但容易忽略细节。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。