通过select语句可以监听channel上的数据流动
Golang的select
语句类似于UNIX的select()
函数的轮询机制,在UNIX中可通过调用select()
函数来监控一系列的文件句柄(文件描述符,数量有限),一旦某个文件句柄发生了I/O动作select()
调用就会被返回。该机制也被用于高并发的Socket服务器程序中。
select
来源于网络I/O模型中的select
,本质上I/O多路复用技术,只不过Golang中的select
基于的并非网络而是channel
。
select {
case communication clause:
statement(s)
case communication clause:
statement(s)
default:
statement(s)
}
select
语句是Golang中的控制结构,类似用于通信的switch
语句,也被称为channel
开关。select
语句会等待channel
准备就绪(收发操作),以便在不同的case
下执行。由select
开始的一个新的选择块,每个选择条件由case
语句来描述。与switch
语句相比,select
有较多限制,最大的限制在于每个case
语句必须是一个I/O操作。
select{
case v1 := <-ch1:
fmt.Printf("[CH1] received: %v\n", v1)
case v2,ok := <-ch2:
if ok{
fmt.Printf("[CH2] received: %v\n", v2)
}else{
fmt.Printf("[CH2] closed\n")
}
case ch3 <-msg:
fmt.Printf("[CH3] sent: %v\n", msg)
default:
fmt.Printf("NO Communicating\n")
}
例如:上例select
语句拥有四个case
子句,前两个是channel
的receive
接收操作,第三个是channel
的send
发送操作,最后一个default
默认操作。当代码执行到select
语句时,case
子句会按源代码的顺序进行评估,且只评估一次。评估结果会出现下面几种情况:
- 除
default
外,若只有一个case
评估通过则执行此case
中的语句。 - 除
default
外,若用多个case
评估通过则随机挑选一个执行。 - 除
default
外,所有的case
评估都不通过,则执行default
。 - 若没有
default
则select
代码块发生阻塞,直到有一个case
通过评估,否则一直阻塞。 - 若
case
中receive
接收的是nil
则也会发生阻塞
监听
Golang的select
用来监听和channel
有关的I/O操作,可监听进入channel
时的数据,也可以是用channel
发送值时。当I/O操作发生时会触发相应地动作,因此每个case
都必须是一个I/O操作,确切的说应该是一个面向channel
的I/O操作。
例如:定时器
ticker1 := time.NewTicker(time.Second * 1)
ticker2 := time.NewTicker(time.Second * 3)
for{
select{
case <-ticker1.C:
fmt.Printf("[1] TICK\n")
case <-ticker2.C:
fmt.Printf("[2] TICK\n")
}
}
在执行select
语句时,运行时系统会自上而下地判断每个case
中的发送或接收操作是否可以被立即执行,所谓立即执行即当前goroutine
不会因此操作而被阻塞。
select
语句只能用于channel
的读写操作
例如:使用select
来检测channel
是否已满
ch := make(chan int, 1)
ch<-1
select{
case ch<-2:
fmt.Printf("send to channel\n")
default:
fmt.Printf("channel is full\n")
}
特性
- 每个
case
都必须是一个channel
- 所有
channel
表达式都会被求值 - 所有被发送的表达式都会被求值
- 若任意某个
channel
可执行就执行,其它被忽略。
例如:获取斐波拉兹数列
func fib(ch, quit chan int){
x,y := 0,1
for{
select{
case ch <- x:
x,y = y, x+y
case <-quit:
fmt.Printf("QUIT\n")
return
}
}
}
func main(){
ch := make(chan int)
quit := make(chan int)
go func(){
for i:=0; i<10; i++{
fmt.Println(<-ch)
}
quit<-0
}()
fib(ch, quit)
}
case
-
select
中每个case
必须是一个channel
操作,要么是发送要么是接收。
select
执行过程中必须命中某一case
分支,若在遍历所有case
后都没有命中,则会进入default
分支。若没有default
分支则select
发生阻塞,直到某个case
可以命中。若一直都没有命中,则select
抛出deadlock
死锁错误。
- 循环中每次
select
都会对所有channel
表达式求值
例如:通过time.After
实现定时器,定时任务可通过done channel
停止。
done := make(chan bool, 1)
close(done)
for{
select{
case <-time.After(time.Second):
fmt.Printf("Time after\n")
case <-done:
//读取零值 false
fmt.Printf("Read done\n")
}
}
- 若多个
case
满足读写条件,select
会随机选择一个case
来执行。
select
会随机执行一个可运行的case
,若没有case
可以运行则阻塞,直到有case
可以运行。
select
可用于多个channel
进行读写操作时仅需一次只处理一个的情况。
ch := make(chan int, 1024)
go func(ch chan int){
for{
v := <-ch
fmt.Printf("value = %v\n", v)
}
}(ch)
ticker := time.NewTicker(time.Second * 1)
for i:=0; i<5; i++{
select{
case ch<-i:
case <-ticker.C:
fmt.Printf("%d: Ticker\n", i)
}
time.Sleep(time.Microsecond * 500)
}
close(ch)
ticker.Stop()
若ticker.C
和ch
同时满足读写条件时,select
会随机地选择一个来执行,导致看起来一些数据丢了。
- 对于
case
条件语句中若存在channel
值为nil
的读写操作,则该分支会被忽略。
var ch chan int
go func(ch chan int){
ch <- 100
}(ch)
select {
case <-ch:
fmt.Printf("Channel recieved\n")
}
发生错误:fatal error: all goroutines are asleep - deadlock!
default
select
语句会被阻塞,直到其中一个case
被执行。若select
中没有任何case
,它将永远阻塞,从而导致死锁。
例如:空select{}
引发死锁
func main(){
select {
}
}
对于空的select
语句,程序会被阻塞,准确来说是当前goroutine
会被阻塞。Golang自带死锁检测机制,发现当前goroutine
再也没有机会被唤醒时,则会panic
。
fatal error: all goroutines are asleep - deadlock!
通过带default
的select
实现非阻塞读写,用于防止select
发生阻塞。
select{
default:
}
多个case
运行时select
会随机公平地选出一个执行,其它不会执行。若存在default
子句则会执行该语句。若没有default
则select
阻塞,直到某个通信可以运行。Go不会重新对channel
或值进行求值。
当select
语句永远阻塞,没有其它goroutine
写入此channel
时,将导致死锁。
ch := make(chan int)
select{
case <-ch:
}
fatal error: all goroutines are asleep - deadlock!
若存在默认情况default
则不会发生死锁deadlock
,因为在没有其它case
准备就绪时将执行default
默认情况。
ch := make(chan int)
select{
case <-ch:
default:
fmt.Println("default case executed")
}
例如:典型生产者消费者模式
func main(){
ch1 := make(chan int)
ch2 := make(chan int)
//生产者
go pump1(ch1)
go pump2(ch2)
//消费者
go suck(ch1, ch2)
time.Sleep(1e9)
}
-
ch1
和ch2
在无限循环中通过pump1()
和pump2()
填充整数
func pump1(ch chan int){
for i:=0; ; i++{
ch <- i * 1
}
}
func pump2(ch chan int){
for i:=0; ; i++{
ch <- i * 2
}
}
-
suck()
在无限循环中轮询输入项,通过select
语句获取不同信道的整数并输出。
func suck(ch1,ch2 chan int){
for{
select{
case v := <-ch1:
fmt.Printf("[CH1] receive %d\n", v)
case v := <-ch2:
fmt.Printf("[CH2] receive %d\n", v)
default:
fmt.Printf("NO Communicating\n")
}
}
}
选择select
的哪一个case
取决于哪个信道接收到了消息。
timeout
当case
中的channel
始终没有接收到数据,同时也没有提供default
语句时,select
语句整体会发生阻塞。有时并不希望select
一直阻塞下去,此时可手动设置一个超时时间。
func expire(ch chan bool, t int){
time.Sleep(time.Second * time.Duration(t))
ch <- true
}
func main(){
timeout := make(chan bool, 1)
go expire(timeout, 2)
ch1 := make(chan string, 1)
ch2 := make(chan string, 1)
select{
case msg1 := <-ch1:
fmt.Printf("[CH1] received: %s\n", msg1)
case msg2 := <-ch2:
fmt.Printf("[CH2] received: %s\n", msg2)
case <-timeout:
fmt.Printf("[EXPIRE] exit\n")
}
}
例如:使用select
实现channel
的读取超时机制,不能使用default
否则3秒超时未到,就会直接执行default
。
timeout := make(chan bool, 1)
go func(){
time.Sleep(time.Second * 3)
timeout <- true
}()
ch := make(chan int)
select{
case <-ch:
case <-timeout:
fmt.Printf("TIMEOUT\n")
}
可使用time.After
实现超时控制
ch := make(chan int)
select{
case <-ch:
fmt.Printf("read from ch\n")
case <-time.After(time.Second * time.Duration(3)):
fmt.Printf("[TIMEOUT] exit\n")
}
for
当for
和select
结合时,break
是无法跳出for
之外的,若需break
出来需添加标签使用goto
,或break
到具体为止。
- 解决方案1:使用Golang中
break
的特性在外层for
上添加一个标签 - 解决方案2:使用
goto
直接跳出循环到指定标记位置
- 对于
for
中空的select{}
也有可能会引起CPU占用过高的问题
ch := make(chan bool)
for i:=0; i
一般来说,使用select
监听各个case
的I/O事件,每个case
都是阻塞的。上例中原本希望select
在获取到ch
里的数据时立即退出循环,但由于在for
循环中,第一次读取ch
后仅仅退出了select
但并未退出for
,因此下次哈希继续执行select
逻辑,此时将永远是执行default
,直到ch
里读取到数据。否则会一直在一个死循环中运行,因此即便只是放到一个goroutine
中运行,也会占满所有的CPU。解决的方式直接把default
拿掉,这样select
会一直阻塞在ch
通道的I/O上,当ch
有数据时就可以随时响应通道中的信息。
select
实现了一种监听模式,通常用在(无限)循环中,在某中情况下可通过break
语句使循环退出。
ch := make(chan int)
//定时2s
ticker := time.NewTicker(time.Second * 2)
defer ticker.Stop()
//发送信号
go func(ch chan int){
time.Sleep(time.Second * 5)
ch <- 1
}(ch)
//监听I/O
for{
select{
case <-ticker.C:
fmt.Printf("task running...\n")
case result,ok := <-ch:{
if ok{
fmt.Printf("chan number is %v\n", result)
break
}
}
}
}
fmt.Printf("END\n")
例如:
ch := make(chan int)
quit := make(chan bool)
//写数据
go func() {
//循环写入
for i := 0; i < 5; i++ {
ch <- i
time.Sleep(time.Second)
}
//关闭channel
close(ch)
//通知主goroutine推出
quit <- true
//退出当前goroutine
runtime.Goexit()
}()
//主goroutine 读数据
for {
select {
case num := <-ch:
fmt.Printf("received number is %d\n", num)
case <-quit:
fmt.Printf("quit\n")
//break //跳出select
return //终止进程
}
fmt.Printf("==================\n")
}
多路复用
select
是Golang在语言层面提供的多路I/O复用机制,它可检测多个channel
是否ready
(是否可读或可写)。
select
是如何实现多路复用的,为什么没有在第一个channel
操作时阻塞,从而导致后面的case
都执行不了。