早期的select函数是用来监控一系列的文件句柄,一旦其中一个文件句柄发生IO操作,该select调用就会被返回。golang在语言级别直接支持select,用于处理异步IO问题。
select用法同switch类似,如下:
timeout := make (chan bool, 1)
ch := make(chan int)
select {
case <-ch:
case <-timeout:
fmt.Println("timeout!")
default:
fmt.Println("default case is running")
}
可以看出,ch初始化后,case1读取失败,timeout同样失败,因为channel中无数据,直接跳至default执行并返回。
注意,如果没有default,select 会一直等待等到某个 case 语句完成, 也就是等到成功从 ch 或者 timeout 中读到数据,否则一直阻塞。
基于这种机制,可以使用select实现channel读取超时的机制
package main import (
"fmt"
"time"
) func main() {
timeout := make(chan bool, )
go func() {
time.Sleep(3e9) // sleep 3 seconds
timeout <- true
}()
ch := make(chan int)
select {
case <-ch:
case <-timeout:
fmt.Println("timeout!")
}
}
注意这里一定不能用default,否则3s超时还未到直接执行default,case2便不会执行,超时机制便不会实现。timeout会在3s超时后读取到数据。
使用select判断channel是否存满
ch1 := make(chan int, )
ch2 := make(chan int, )
select {
case <-ch1:
fmt.Println("ch1 pop one element")
case <-ch2:
fmt.Println("ch2 pop one element")
default:
fmt.Println("default")
}
如果case1、case2均未执行,则说明ch1&ch2已满,over.....