使用 select 切换协程

在 Go 中，可以通过关键字

select

来完成从不同的并发执行的协程中获取值，它和

switch

控制语句非常相似，也被称作通信开关；它的行为像是“你准备好了吗”的轮询机制；

select

监听进入通道的数据，也可以是用通道发送值的时候。

语法格式：

select {
case u:= <- ch1:
 ...
case v:= <- ch2:
 ...
 ...
default: // no value ready to be received
 ...
}


default

 语句是可选的；fallthrough 行为，和普通的 switch 相似，是不允许的。在任何一个 case 中执行 

break

 或者 

return

，select 就结束了。

select

 做的就是：选择处理列出的多个通信情况中的一个。

• 如果都阻塞了，会等待直到其中一个可以处理

• 如果多个可以处理，随机选择一个

• 如果没有通道操作可以处理并且写了
  default
   语句，它就会执行：
  default
   永远是可运行的（这就是准备好了，可以执行）。

在 

select

 中使用发送操作并且有 

default

 可以确保发送不被阻塞！如果没有 

default

，select 就会一直阻塞。

select

 语句实现了一种监听模式，通常用在（无限）循环中；在某种情况下，通过 

break

 语句使循环退出。

程序示例

package main

import (
 "fmt"
 "time"
)

func main() {
 ch1 := make(chan int)
 ch2 := make(chan int)

 go pump1(ch1)
 go pump2(ch2)
 go suck(ch1, ch2)

 time.Sleep(1e9)
}

func pump1(ch chan int) {
 for i := 0; ; i++ {
 ch <- i * 2
 }
}

func pump2(ch chan int) {
 for i := 0; ; i++ {
 ch <- i + 5
 }
}

func suck(ch1, ch2 chan int) {
 for {
 select {
 case v := <-ch1:
 fmt.Printf(&quot;Received on channel 1: %d\n&quot;, v)
 case v := <-ch2:
 fmt.Printf(&quot;Received on channel 2: %d\n&quot;, v)
 }
 }
}


在程序 

goroutine_select.go

 中有 2 个通道 

ch1

 和 

ch2

，三个协程 

pump1()

、

pump2()

 和 

suck()

。这是一个典型的生产者消费者模式。在无限循环中，

ch1

 和 

ch2

 通过 

pump1()

 和 

pump2()

 填充整数；

suck()

 也是在无限循环中轮询输入的，通过 

select

 语句获取 

ch1

 和 

ch2

 的整数并输出。选择哪一个 case 取决于哪一个通道收到了信息。程序在 main 执行 1 秒后结束。

运行结果：

Received on channel 2: 148120
Received on channel 2: 148121
Received on channel 2: 148122
Received on channel 2: 148123
Received on channel 2: 148124
Received on channel 2: 148125
Received on channel 2: 148126
Received on channel 1: 296784
Received on channel 2: 148127
Received on channel 2: 148128
Received on channel 2: 148129
Received on channel 1: 296786
Received on channel 1: 296788


一秒内的输出非常惊人，如果我们给它计数（goroutine_select2.go），得到了 296788 个左右的数字。