goroutine模拟了线程级别的返场的能力,但它的执行需要主协程给机会。一般的作法用sleep,chan阻塞,看起来让人不爽,本文介绍sync.WaitGroup 类型结合 defer 的特性,给出优雅的解决方案。
缘起
下面这段代码众所周知不会打招呼
....
func main(){
go sayHi(){
fmt.Println("say hello......")
}()
fmt.Println("main groutine....")
}
等待
上述代码不会讲hello,在于主协程main无等待(阻塞),子协程sayHi没有露脸的机会。
为了让子协程sayHai上场,通常在主协程末了加这么一句,让它睡会儿
time.Sleep(1e9)
但想想不科学,如果子协程在它睡期间,没能完成任务,超时了,子协程仍然打不了招呼。又或者子协程完成了,主协程还在睡,岂不是主线程不作为?睡的时间多久怎样才合理...
通道
用通道可解决阻塞时间合理性质疑。
若启用了多个子协程,可以这样实现主协程等待子协程执行完毕并退出的:声明一个和子协程数量一致的通道数组,然后为每个子协程分配一个通道元素,在子协程执行完毕时向对应的通道发送数据;然后在主协程中,依次读取这些通道接收子协程发送的数据,只有所有通道都接收到数据才会退出主协程。
代码看起来像这样
chs := make([]chan in