Golang如何使用WaitGroup等待goroutine完成_石家庄创泽智能科技有限公司

新闻中心 NEWS CENTER

您当前位置：首页 > 新闻中心 > 网络学院

Golang如何使用WaitGroup等待goroutine完成

2025-11-01

浏览次数：次

返回列表

WaitGroup用于等待多个goroutine完成，通过Add增加计数、Done减少计数、Wait阻塞直至计数为零。

golang如何使用waitgroup等待goroutine完成

在Go语言中，WaitGroup 是 sync 包提供的一个同步原语，用于等待一组并发的 goroutine 执行完成。它特别适合在主协程需要等待多个子协程结束时使用，而不需要关闭通道或处理复杂的信号机制。

基本用法和核心方法

WaitGroup 有三个主要方法：

Add(delta int)：增加计数器，通常传入要启动的 goroutine 数量。
Done()：在每个 goroutine 结束时调用，相当于计数器减1。
Wait()：阻塞当前协程，直到计数器归零。

典型结构是：主协程调用 Add(n) 设置任务数量，每个子协程执行完后调用 Done()，主协程通过 Wait() 阻塞等待全部完成。

简单示例：等待多个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 <= 3; i++ {
        wg.Add(1)           // 增加计数
        go worker(i, &wg)   // 启动goroutine
    }

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

输出会显示三个 worker 依次开始和结束，最后打印“All workers finished”。