0
点赞
收藏
分享

微信扫一扫

使用 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 个左右的数字。

举报

相关推荐

Kotlin协程的使用

Android协程的使用

0 条评论