Go WaitGroup的使用

Posted 2020-11-19 lanyangsh

开发过程中，经常task之间的同步问题。例如，多个子task并发完成一部分任务，主task等待他们最后结束。

在Go语言，实现同步的一种方式就是WaitGroup。

Example

package main 

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



func main() {

    var wg sync.WaitGroup


    wg.Add(3)
    go func(n int){
        fmt.Println("n:", n)
        t := time.Duration(n)*time.Second
        time.Sleep(t)

        wg.Done()
    }(1)


    go func(n int){
        fmt.Println("n:", n)
        t := time.Duration(n)*time.Second
        time.Sleep(t)
        time.Sleep(t)

        wg.Done()
    }(2)

    go func(n int){
        fmt.Println("n:", n)
        t := time.Duration(n)*time.Second
        time.Sleep(t)

        wg.Done()
    }(3)

    wg.Wait()

    fmt.Println("main exit...")
}

output:

n: 3

n: 1

n: 2

main exit...

注意

WaitGroup变量定义后，是不允许被拷贝的，即不允许作为函数参数或者赋值给其他变量。