golang的select用法

Posted 2020-07-14

早期的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.....