主协程如何等待其余协程完再操作

题目序号：(421, 3235)

题目来源：腾讯

频次：2

答案1：(dema)

1. Sleep(不常用)，在main方法退出之前 sleep 一段时间

   package main
   
   import (
       "fmt"
       "time"
   )
   
   func main() {
       go func1()
       fmt.Println("main")
       // 这里睡眠一秒，等待其他协程结束
       time.Sleep(1 * time.Second)
   }
   
   func func1() {
       fmt.Println("func1")
   }

2. 通道(不常用)，每个活动结束从通道中取出一个值，当所有活动的协程结束，关闭通道

   package main
   
   import (
       "fmt"
   )
   
   var (
       ch    = make(chan struct{})
       count = 1 // 表示协程数
   )
   
   func main() {
       go func1()
       fmt.Println("main")
       for range ch {
           // 每次取值将count减1，当count为0时关闭通道
           count--
           if count == 0 {
               close(ch)
           }
       }
   }
   
   func func1() {
       fmt.Println("func1")
       ch <- struct{}{}
   }

3. sync.WaitGroup(常用)

   package main
   
   import (
       "fmt"
       "sync"
   )
   
   var wg sync.WaitGroup
   
   func main() {
       wg.Add(1) // 添加计数
       go func1()
       fmt.Println("main")
       wg.Wait() // 等待计数变为0
   }
   
   func func1() {
       fmt.Println("func1")
       wg.Done() // 将计数减1
   }