使用 select 切换协程

于今腐草无萤火,终古垂杨有暮鸦。这篇文章主要讲述使用 select 切换协程相关的知识,希望能为你提供帮助。
?
【使用 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("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 个左右的数字。

    推荐阅读