1. 前言

channel一般用于协程之间的通信,channel也可以用于并发控制。比如主协程启动N个子协程,主协程等待所有子协程退出后再继续后续流程,这种场景下channel也可轻易实现。

2. 场景示例

下面程序展示一个使用channel控制子协程的例子:

  1. package main
  2. import (
  3. "time"
  4. "fmt"
  5. )
  6. func Process(ch chan int) {
  7. //Do some work...
  8. time.Sleep(time.Second)
  9. ch <- 1 //管道中写入一个元素表示当前协程已结束
  10. }
  11. func main() {
  12. channels := make([]chan int, 10) //创建一个10个元素的切片,元素类型为channel
  13. for i:= 0; i < 10; i++ {
  14. channels[i] = make(chan int) //切片中放入一个channel
  15. go Process(channels[i]) //启动协程,传一个管道用于通信
  16. }
  17. for i, ch := range channels { //遍历切片,等待子协程结束
  18. <-ch
  19. fmt.Println("Routine ", i, " quit!")
  20. }
  21. }

上面程序通过创建N个channel来管理N个协程,每个协程都有一个channel用于跟父协程通信,父协程创建完所有协程中等待所有协程结束。

这个例子中,父协程仅仅是等待子协程结束,其实父协程也可以向管道中写入数据通知子协程结束,这时子协程需要定期的探测管道中是否有消息出现。

3. 总结

使用channel来控制子协程的优点是实现简单,缺点是当需要大量创建协程时就需要有相同数量的channel,而且对于子协程继续派生出来的协程不方便控制。

后面继续介绍的WaitGroup、Context看起来比channel优雅一些,在各种开源组件中使用频率比channel高得多。