select作用
Go里面提供了一个关键字select，通过select可以监听channel上的数据流动。
select的用法与switch语言非常类似，由select开始一个新的选择块，每个选择条件由case语句来描述。
与switch语句可以选择任何可使用相等比较的条件相比， select有比较多的限制，其中最大的一条限制就是每个case语句里必须是一个IO操作，大致的结构如下：
select {
case <-chan1:
// 如果chan1成功读到数据，则进行该case处理语句
case chan2 <- 1:
// 如果成功向chan2写入数据，则进行该case处理语句
default:
// 如果上面都没有成功，则进入default处理流程
}
在一个select语句中，Go语言会按顺序从头至尾评估每一个发送和接收的语句。
如果其中的任意一语句可以继续执行(即没有被阻塞)，那么就从那些可以执行的语句中任意选择一条来使用。
如果没有任意一条语句可以执行(即所有的通道都被阻塞)，那么有两种可能的情况：
l 如果给出了default语句，那么就会执行default语句，同时程序的执行会从select语句后的语句中恢复。
l 如果没有default语句，那么select语句将被阻塞，直到至少有一个通信可以进行下去。
示例代码：
func fibonacci(c, quit chan int) {
x, y := 1, 1
for {
select {
case c <- x:
x, y = y, x+y
case <-quit:
fmt.Println("quit")
return
}
}
}
func main() {
c := make(chan int)
quit := make(chan int)
go func() {
for i := 0; i < 6; i++ {
fmt.Println(<-c)
}
quit <- 0
}()
fibonacci(c, quit)
}
运行结果如下：
图片3.png
超时
有时候会出现goroutine阻塞的情况，那么我们如何避免整个程序进入阻塞的情况呢？我们可以利用select来设置超时，通过如下的方式实现：
func main() {
c := make(chan int)
o := make(chan bool)
go func() {
for {
select {
case v := <-c:
fmt.Println(v)
case <-time.After(5 * time.Second):
fmt.Println("timeout")
o <- true
break
}
}
}()
//c <- 666 // 注释掉，引发 timeout
<-o
}
代码如下所示：
// code_049_select_and_timeout project main.go
package main
import (
"fmt"
"time"
)
/*
1)Go里面提供了一个关键字select，通过select可以监听channel上的数据流动。
2)select的用法与switch语言非常类似，由select开始一个新的选择块，每个选择条件由case语句来描述。
3)与switch语句可以选择任何可使用相等比较的条件相比， select有比较多的限制，其中最大的一条限制就是每个case语句里必须是一个IO操作
*/
func fibonacci(c, quit chan int) {
x, y := 1, 1
for {
select {
case c <- x: //此处执行了发送,如果成功向c写入数据，则进行该case处理语句
x, y = y, x+y
case <-quit: //此处执行了接收, 如果quit成功读到数据，则进行该case处理语句
//从输出结果可以看到, 此代码一直不断地向channel中写入数据。 在main函数中只是读取6个数据, 而写入的数据多于读取的数据
fmt.Println("quit：y=", y)
return
}
}
}
func main() {
//1) select的使用
c1 := make(chan int)
quit := make(chan int)
go func() {
for i := 0; i < 6; i++ {
fmt.Println(<-c1) //读取数据
}
quit <- 0
}()
fibonacci(c1, quit)
//2) goroutine阻塞,利用select设置超时
c := make(chan int)
o := make(chan bool)
go func() {
for {
select {
case v := <-c:
fmt.Println(v)
case <-time.After(5 * time.Second): //返回值为channel类型,输出值为Time当前时间
fmt.Println("timeout")
o <- true
break
}
}
}()
//c <-666 //注释掉,引发timeout
<-o
}
©著作权归作者所有：来自51CTO博客作者ck_god的原创作品，如需转载，请注明出处，否则将追究法律责任
随时随地看视频
