golang 中 sync包的 WaitGroup

标签：子线程   height   int   span   line   col   main   一个   style   

　　golang 中的 sync 包有一个很有用的功能，就是 WaitGroup

　　先说说 WaitGroup 的用途：它能够一直等到所有的 goroutine 执行完成，并且阻塞主线程的执行，直到所有的 goroutine 执行完成。

　　WaitGroup 总共有三个方法：Add(delta int), Done(), Wait()。简单的说一下这三个方法的作用。

　　Add：添加或者减少等待 goroutine 的数量

　　Done：相当于Add(-1)

　　Wait：执行阻塞，直到所有的WaitGroup数量变成 0

请看例子：

package main

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

func main() {
    var wg sync.WaitGroup

    for i := 0; i < 5; i = i + 1 {
        wg.Add(1)
        go func(n int) {
            // defer wg.Done()
            defer wg.Add(-1)
            EchoNumber(n)
        }(i)
    }

    wg.Wait()
}

func EchoNumber(i int) {
    time.Sleep(3e9)
    fmt.Println(i)
}

结果如下所示：

0
1
2
3
4

　　程序很简单，只是将每次循环的数量过 3秒钟 输出。那么，这个程序如果不用 WaitGroup，那么将看不见输出结果。因为 goroutine 还没执行完，主线程已经执行完毕。注释的 defer wg.Done() 和 defer wg.Add(-1) 作用一样。这个很好，原来执行脚本，都是使用 time.Sleep，用一个估计的时间等到子线程执行完。WaitGroup 很好。虽然 chanel 也能实现，但是觉得如果涉及不到子线程与主线程数据同步，这个感觉不错。

golang 中 sync包的 WaitGroup

标签：子线程   height   int   span   line   col   main   一个   style   

原文地址：http://www.cnblogs.com/liang1101/p/7629916.html