golang 使用Sync.WaitGroup来解决等待问题

面对goroutine我们都需要等待它完成交给它的事情，等待它计算完成或是执行完毕，所以不得不在程序需要等待的地方使用time.Sleep()来睡眠一段时间，等待其他goroytine执行完毕，下面的代码打印1到100的for循环可以在很快的时间内运行完毕，但是我们必须添加time.Sleep()来等待其打印完毕,如果我们不等待仿佛什么也没有发生一样。。。。。这肯定不是我们想要的！

 

func main(){
    for i := 0; i < 100 ; i++{
        go fmt.Println(i)
    }
    time.Sleep(time.Second)
}

这就是我最开始学习goroutine想到的笨方法。。。。死等！ 等不到 time.Second*2 在等不到*3

但是对于我们实际的使用场景来说，1秒也许等待的时间有点长，而有时候是远远不够的，最主要的我们无法预知gorotine的运行时间长短，所以也无法指定Sleep睡眠的时间。

正好前几天学习了channel，那么现在我们就可以使用channel来达到我们的目的了：

func main() {
    c := make(chan bool, 100)
    for i := 0; i < 100; i++ {
        go func(i int) {
            fmt.Println(i)
            c <- true
        }(i)
    }

    for i := 0; i < 100; i++ {
        <-c
    }
}

 

 

完美输出！这里我们可以从控制台看到我们的胜利成果，但是我们有更好的处理方式：sync.WaitGroup

WaitGroup对象内部有一个计数器，最初从0开始，它有三个方法：Add() Done() Wait() 用来控制计数器的数量从而达到等待的目的。

func main() {
   wg := sync.WaitGroup{}
   wg.Add(100)
   for i := 0; i < 100; i++ {
      go func(i int) {
         defer wg.Done()
         fmt.Print(i)
      }(i)
   }
   wg.Wait()
}

 

首先使用 add(100) 把计数器的数量设置为100，每次循环运行完毕都把计数器减1，主函数使用Wait(）

阻塞，直到计数器归0结束主线程结束阻塞。

这里会阻塞主线程！！！需要注意。

另外：以上三种方式打印出的数字皆无序

type WaitGroup struct {
	noCopy noCopy
	state1 [3]uint32
}

它只是一个struct所以当参数传递的时候需要传递指针。