【GOLANG】Go语言学习-select用法

发布时间：2020-12-16 18:12:53 所属栏目：大数据来源：网络整理

导读：【GOLANG】Go语言学习-select用法 golang 的 select 的功能和 select,poll,epoll 相似，就是监听 IO 操作，当 IO 操作发生时，触发相应的动作。示例： ch1 := make (chan int , 1 )ch2 := make (chan int ,0)">1 )... select { case -ch1: fmt.Println( "c

golang 的 select 的功能和select,poll,epoll相似，就是监听 IO 操作，当 IO 操作发生时，触发相应的动作。

示例：

ch1 := make (chan int,1)
ch2 := make (chan int,0)">1)

...

select {
case <-ch1:
    fmt.Println("ch1 pop one element")
case <-ch2:
    fmt.Println("ch2 pop one element")
}

注意到 select 的代码形式和 switch 非常相似，不过 select 的 case 里的操作语句只能是【IO 操作】。

此示例里面 select 会一直等待等到某个 case 语句完成，也就是等到成功从 ch1 或者 ch2 中读到数据。则 select 语句结束。

【使用 select 实现 timeout 机制】

如下：

timeout := make (chan bool,0)">1)
go func() {
    time.Sleep(1e9) // sleep one second
    timeout <- true
}()
ch := make (chan int)
select {
case <- ch:
case <- timeout:
    fmt.Println("timeout!")
} 

当超时时间到的时候，case2 会操作成功。所以 select 语句则会退出。而不是一直阻塞在 ch 的读取操作上。从而实现了对 ch 读取操作的超时设置。

下面这个更有意思一点。

当 select 语句带有 default 的时候：

1)

select {
case <-ch1:
    fmt.Println("ch2 pop one element")
default:
    fmt.Println("default")
} 

此时因为 ch1 和 ch2 都为空，所以 case1 和 case2 都不会读取成功。则 select 执行 default 语句。

就是因为这个 default 特性，我们可以使用 select 语句来检测 chan 是否已经满了。

ch := make (chan int,0)">1) ch <- 1 select { case ch <- 2: default: fmt.Println("channel is full !") }

因为 ch 插入 1 的时候已经满了，当 ch 要插入 2 的时候，发现 ch 已经满了（case1 阻塞住），则 select 执行 default 语句。这样就可以实现对 channel 是否已满的检测，而不是一直等待。

比如我们有一个服务，当请求进来的时候我们会生成一个 job 扔进 channel，由其他协程从 channel 中获取 job 去执行。但是我们希望当 channel 瞒了的时候，将该 job 抛弃并回复【服务繁忙，请稍微再试。】就可以用 select 实现该需求

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!