var wg sync.WaitGroup

// Add(n): 計(jì)數(shù)器加 n，表示"我又派了 n 個(gè)任務(wù)"
wg.Add(3)

for i := 0; i < 3; i++ {
    go func(id int) {
        defer wg.Done() // 等價(jià)于 Add(-1)，表示"我干完了"
        // ... 干活 ...
    }(i)
}

wg.Wait() // 阻塞直到計(jì)數(shù)器歸零

// ? 錯(cuò)誤：wg.Add 在 go 里面
for i := 0; i < 5; i++ {
    go func() {
        wg.Add(1)  // 嚴(yán)重錯(cuò)誤！Wait 可能在這之前就返回了
        defer wg.Done()
        doWork()
    }()
}
wg.Wait()

// ? 正確
for i := 0; i < 5; i++ {
    wg.Add(1)
    go func() {
        defer wg.Done()
        doWork()
    }()
}
wg.Wait()

// ? 值傳遞：計(jì)數(shù)器的修改對(duì)原始 wg 不可見(jiàn)
func run(wg sync.WaitGroup) {
    defer wg.Done()
}

// ? 指針傳遞
func run(wg *sync.WaitGroup) {
    defer wg.Done()
}

// ? 數(shù)據(jù)競(jìng)爭(zhēng)！
go func() {
    wg.Add(1) // 與下面的 wg.Wait 并發(fā)
    defer wg.Done()
    doWork()
}()
wg.Wait() // 此時(shí)可能還沒(méi)有 Add 完成

package main

import (
    "fmt"
    "math/rand"
    "sync"
    "time"
)

// 模擬從多個(gè)數(shù)據(jù)源獲取數(shù)據(jù)
type DataFetcher struct {
    name    string
    latency time.Duration // 模擬耗時(shí)
}

func (f DataFetcher) Fetch() string {
    time.Sleep(f.latency)
    return fmt.Sprintf("[%s 數(shù)據(jù)]", f.name)
}

func main() {
    rand.Seed(time.Now().UnixNano())

    // 三個(gè)數(shù)據(jù)源，各自耗時(shí)不同
    sources := []DataFetcher{
        {name: "數(shù)據(jù)庫(kù)", latency: 200 * time.Millisecond},
        {name: "Redis", latency: 50 * time.Millisecond},
        {name: "遠(yuǎn)程API", latency: 300 * time.Millisecond},
    }

    // 結(jié)果收集
    results := make([]string, len(sources))
    var wg sync.WaitGroup

    for i, src := range sources {
        wg.Add(1) // 在 go 之前 Add
        go func(idx int, fetcher DataFetcher) {
            defer wg.Done()
            results[idx] = fetcher.Fetch()
        }(i, src) // 注意：把循環(huán)變量傳給閉包
    }

    fmt.Println("等待所有數(shù)據(jù)源返回...")
    wg.Wait()
    fmt.Println("全部就緒！\n")

    for i, r := range results {
        fmt.Printf("數(shù)據(jù)源 %d (%s): %s\n", i, sources[i].name, r)
    }

    // —— 進(jìn)階：帶超時(shí)的 WaitGroup ——
    fmt.Println("\n——— 超時(shí)模式演示 ———")

    var wg2 sync.WaitGroup
    wg2.Add(1)
    go func() {
        defer wg2.Done()
        time.Sleep(2 * time.Second) // 模擬很慢的任務(wù)
        fmt.Println("慢任務(wù)完成")
    }()

    // 用一個(gè) channel 來(lái)通知"WaitGroup 已經(jīng)等完了"
    done := make(chan struct{})
    go func() {
        wg2.Wait()
        close(done)
    }()

    select {
    case <-done:
        fmt.Println("所有任務(wù)在超時(shí)前完成")
    case <-time.After(500 * time.Millisecond):
        fmt.Println("超時(shí)！不等了")
    }
}