Golang并发编程:使用sync.WaitGroup实现线程的等待 - Go语言中文社区

Golang并发编程:使用sync.WaitGroup实现线程的等待


首先看如下的代码:

func main() {

	go func() {
		fmt.Println("123")
	}()

	go func() {
		fmt.Println("ABC")
	}()

}

发现最后控制台没有任何输出,如图:

控制台显示情况
可以使用通道解决:

func main() {

	ch := make(chan int)
	count := 2

	go func() {
		fmt.Println("123")
		ch <- 1
	}()

	go func() {
		fmt.Println("ABC")
		ch <- 1
	}()

	for range ch {
		count--
		if count == 0 {
			close(ch)
		}
	}

}

可以解决:

123
ABC

其实,golang中内置了一套简单用于等待的编程方式:sync.WaitGroup,使用方式如下:

func main() {
	wg := &sync.WaitGroup{}
	wg.Add(2)
	go func() {
		fmt.Println("123")
		wg.Done()
	}()

	go func() {
		fmt.Println("ABC")
		wg.Done()
	}()

	wg.Wait()

}

可以解决:

123
ABC

总结:
WaitGroup内部实现了一个计数器,用来记录未完成的操作个数,类似于方法一的“count” ,共提供了3个方法:
Add() : 用于添加计数个数
Done(): 用于表示某任务已经完成,未完成任务计数减1
Wait(): 用来等待所有的操作结束(即计数变为0),该函数会在计数不为零的时候等待,在计数为零的时候立即返回。

可见用sync.WaitGroup是最简单的方式。

版权声明:本文来源CSDN,感谢博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
原文链接:https://blog.csdn.net/wxs19970115/article/details/98487904
站方申明:本站部分内容来自社区用户分享,若涉及侵权,请联系站方删除。
  • 发表于 2020-02-25 01:17:57
  • 阅读 ( 940 )
  • 分类:Go

0 条评论

请先 登录 后评论

官方社群

GO教程

猜你喜欢