在并發(fā)編程中，經(jīng)常需要在主協(xié)程中等待多個(gè)子協(xié)程執(zhí)行完畢再進(jìn)行下一步操作。Go 提供了一個(gè)非常實(shí)用的同步原語(yǔ)：sync.WaitGroup，可以幫助我們優(yōu)雅地解決這個(gè)問(wèn)題。

一、什么是 WaitGroup？

sync.WaitGroup 是 Go 標(biāo)準(zhǔn)庫(kù)中的一個(gè)結(jié)構(gòu)體，用于等待一組 goroutine 執(zhí)行完成。

常用方法：

二、為什么需要 WaitGroup？

在并發(fā)場(chǎng)景中，如果主函數(shù)先于子協(xié)程執(zhí)行完畢，程序?qū)?huì)直接退出，導(dǎo)致協(xié)程未完成任務(wù)。例如：

func main() {
    go func() {
        fmt.Println("Hello from goroutine")
    }()
    // 沒(méi)有等待，主函數(shù)直接退出
}

輸出往往為空或者輸出不完整，根本原因是主 goroutine 沒(méi)有等待子 goroutine。

三、實(shí)戰(zhàn)案例：等待多個(gè)協(xié)程完成任務(wù)

示例：開(kāi)啟 5 個(gè) goroutine，每個(gè)打印一條消息，主程序等待所有協(xié)程完成后再退出。

package main

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

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 執(zhí)行完畢，通知 WaitGroup
    fmt.Printf("Worker %d starting\n", id)
    time.Sleep(time.Second) // 模擬耗時(shí)操作
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    var wg sync.WaitGroup

    for i := 1; i <= 5; i++ {
        wg.Add(1) // 啟動(dòng)一個(gè) goroutine 前調(diào)用 Add(1)
        go worker(i, &wg)
    }

    wg.Wait() // 阻塞直到所有 Done() 被調(diào)用
    fmt.Println("所有任務(wù)已完成")
}

輸出示例

Worker 1 starting
Worker 2 starting
Worker 3 starting
Worker 4 starting
Worker 5 starting
Worker 2 done
Worker 4 done
Worker 1 done
Worker 3 done
Worker 5 done
所有任務(wù)已完成

輸出順序可能每次不同，因?yàn)?goroutine 是并發(fā)執(zhí)行的。

四、使用建議與注意事項(xiàng)

? 正確使用方式：

• Add(n) 必須在 啟動(dòng) goroutine 之前 調(diào)用。
• 每個(gè) goroutine 結(jié)束時(shí)必須調(diào)用 Done()。
• Wait() 通常放在主 goroutine 中，阻塞直到所有任務(wù)完成。

?? 易犯錯(cuò)誤：

1. 1. 忘記 Done() ：

   go func() {
       fmt.Println("hello")
       // 忘記 wg.Done()，會(huì)導(dǎo)致 wg.Wait() 永久阻塞
   }()
   

2. 2. 在 goroutine 內(nèi)調(diào)用 Add() ：

   go func() {
       wg.Add(1) // ?不推薦：會(huì)出現(xiàn)數(shù)據(jù)競(jìng)爭(zhēng)
       ...
   }()
   

   正確方式是主 goroutine 中統(tǒng)一 Add()。

五、封裝使用：任務(wù)調(diào)度器

我們可以封裝一個(gè)并發(fā)調(diào)度器，更加靈活：

type TaskScheduler struct {
    wg sync.WaitGroup
}

func (t *TaskScheduler) Run(task func()) {
    t.wg.Add(1)
    go func() {
        defer t.wg.Done()
        task()
    }()
}

func (t *TaskScheduler) Wait() {
    t.wg.Wait()
}

使用示例：

func main() {
    scheduler := TaskScheduler{}

    for i := 0; i < 5; i++ {
        index := i
        scheduler.Run(func() {
            fmt.Printf("Task %d running\n", index)
            time.Sleep(500 * time.Millisecond)
        })
    }

    scheduler.Wait()
    fmt.Println("All scheduled tasks completed.")
}

六、與其他同步機(jī)制對(duì)比

七、結(jié)語(yǔ)

在并發(fā)控制中，sync.WaitGroup 是一種簡(jiǎn)單卻非常強(qiáng)大的工具，能幫助我們控制 goroutine 的生命周期，避免程序提前退出或資源泄漏等問(wèn)題。掌握其使用方法，是每位 Go 開(kāi)發(fā)者的必備技能。

? 實(shí)戰(zhàn)建議：在進(jìn)行網(wǎng)絡(luò)抓取、任務(wù)分發(fā)、并發(fā)計(jì)算等場(chǎng)景時(shí)，推薦優(yōu)先考慮 WaitGroup 來(lái)管理協(xié)程。

到此這篇關(guān)于Go語(yǔ)言使用WaitGroup等待多個(gè)協(xié)程完成的文章就介紹到這了,更多相關(guān)Go WaitGroup等待多協(xié)程完成內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！ 

最新評(píng)論