从零学习 Go 语言(27):学习 Go 协程里的WaitGroup
在前两篇文章里,我们学习了 协程
和 信道
的内容,里面有很多例子,当时为了保证 main goroutine 在所有的 goroutine 都执行完毕后再退出,我使用了 time.Sleep 这种简单的方式。
由于写的 demo 都是比较简单的, sleep 个 1 秒,我们主观上认为是够用的。
但在实际开发中,开发人员是无法预知,所有的 goroutine 需要多长的时间才能执行完毕,sleep 多了吧主程序就阻塞了, sleep 少了吧有的子协程的任务就没法完成。
因此,使用time.Sleep 是一种极不推荐的方式,今天主要就要来介绍 一下如何优雅的处理这种情况。
1. 使用信道来标记完成
“不要通过共享内存来通信,要通过通信来共享内存”
学习了信道后,我们知道,信道可以实现多个协程间的通信,那么我们只要定义一个信道,在任务完成后,往信道中写入true,然后在主协程中获取到true,就认为子协程已经执行完毕。
import "fmt"
func main() {
done := make(chan bool)
go func() {
for i := ; i < 5; i++ {
fmt.Println(i)
}
done <- true
}()
<-done
}
相关文章