Go语言并发编程中 WaitGroup 用法

发布于 1周前 作者 eggper 最后一次编辑是 5天前 来自 问答

Go语言并发编程中 WaitGroup 用法

一个 WaitGroup 对象可以等待一组协程结束。使用方法是：

  1. main 协程通过调用 wg.Add(delta int) 设置 worker 协程的个数，然后创 建 worker 协程； 2.worker 协程执行结束以后，都要调用 wg.Done()； 3.main 协程调用 wg.Wait() 且被 block，直到所有 worker 协程全部执行结束 后返回。
1 回复

在Go语言的并发编程中，sync.WaitGroup是一个非常有用的工具，用于等待一组goroutines的完成。它允许你启动任意数量的goroutines，并在所有这些goroutines完成后再继续执行。这对于需要等待所有并发任务完成后再进行下一步操作的场景特别有用。

WaitGroup的基本用法

  1. 初始化：使用var wg sync.WaitGroup声明一个WaitGroup实例。
  2. 增加计数器：在启动goroutine之前，使用wg.Add(1)（或更多，取决于你打算启动多少个goroutine）来增加WaitGroup的计数器。
  3. 等待完成：在所有goroutine中，使用defer wg.Done()来在goroutine结束时减少WaitGroup的计数器，并在主函数中调用wg.Wait()来等待所有goroutine完成。

示例代码

以下是一个使用sync.WaitGroup的简单示例，它启动了两个goroutine，每个goroutine都执行一个简单的打印任务，并在完成后减少WaitGroup的计数器。主函数会等待这两个goroutine都完成后才继续执行。

package main

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

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 确保goroutine结束时减少WaitGroup的计数器
    fmt.Printf("Worker %d starting\n", id)

    // 模拟耗时任务
    time.Sleep(time.Second)
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    var wg sync.WaitGroup

    // 启动两个goroutine
    for i := 1; i <= 2; i++ {
        wg.Add(1) // 增加WaitGroup的计数器
        go worker(i, &wg)
    }

    // 等待所有goroutine完成
    wg.Wait()
    fmt.Println("All workers finished")
}

在这个例子中，main函数首先初始化了一个WaitGroup实例。然后，它使用wg.Add(1)为每个将要启动的goroutine增加计数器的值。每个goroutine在被启动时都接收到了WaitGroup的指针，并在其函数体开始处使用defer wg.Done()来确保在goroutine结束时减少计数器的值。最后，main函数通过调用wg.Wait()来等待所有goroutine的完成。

回到顶部