select的作用
Go 里面提供了一个关键字 select,通过 select 可以监听 channel上的数据流动。
select 的用法与switch 语言非常类似,由 select 开始一个新的选择块,每个选择条件由 case语句来描述。
与 switch 语句可以选择任何可使用相等比较的条件相比, select 有比较多的限制,其中最大的一条限制就是每个 case 语句里必须是一个 IO 操作,大致的结构如下:
在一个 select 语句中,Go语言会按顺序从头至尾评估每一个发送和接收的语句。
如果其中的任意一语句可以继续执行(即没有被阻塞),那么就从那些可以执行的语句中任意选择一条来使用。
如果没有任意一条语句可以执行(即所有的通道都被阻塞),那么有两种可能的情况:如果给出了 * default 语句,那么就会执行 default 语句,同时程序的执行会从 select 语句后的语句中恢复。
- 如果没有 default 语句,那么 select 语句将被阻塞,直到至少有一个通信可以进行下去。
利用select实现fibonacci
package mainimport ("fmt"
)// ch只写,quit只读
func fibonacci(ch chan<- int, quit <-chan bool) {x, y := 1, 1for {select {case ch <- x:x, y = y, x+ycase flag := <-quit:fmt.Println("flag=", flag)return}}
}
func main() {ch := make(chan int) //数字通信quit := make(chan bool) //程序是否结束//消费者go func() {for i := 0; i < 8; i++ {num := <-chfmt.Println(num)}//可以停止quit <- true}()//生产者fibonacci(ch, quit)
}
- ch控制着读写的进行。
select实现超时
有时候会出现 goroutine 阻塞的情况,那么我们如何避免整个程序进入阻塞的情况呢?我们可以利用 select 来设置超时,通过如下的方式实现: