WaitGroup用于线程同步,WaitGroup等待一组线程集合完成,才会继续向下执行。 主线程(goroutine)调用Add来设置等待的线程(goroutine)数量。 然后每个线程(goroutine)运行,并在完成后调用Done。 同时,Wait用来阻塞,直到所有线程(goroutine)完成才会向下执行。

对官方的代码做简单修改:

  1. package main  
  2.   
  3. import (  
  4.     "fmt"  
  5.     "sync"  
  6. )  
  7.   
  8. func main() {  
  9.     var wg sync.WaitGroup  
  10.     var urls = []string{  
  11.         "http://www.golang.org/",  
  12.         "http://www.google.com/",  
  13.         "http://www.somestupidname.com/",  
  14.     }  
  15.     for _, url := range urls {  
  16.         // Increment the WaitGroup counter.  
  17.         wg.Add(1)  
  18.         go func(url string) {  
  19.             // Launch a goroutine to fetch the URL.  
  20.             defer wg.Done()  
  21.             // Fetch the URL.  
  22.             fmt.Println(url)  
  23.         }(url)  
  24.     }  
  25.     // Wait for all goroutines to finish.  
  26.     wg.Wait()  
  27.     fmt.Println("Game Over")  
  28. }  

执行结果为:
  1. http://www.somestupidname.com/  
  2. http://www.golang.org/  
  3. http://www.google.com/  
  4. Game Over  
  5. 可以看出3个线程(goroutine)全部执行完成后,wg.Wait()才停止等待,程序继续往下执行,输出Game Over。
  6. 文章转载于:https://blog.csdn.net/aslackers/article/details/62046306