[golang]语法基础之select

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语言会按顺序从头至尾评估每一个发送和接收的语句。

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

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

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

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

例如下面的代码中通过select和channel实现斐波那契数列的计算：

package main

import "fmt"

func main() {
	// 计算斐波那契数列
	ch := make(chan int)
	quit := make(chan bool)

	go func() {
		for i := 0; i < 6; i++ {
			fmt.Println(<-ch)
		}
		quit <- true
	}()
	fibonacci(ch,quit)
}

func fibonacci(ch chan int, quit chan bool) {
	x, y := 1, 1
	for {
		// 通过select监听channel值的传递
		select {
		case ch <- x: // 将x写入到ch
			x, y = y, x+y
		case <-quit:
			fmt.Println("quit")
			return
		}
	}
}

输出结果:

1
1
2
3
5
8
quit

超时

有时候会出现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
}