golang学习之select用法

caiyezi · · 904 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

早期的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") }