go语言中——select的使用方法

tydldd 2020-07-30

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

参考链接:https://studygolang.com/articles/7203

相关推荐

jiong / 0评论 2020-09-17