select作用

Go里面提供了一个关键字select,通过select可以监听channel上的数据流动。

有时候我们希望能够借助channel发送或接收数据,并避免因为发送或者接收导致的阻塞,尤其是当channel没有准备好写或者读时。select语句就可以实现这样的功能。

select的用法与switch语言非常类似,由select开始一个新的选择块,每个选择条件由case语句来描述。

与switch语句相比,select有比较多的限制,其中最大的一条限制就是每个case语句里必须是一个IO操作,大致的结构如下:

  1. select {
  2. case <- chan1:
  3. // 如果chan1成功读到数据,则进行该case处理语句
  4. case chan2 <- 1:
  5. // 如果成功向chan2写入数据,则进行该case处理语句
  6. default:
  7. // 如果上面都没有成功,则进入default处理流程
  8. }

在一个select语句中,Go语言会按顺序从头至尾评估每一个发送和接收的语句。

如果其中的任意一语句可以继续执行(即没有被阻塞),那么就从那些可以执行的语句中任意选择一条来使用。

如果没有任意一条语句可以执行(即所有的通道都被阻塞),那么有两种可能的情况:

l 如果给出了default语句,那么就会执行default语句,同时程序的执行会从select语句后的语句中恢复。

l 如果没有default语句,那么select语句将被阻塞,直到至少有一个通信可以进行下去。

示例代码:

  1. package main
  2. import (
  3. "fmt"
  4. )
  5. func fibonacci(c, quit chan int) {
  6. x, y := 1, 1
  7. for {
  8. select {
  9. case c <- x:
  10. x, y = y, x+y
  11. case <-quit:
  12. fmt.Println("quit")
  13. return
  14. }
  15. }
  16. }
  17. func main() {
  18. c := make(chan int)
  19. quit := make(chan int)
  20. go func() {
  21. for i := 0; i < 6; i++ {
  22. fmt.Println(<-c)
  23. }
  24. quit <- 0
  25. }()
  26. fibonacci(c, quit)
  27. }

运行结果如下:

select