最近公司工做有点多,Golang的select进阶就这样被拖沓啦,今天坚持把时间挤一挤,把吹的牛皮补上。golang
前一篇文章《Golang并发模型:轻松入门select》介绍了select的做用和它的基本用法,此次介绍它的3个进阶特性。后端
nil
的通道永远阻塞for-select
select{}
阻塞nil
的通道永远阻塞当case
上读一个通道时,若是这个通道是nil
,则该case
永远阻塞。这个功能有1个妙用,select
一般处理的是多个通道,当某个读通道关闭了,但不想select
再继续关注此case
,继续处理其余case
,把该通道设置为nil
便可。
下面是一个合并程序等待两个输入通道都关闭后才退出的例子,就使用了这个特性。缓存
func combine(inCh1, inCh2 <-chan int) <-chan int { // 输出通道 out := make(chan int) // 启动协程合并数据 go func() { defer close(out) for { select { case x, open := <-inCh1: if !open { inCh1 = nil continue } out<-x case x, open := <-inCh2: if !open { inCh2 = nil continue } out<-x } // 当ch1和ch2都关闭是才退出 if inCh1 == nil && inCh2 == nil { break } } }() return out }
break
在select
内的并不能跳出for-select
循环。看下面的例子,consume
函数从通道inCh
不停读数据,期待在inCh
关闭后退出for-select
循环,但结果是永远没有退出。bash
func consume(inCh <-chan int) { i := 0 for { fmt.Printf("for: %d\n", i) select { case x, open := <-inCh: if !open { break } fmt.Printf("read: %d\n", x) } i++ } fmt.Println("combine-routine exit") }
运行结果:并发
➜ go run x.go for: 0 read: 0 for: 1 read: 1 for: 2 read: 2 for: 3 gen exit for: 4 for: 5 for: 6 for: 7 for: 8 ... // never stop
既然break
不能跳出for-select
,那怎么办呢?给你3个锦囊:less
case
内,使用return
,若是有结尾工做,尝试交给defer
。select
外for
内使用break
挑出循环,如combine
函数。goto
。select{}
永远阻塞select{}
的效果等价于建立了1个通道,直接从通道读数据:函数
ch := make(chan int) <-ch
可是,这个写起来多麻烦啊!没select{}
简洁啊。
可是,永远阻塞能有什么用呢!?
当你开发一个并发程序的时候,main
函数千万不能在子协程干完活前退出啊,否则全部的协程都被迫退出了,还怎么提供服务呢?
好比,写了个Web服务程序,端口监听、后端处理等等都在子协程跑起来了,main
函数这时候能退出吗?spa
最后,介绍下我经常使用的select
场景:code
select
本色:多通道处理
- 若是这篇文章对你有帮助,请点个赞/喜欢,鼓励我持续分享,感谢。
- 个人文章列表,点此可查看
- 若是喜欢本文,随意转载,但请保留此原文连接。