func main() {
    var waitGroup sync.WaitGroup

    start := time.Now()
    waitGroup.Add(5)
    for i := 0; i < 5; i++ {
        go func() {
            defer waitGroup.Done()
            time.Sleep(time.Second)
            fmt.Println("done")
        }()
    }

    waitGroup.Wait()
    fmt.Println(time.Now().Sub(start).Seconds())
}

/*
done
done
done
done
done
1.000306089
*/

(wg *WaitGroup) Add(delta int)
(wg *WaitGroup) Done()
(wg *WaitGroup) Wait()

type WaitGroup struct {
    noCopy noCopy // noCopy 字段標(biāo)識(shí)，由于 WaitGroup 不能復(fù)制，方便工具檢測(cè)

    state1 [3]uint32  // 12個(gè)字節(jié)，8個(gè)字節(jié)標(biāo)識(shí) 計(jì)數(shù)值和等待數(shù)量，4個(gè)字節(jié)用于標(biāo)識(shí)信號(hào)量
}

func (wg *WaitGroup) state() (statep *uint64, semap *uint32) {
    
    // 首地址是8的倍數(shù)時(shí)，前8個(gè)字節(jié)為 statep, 后四個(gè)字節(jié)為 semap
    if uintptr(unsafe.Pointer(&wg.state1))%8 == 0 {
        return (*uint64)(unsafe.Pointer(&wg.state1)), &wg.state1[2]
    } else { 
        
    // 后8個(gè)字節(jié)為 statep, 前四個(gè)字節(jié)為 semap    
        return (*uint64)(unsafe.Pointer(&wg.state1[1])), &wg.state1[0]
    }
}

func (wg *WaitGroup) Add(delta int) {

    // 拿到計(jì)數(shù)值等待者變量 statep 和 信號(hào)量 semap
    statep, semap := wg.state()

    // 計(jì)數(shù)值加上 delta: statep 的前四個(gè)字節(jié)是計(jì)數(shù)值，因此將 delta 前移 32位
    state := atomic.AddUint64(statep, uint64(delta)<<32)

    // 計(jì)數(shù)值
    v := int32(state >> 32)

    // 等待者數(shù)量
    w := uint32(state)

    // 如果加上 delta 之后，計(jì)數(shù)值變?yōu)樨?fù)數(shù)，不合法，panic
    if v < 0 {
        panic("sync: negative WaitGroup counter")
    }

    // delta > 0 && v == int32(delta) : 表示從 0 開(kāi)始添加計(jì)數(shù)值
    // w!=0 ：表示已經(jīng)有了等待者
    // 說(shuō)明在添加計(jì)數(shù)值的時(shí)候，同時(shí)添加了等待者，非法操作。添加等待者需要在添加計(jì)數(shù)值之后
    if w != 0 && delta > 0 && v == int32(delta) {
        panic("sync: WaitGroup misuse: Add called concurrently with Wait")
    }

    // v>0 : 計(jì)數(shù)值不等于0，不需要喚醒等待者，直接返回
    // w==0: 沒(méi)有等待者，不需要喚醒，直接返回
    if v > 0 || w == 0 {
        return
    }

    // 再次檢查數(shù)據(jù)是否一致
    if *statep != state {
        panic("sync: WaitGroup misuse: Add called concurrently with Wait")
    }

    // 到這里說(shuō)明計(jì)數(shù)值為0，且等待者大于0，需要喚醒所有的等待者，并把系統(tǒng)置為初始狀態(tài)（0狀態(tài)）
  
  // 將計(jì)數(shù)值和等待者數(shù)量都置為0
    *statep = 0

    // 喚醒等待者
    for ; w != 0; w-- {
        runtime_Semrelease(semap, false, 0)
    }
}

// 完成一個(gè)任務(wù)，將計(jì)數(shù)值減一，當(dāng)計(jì)數(shù)值減為0時(shí)，需要喚醒所有的等待者
func (wg *WaitGroup) Done() {
    wg.Add(-1)
}

// 調(diào)用 Wait 方法會(huì)被阻塞，直到 計(jì)數(shù)值 變?yōu)?
func (wg *WaitGroup) Wait() {

    // 獲取計(jì)數(shù)、等待數(shù)和信號(hào)量
    statep, semap := wg.state()

    for {
        state := atomic.LoadUint64(statep)

        // 計(jì)數(shù)值
        v := int32(state >> 32)

        // 等待者數(shù)量
        w := uint32(state)

        // 計(jì)數(shù)值數(shù)量為0，直接返回，無(wú)需等待
        if v == 0 {
            return
        }

        // 到這里說(shuō)明計(jì)數(shù)值數(shù)量大于0
        // 增加等待者數(shù)量：這里會(huì)有競(jìng)爭(zhēng)，比如多個(gè) Wait 調(diào)用，或者在同時(shí)調(diào)用 Add 方法，增加不成功會(huì)繼續(xù) for 循環(huán)
        if atomic.CompareAndSwapUint64(statep, state, state+1) {
            // 增加成功后，阻塞在信號(hào)量這里，等待被喚醒
            runtime_Semacquire(semap)

            // 被喚醒的時(shí)候，應(yīng)該是0狀態(tài)。如果重用 WaitGroup，需要等 Wait 返回
            if *statep != 0 {
                panic("sync: WaitGroup is reused before previous Wait has returned")
            }
            return
        }
    }
}

func main() {
var wg sync.WaitGroup
wg.Add(1)
wg.Add(-1)
wg.Add(-1)
}

func main() {
    var wg sync.WaitGroup

    wg.Add(1)
    go func() {
        fmt.Println("test")
        wg.Done()
        wg.Done()
    }()

    time.Sleep(time.Second)
    wg.Wait()
}

func doSomething(wg *sync.WaitGroup) {
    wg.Add(1)
    fmt.Println("do  something")
    defer wg.Done()
}

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 3; i++ {
        go doSomething(&wg)
    }

    wg.Wait()
    fmt.Println("main")
}

//main
//do  something
//do  something

func doSomething(wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Println("do  something")
}

func main() {
    var wg sync.WaitGroup
    wg.Add(3)
    for i := 0; i < 3; i++ {
        go doSomething(&wg)
    }

    wg.Wait()
    fmt.Println("main")
}

//do  something
//do  something
//do  something
//main

func main() {
    var wg sync.WaitGroup

    wg.Add(1)
    go func() {
        fmt.Println("do something")
        wg.Done()
        wg.Add(1)
    }()

    wg.Wait()
}

// 錯(cuò)誤使用方式，沒(méi)有使用指針
func doSomething(wg sync.WaitGroup) {
    fmt.Println("do  something")
    defer wg.Done()
}

func main() {
    var wg sync.WaitGroup
    wg.Add(3)
    for i := 0; i < 3; i++ {
        // 這里沒(méi)使用指針，wg狀態(tài)一直不會(huì)改變，導(dǎo)致 Wait 一直阻塞
        go doSomething(wg)
    }

    wg.Wait()
    fmt.Println("main")
}