27. 学习 Go 协程：WaitGroup

Hi，你们好，我是明哥。

在本身学习 Golang 的这段时间里，我写了详细的学习笔记放在个人我的微信公众号 《Go编程时光》，对于 Go 语言，我也算是个初学者，所以写的东西应该会比较适合刚接触的同窗，若是你也是刚学习 Go 语言，不防关注一下，一块儿学习，一块儿成长。

个人在线博客：http://golang.iswbm.com
个人 Github：github.com/iswbm/GolangCodingTime

---

在前两篇文章里，咱们学习了 协程 和 信道 的内容，里面有不少例子，当时为了保证 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 := 0; i < 5; i++ {
			fmt.Println(i)
		}
		done <- true
	}()
	<-done
}

输出以下

0
1
2
3
4

2. 使用 WaitGroup

上面使用信道的方法，在单个协程或者协程数少的时候，并不会有什么问题，但在协程数多的时候，代码就会显得很是复杂，有兴趣能够本身尝试一下。

那么有没有一种更加优雅的方式呢？

有，这就要说到 sync包 提供的 WaitGroup 类型。

WaitGroup 你只要实例化了就能使用

var 实例名 sync.WaitGroup

实例化完成后，就可使用它的几个方法：

• Add：初始值为0，你传入的值会往计数器上加，这里直接传入你子协程的数量
• Done：当某个子协程完成后，可调用此方法，会从计数器上减一，一般可使用 defer 来调用。
• Wait：阻塞当前协程，直到实例里的计数器归零。

举一个例子：

import (
	"fmt"
	"sync"
)

func worker(x int, wg *sync.WaitGroup) {
	defer wg.Done()
	for i := 0; i < 5; i++ {
		fmt.Printf("worker %d: %d\n", x, i)
	}
}

func main() {
	var wg sync.WaitGroup

	wg.Add(2)
	go worker(1, &wg)
	go worker(2, &wg)

	wg.Wait()
}

输出以下

worker 2: 0
worker 2: 1
worker 2: 2
worker 2: 3
worker 2: 4
worker 1: 0
worker 1: 1
worker 1: 2
worker 1: 3
worker 1: 4

以上就是咱们在 Go 语言中实现一主多子的协程协做方式，推荐使用 sync.WaitGroup。。

系列导读

01. 开发环境的搭建（Goland & VS Code）

02. 学习五种变量建立的方法

03. 详解数据类型：****整形与浮点型

04. 详解数据类型：byte、rune与string

05. 详解数据类型：数组与切片

06. 详解数据类型：字典与布尔类型

07. 详解数据类型：指针

08. 面向对象编程：结构体与继承

09. 一篇文章理解 Go 里的函数

10. Go语言流程控制：if-else 条件语句

11. Go语言流程控制：switch-case 选择语句

12. Go语言流程控制：for 循环语句

13. Go语言流程控制：goto 无条件跳转

14. Go语言流程控制：defer 延迟调用

15. 面向对象编程：接口与多态

16. 关键字：make 和 new 的区别？

17. 一篇文章理解 Go 里的语句块与做用域

18. 学习 Go 协程：goroutine

19. 学习 Go 协程：详解信道/通道

20. 几个信道死锁经典错误案例详解

21. 学习 Go 协程：WaitGroup

22. 学习 Go 协程：互斥锁和读写锁

23. Go 里的异常处理：panic 和 recover

24. 超详细解读 Go Modules 前世此生及入门使用

25. Go 语言中关于包导入必学的 8 个知识点

26. 如何开源本身写的模块给别人用？

27. 说说 Go 语言中的类型断言？

28. 这五点带你理解Go语言的select用法

---