早期的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, 1) 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, 1) ch2 := make(chan int, 1) 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.....
有疑问加站长微信联系(非本文作者)