select 是 Golang 中的一个控制结构，语法上类似于 switch 语句，只不过 select 是用于 goroutine 间通信的 ，每个 case 必须是一个通信操作，要么是发送要么是接收，select 会随机执行一个可运行的 case。如果没有 case 可运行，goroutine 将阻塞，直到有 case 可运行。
select 多路选择
select 写法上跟 switch case 的写法基本一致，只不过 golang 的 select 是通信控制语句。select 的执行必须有通信的发送或者接受，如果没有就一直阻塞。
ch := make(chan bool, 0) ch1 := make(chan bool, 0) select { case ret := <-ch: fmt.Println(ret) case ret := <-ch1: fmt.Println(ret) }
复制代码
如果 ch 和 ch1 都没有通信数据发送，select 就一直阻塞，直到 ch 或者 ch1 有数据发送，select 就执行相应的 case 来接受数据。
select 实现超时控制
我们可以利用 select 机制实现一种简单的超时控制。先看下程序完整执行的代码
func service(ch chan bool) { time.Sleep(time.Second*3) ch<-true}func main() { ch := make(chan bool, 0) go service(ch) select { case ret := <-ch: fmt.Println(ret) case <-time.After(time.Second*5): fmt.Println("timeout") }}
___go_build_main_go #gosetuptrue
复制代码
可以看到使用 time.After 超时定义了 5S，service 程序执行 3S，所以肯定没有超时，跟预想的一致。我们再看看超时的执行，我们将 service 程序执行时间该为 6S。超时控制继续是 5S，再看下执行效果
func service(ch chan bool) { time.Sleep(time.Second*6) ch<-true}func main() { ch := make(chan bool, 0) go service(ch) select { case ret := <-ch: fmt.Println(ret) case <-time.After(time.Second*5): fmt.Println("timeout") }}
___go_build_main_go #gosetuptimeout
复制代码
执行到了超时的 case，跟预想的其实是一致的。
select 判断 channel 是否关闭
先看下接受数据的语法
val,ok <- chok true 正常接收数据ok false 通道关闭
复制代码
可以看到接受数据其实有两个参数，第二个 bool 值会反应 channel 是否关闭，是否可以正常接受数据。
看下测试代码我们写了一个数据发送者，两个数据接收者，当发送者关闭 channel 的时候，两个接收者的 goroutine 可以通过以上的语法判断 channel 是否关闭，决定自己的 goroutine 是否结束。
func sender(ch chan int, wg *sync.WaitGroup) { for i:=0;i<10;i++ { ch<-i } close(ch) wg.Done()}func receiver(ch chan int, wg *sync.WaitGroup) { for { if val,ok := <-ch;ok { fmt.Println(fmt.Sprintf("%d,%s",val, "revevier")) } else { fmt.Println("quit recevier") break; } } wg.Done()}func receiver2(ch chan int, wg *sync.WaitGroup) { for { if val,ok := <-ch;ok { fmt.Println(fmt.Sprintf("%d,%s",val, "revevier2")) } else { fmt.Println("quit recevier2") break; } } wg.Done()}func main() { ch := make(chan int, 0) wg := &sync.WaitGroup{} wg.Add(1) go sender(ch, wg) wg.Add(1) go receiver(ch, wg) wg.Add(1) go receiver2(ch, wg) wg.Wait()}
复制代码
执行结果
0,revevier22,revevier23,revevier24,revevier25,revevier26,revevier27,revevier21,revevier9,revevierquit recevier8,revevier2quit recevier2
复制代码
可以看到一个数据发送者，两个数据接收者，当 channel 关闭的时候，两个数据接收者都收到了 channel 关闭的通知。需要注意的是，给一个已经关闭的 channel 发送数据，程序会 panic，从一个已经关闭的 channel 接收数据，会接收到没有参考意义的 channel 类型的 0 值数据，Int 是 0，string 是空...
select 退出计时器等程序
开发中经常会经常会使用轮训计时器，但是当程序退出时，轮训计时器无法关闭的问题。其实 select 是可以解决这个问题的。如果我们有一个轮训任务，需要一个 timer，每隔 3S 执行逻辑，过完 10S 之后关闭这个 timer。
看下代码
func TimeTick(wg *sync.WaitGroup,q chan bool) { defer wg.Done() t := time.NewTicker(time.Second*3) defer t.Stop() for { select { case <-q: fmt.Println("quit") return case <-t.C: fmt.Println("seconds timer") } }}func main() { q := make(chan bool) wg := new(sync.WaitGroup) wg.Add(1) go TimeTick(wg,q) time.Sleep(time.Second*10) close(q) wg.Wait()}
复制代码
执行结果
seconds timerseconds timerseconds timerquit
复制代码
通过关闭 channel 退出了轮训计时器 goroutine，
评论