多路复用channel的时候,如何优雅的关闭通道

最近在学习Go语言,学了有一个月时间,从微信公众号“码洞”上看到了快学系列教材,很棒的文章。最新的一篇是讲通道,里面提到了多路复用channel,但是没有讲在多路复用中,如何优雅的关闭通道,我在这里做了一些尝试,发出来供交流。

《快学 Go 语言》第 12 课 —— 神秘的地下通道 的访问地址如下,先阅读:
https://mp.weixin.qq.com/s/eZTFa-8drmkZUXsvHycorQ

然后下面是我增加通道关闭后的代码,详见注释:

package main
import (
  "fmt"
  "time"
)
func send(ch chan int, gap time.Duration) {
  //每个通道模拟发送5个消息就结束
  for i := 0;i<5;i++{
    //i++
    ch <- i
    time.Sleep(gap)
  }
  //关闭通道
  close(ch)
}
func recv(ch1 chan int, ch2 chan int) {
  //定义临时变量,记录通道状况,一旦关闭就标记
  b1,b2:=true,true
  for i := 0;;i++{
    if(b1==false && b2==false ){
      //如果多个通道都关闭了,退出
      return
    }
    select {
      case v, ok := <- ch1:
        //fmt.Printf("recv %d from ch1\n", v)
        if (ok){
          fmt.Printf("%d :recv %d from ch1 %v \n", i,v,ok)
        } else {
          //标记ch1关闭
          b1 = ok
        }
        
      case v, ok := <- ch2:
        //fmt.Printf("recv %d from ch2\n", v)
        if (ok){
          fmt.Printf("%d :recv %d from ch2 %v \n", i,v,ok)
        } else {
          //标记ch2关闭
          b2 = ok
        }
    }
  }
}
func main() {
  var ch1 = make(chan int)
  var ch2 = make(chan int)
  //启动ch1通道
  go send(ch1, time.Second)
  //启动ch2通道
  go send(ch2, 2 * time.Second)
  recv(ch1, ch2)
}

• 编辑
• 预览