在 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("Received on channel 1: %d\n", v) case v := <-ch2: fmt.Printf("Received on channel 2: %d\n", 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 个左右的数字。