Go 语言内置 I/O 多路复用机制

简介: Go 语言内置 I/O 多路复用机制

介绍

Go 协程之间通过 channel 通信,但是 channel 读写取决于自身特性,即是否有可写入缓冲区、缓冲区中是否有数据、是否已关闭...

为了检测 channel 的特性,Go 提供了一个关键字 select,可用于实现 I/O 多路复用机制。

本文我们介绍 Go 关键字 select 的使用方式。

使用方式

Go 关键字 select 中包含 case 语句和 default 语句,其中 default 语句可以认为是一种特殊的 case 语句。

因为 default 语句不负责处理 channel 的读写,它可以在 select 中的任意位置,且仅能包含一个 default 语句。在所有 case 语句都不满足执行条件时,default 语句将被执行(建议尽量不要省略 default 语句)。

我们通过代码片段,分别介绍 select 在检测到 channel 不同特性时,得到的运行结果。

接下来,我们阅读一段代码。

func main() {
 fmt.Println("Golang 语言开发栈")
 go func() {
  fmt.Println("Golang 公众号")
 }()
}

阅读上面这段代码,读者朋友们认为 Go 协程中的打印语句可以正常输出吗?

读者朋友们如果运行代码,会发现 Go 协程中的打印语句还没有执行,程序就已经退出了,这是因为 main 函数中的打印语句已经执行完成,所以会退出程序。

如果我们希望 Go 协程中的打印语句也执行,可以在 main 函数中使用 select{}main 阻塞,Go 协程中的打印语句就有机会执行了。但是,这会导致死锁(可以根据实际应用场景选择是否使用)。

接下来,我们再读一段可以导致死锁的代码:

func main() {
 c := make(chan string)
 DoChannel(c)
}
func DoChannel(c chan string) {
 var receive string
 send := "golang"
 select {
 case receive = <-c:
  fmt.Println(receive)
 case c <- send:
  fmt.Println(send)
 }
}

阅读上面这段代码,我们定义一个函数 DoChannel(),该函数接收的参数是一个 string 类型的 channel,函数体中使用 select 中的两个 case 语句,分别对参数进行接收和发送操作。

因为,我们传参的 c 是无缓冲 channel,所以它即不能读也不能写,两个 case 语句都不执行,select 陷入阻塞,导致死锁(此处为了行文,故意没有 default 语句)。

我们将上面这段代码,稍微修改一下,将入参的 c 改为 1 个缓冲区大小的 channel(未写入数据)。代码如下:

func main() {
 c := make(chan string, 1)
 DoChannel(c)
}

运行代码,写执行,读未执行。

select 中的对入参 channel 进行发送操作的 case 语句被执行,因为入参 c 是一个有 1 个缓冲区大小的 channel,并且该 channel 中还没有数据,所以读取操作的 case 语句没有读取到数据,不满足执行条件。

有缓冲区,已写满数据 channel

我们再修改一下入参 c,将入参的 c 改为 1 个缓冲区大小的 channel,并且写入字符串 Go。代码如下:

func main() {
 c := make(chan string, 1)
 c <- "Go"
 DoChannel(c)
}

运行代码,读执行,写未执行。

select 中的对入参 channel 进行接收操作的 case 语句被执行,因为入参 c 是一个有 1 个缓冲区大小,并且已写满数据,所以读取操作的 case 语句可以读取到数据,满足执行条件。

而写入操作的 case 无法写入数据,不满足执行条件。

有缓冲区,有数据,可写数据 channel

最后一种场景是既能读取也能写入的 channel,我们修改一下入参 c,将入参 c 改为 2 个缓冲区大小的 channel,其中 1 个缓冲区写入字符串 Go,另外 1 个缓冲区还可以写入数据。代码如下:

func main() {
 c := make(chan string, 2)
 c <- "Go"
 DoChannel(c)
}

通过多次运行代码,会发现读取和写入的 case 语句都有机会执行,因为两个 case 语句都满足执行条件,但是只能有 1 个 case 语句执行,select 会随机执行其中 1 个 case 语句。

至此,我们已经介绍了 5 种 channelselect 中的运行结果。

case 语句中声明变量

上面的代码中,我们发现在两个 case 语句中,读操作我们将读取到的数据赋值给变量 receive,实际上,我们也可以省略变量赋值操作。

如果我们需要将读取到的数据,赋值给变量的话,一般建议将读取 channel 返回的两个值全部接收,其中一个是读取到的数据,另外一个是布尔值,代表 channel 中没有数据,并且已被关闭。代码如下:

func main() {
 c := make(chan string)
 close(c)
 DoChannelV2(c)
}
func DoChannelV2(c chan string) {
 var (
  receive string
  ok      bool
 )
 select {
 case receive, ok = <-c:
  if !ok {
   fmt.Println("no data")
  } else {
   fmt.Println(receive)
  }
 }
}

阅读上面这段代码,我们使用 closec 关闭。select 中的读操作 case 语句,可以通过 ok 的值,得到 channel 中没有数据,且已被关闭,不必打印空数据。

03

总结

本文我们了解到 select 中的 case 语句可以读取 channel,多个 case 语句仅能其中 1 个被执行。

每个 case 语句仅能对 1 个 channel 进行读写操作,如果读操作未读取到数据将陷入阻塞,如果写操作无法写入数据将陷入阻塞,如果所有 case 语句中的 channel 都陷入阻塞时,select 也会陷入阻塞。

为了避免 select 陷入阻塞,我们可以使用 default 语句,需要注意的是,default 语句可以在 select 的任意位置,但是仅能包含 1 个,而 case 语句可以包含多个。


目录
相关文章
|
9天前
|
存储 监控 算法
员工上网行为监控中的Go语言算法:布隆过滤器的应用
在信息化高速发展的时代,企业上网行为监管至关重要。布隆过滤器作为一种高效、节省空间的概率性数据结构,适用于大规模URL查询与匹配,是实现精准上网行为管理的理想选择。本文探讨了布隆过滤器的原理及其优缺点,并展示了如何使用Go语言实现该算法,以提升企业网络管理效率和安全性。尽管存在误报等局限性,但合理配置下,布隆过滤器为企业提供了经济有效的解决方案。
45 8
员工上网行为监控中的Go语言算法:布隆过滤器的应用
|
29天前
|
存储 Go 索引
go语言中数组和切片
go语言中数组和切片
39 7
|
28天前
|
Go 开发工具
百炼-千问模型通过openai接口构建assistant 等 go语言
由于阿里百炼平台通义千问大模型没有完善的go语言兼容openapi示例,并且官方答复assistant是不兼容openapi sdk的。 实际使用中发现是能够支持的,所以自己写了一个demo test示例,给大家做一个参考。
|
29天前
|
程序员 Go
go语言中结构体(Struct)
go语言中结构体(Struct)
101 71
|
28天前
|
存储 Go 索引
go语言中的数组(Array)
go语言中的数组(Array)
105 67
|
3天前
|
算法 安全 Go
Go 语言中实现 RSA 加解密、签名验证算法
随着互联网的发展,安全需求日益增长。非对称加密算法RSA成为密码学中的重要代表。本文介绍如何使用Go语言和[forgoer/openssl](https://github.com/forgoer/openssl)库简化RSA加解密操作,包括秘钥生成、加解密及签名验证。该库还支持AES、DES等常用算法,安装简便,代码示例清晰易懂。
29 16
|
29天前
|
存储 Go
go语言中映射
go语言中映射
36 11
|
30天前
|
Go 索引
go语言修改元素
go语言修改元素
31 6
|
6天前
|
监控 算法 安全
解锁企业计算机监控的关键:基于 Go 语言的精准洞察算法
企业计算机监控在数字化浪潮下至关重要,旨在保障信息资产安全与高效运营。利用Go语言的并发编程和系统交互能力,通过进程监控、网络行为分析及应用程序使用记录等手段,实时掌握计算机运行状态。具体实现包括获取进程信息、解析网络数据包、记录应用使用时长等,确保企业信息安全合规,提升工作效率。本文转载自:[VIPShare](https://www.vipshare.com)。
19 0
|
20天前
|
Go 数据安全/隐私保护 UED
优化Go语言中的网络连接:设置代理超时参数
优化Go语言中的网络连接:设置代理超时参数