Gorountine和channel是Go语言最核心的功能,这个功能保证了go强大的并发执行能力 。如果你想在你继续执行你的goroutine之前等待所有的goroutines执行完毕,你该如何使用go语言解决这样一个问题呢?
下面,我将给出三中解决方案 。其中,最有解决方案是sync.WaitGroup 。
参考:https://blog.csdn.net/john_f_lau/article/details/56279803
参考:go语言中文文档:www.topgoer.com
程序函数等待利用time包中的Sleep方法等待一段时间,这样就引起另外一个问题,到底该等待多久呢?
下面是程序的实现 。
func receiveMsg(){msg := make(chan string)go func(){time.Sleep(time.Second * 1)msg <- "goroutine 1"}()go func(){time.Sleep(time.Second * 2)msg <- "goroutine 2"}()go func(){time.Sleep(time.Second * 3)msg <- "goroutine 3"}()go func(){for i := range msg {fmt.Println("message :", i)}}()time.Sleep(time.Second * 4)}func main() {receiveMsg()}12345678910111213141516171819202122232425262728
功能上解决了问题,但是,如果没法知道所有的goroutine总共执行,还是没有从根本上解决问题 。利用channel,可以有一个改良版的解决方案 。
channel等待首先,看看具体的程序实现 。
func receiveMsg2(){msg := make(chan string)done := make(chan bool)go func(){time.Sleep(time.Second * 3)msg <- "goroutine 1"done <- true}()go func(){time.Sleep(time.Second * 2)msg <- "goroutine 2"done <- true}()go func(){time.Sleep(time.Second * 1)msg <- "goroutine 3"done <- true}()go func(){//This statement is to make sure that all message is received.time.Sleep(time.Second * 3)for i := range msg {fmt.Println("message :", i)}}()for j := 0 ; j < 3; j++ {<-done}}func main() {receiveMsg2()}12345678910111213141516171819202122232425262728293031323334353637
从上面程序而言,程序添加了time.Sleep(time.Second * 1) 。添加这行代码是为了保证消息被完全接受(打印到控制台) 。由这个程序的解决方案引出了另外一个问题:加入我们不知道到底有多少个goroutine该怎么办?
sync.WaitGroupsync包下面的WaitGroup结构体就是为解决这样的问题而设计的,官方定义如下:
A WaitGroup waits for a collection of goroutines to finish. The main goroutine calls Add to set the number of goroutines to wait for. Then each of the goroutines runs and calls Done when finished. At the same time, Wait can be used to block until all goroutines have finished.使用sync.WaitGroup有四个步骤:
1. 创建一个新的sync.WaitGroup实例wg(假定代码,下面描述要用到);
2. 当需要使用goroutine的时候,调用wg.Add(1) (使用一次调用一次,如果知道到有N个goroutine,可以直接设置对应的N个);
3. 当goroutine执行完毕前,需要告诉WaitGroup执行完毕,调用对应代码defer wg.Done();
4. 在需要等待所有goroutine执行完毕时,调用代码wg.Wait() 。
func receiveMsg3(){msg := make(chan string)var wg sync.WaitGroupwg.Add(3)go func(){defer wg.Done()time.Sleep(time.Second * 3)msg <- "goroutine 1"}()go func(){defer wg.Done()time.Sleep(time.Second * 2)msg <- "goroutine 2"}()go func(){defer wg.Done()time.Sleep(time.Second * 1)msg <- "goroutine 3"}()go func(){time.Sleep(time.Second * 3)for i := range msg {fmt.Println("message :", i)}}()wg.Wait()}func main() {receiveMsg3()}1234567891011121314151617181920212223242526272829303132
通过上面的代码就是按照这4个步骤来构建整个函数的,运用sync.WaitGroup可以顺利实现并发等待操作 。【Go如何保证gorountine执行完毕后继续执行】
推荐阅读
- Matlab如何调用动态连接库DLL文件的函数?
- 废旧电池如何处理
- 老茶头究竟能不能喝,如何冲泡老茶头
- 如何进行环保装修
- 无法重装系统?笔记本如何引导BIOS设置?
- HTTP长连接是啥?底层是如何工作的?Tomcat是如何实现长连接的?
- 如何科学地选择一台适合 Java 开发的电脑?
- 如何选购iPad
- 如何避免过度设计,正确地使用设计模式?
- 如何为机器学习工程设计Python接口