详解golang开发中select多路选择
select 是 Golang 中的一个控制结构，语法上类似于switch 语句，只不过select是用于 goroutine 间通信的 ，每个 case 必须是一个通信操作，要么是发送要么是接收，select 会随机执行一个可运行的 case。如果没有 case 可运行，goroutine 将阻塞，直到有 case 可运行。
select 多路选择
select写法上跟switch case的写法基本一致，只不过golang的select是通信控制语句。select的执行必须有通信的发送或者接受，如果没有就一直阻塞。
ch := make(chan bool, 0)
ch1 := make(chan bool, 0)
select {
case ret := <-ch:
fmt.Println(ret)
case ret := <-ch1:
fmt.Println(ret)
}
如果ch和ch1都没有通信数据发送，select就一直阻塞，直到ch或者ch1有数据发送，select就执行相应的case来接受数据。
select 实现超时控制
我们可以利用select机制实现一种简单的超时控制。
先看下程序完整执行的代码
func service(ch chan bool) {
time.Sleep(time.Second*3)
ch<-true
}
func main() {
ch := make(chan bool, 0)
go service(ch)
select {
case ret := <-ch:
fmt.Println(ret)
case <-time.After(time.Second*5):
fmt.Println("timeout")
}
}
___go_build_main_go #gosetup
true
可以看到使用time.After超时定义了5S，service程序执行3S，所以肯定没有超时，跟预想的一致。
我们再看看超时的执行，我们将service程序执行时间该为6S。超时控制继续是5S，再看下执行效果
func service(ch chan bool) {
time.Sleep(time.Second*6)
ch<-true
}
func main() {
ch := make(chan bool, 0)
go service(ch)
select {
case ret := <-ch:
fmt.Println(ret)
case <-time.After(time.Second*5):
fmt.Println("timeout")
}
}
___go_build_main_go #gosetup
timeout
执行到了超时的case，跟预想的其实是一致的。
select 判断channel是否关闭
先看下接受数据的语法
val,ok <- ch ok true 正常接收数据 ok false 通道关闭
可以看到接受数据其实有两个参数，第二个bool值会反应channel是否关闭，是否可以正常接受数据。
看下测试代码
我们写了一个数据发送者，两个数据接收者，当发送者关闭channel的时候，两个接收者的 goroutine 可以通过以上的语法判断channel是否关闭，决定自己的 goroutine 是否结束。
func sender(ch chan int, wg *sync.WaitGroup) {
for i:=0;i<10;i++ {
ch<-i
}
close(ch)
wg.Done()
}
func receiver(ch chan int, wg *sync.WaitGroup) {
for {
if val,ok := <-ch;ok {
fmt.Println(fmt.Sprintf("%d,%s",val, "revevier"))
} else {
fmt.Println("quit recevier")
break;
}
}
wg.Done()
}
func receiver2(ch chan int, wg *sync.WaitGroup) {
for {
if val,ok := <-ch;ok {
fmt.Println(fmt.Sprintf("%d,%s",val, "revevier2"))
} else {
fmt.Println("quit recevier2")
break;
}
}
wg.Done()
}
func main() {
ch := make(chan int, 0)
wg := &sync.WaitGroup{}
wg.Add(1)
go sender(ch, wg)
wg.Add(1)
go receiver(ch, wg)
wg.Add(1)
go receiver2(ch, wg)
wg.Wait()
}
执行结果
0,revevier2
2,revevier2
3,revevier2
4,revevier2
5,revevier2
6,revevier2
7,revevier2
1,revevier
9,revevier
quit recevier
8,revevier2
quit recevier2
可以看到一个数据发送者，两个数据接收者，当channel关闭的时候，两个数据接收者都收到了channel关闭的通知。
需要注意的是，给一个已经关闭的channel发送数据，程序会panic，从一个已经关闭的channel接收数据，会接收到没有参考意义的channel类型的0值数据，Int是0，string是空...
select 退出计时器等程序
开发中经常会经常会使用轮训计时器，但是当程序退出时，轮训计时器无法关闭的问题。其实select是可以解决这个问题的。
如果我们有一个轮训任务，需要一个timer，每隔3S执行逻辑，过完10S之后关闭这个timer。
看下代码
func TimeTick(wg *sync.WaitGroup,q chan bool) {
defer wg.Done()
t := time.NewTicker(time.Second*3)
defer t.Stop()
for {
select {
case <-q:
fmt.Println("quit")
return
case <-t.C:
fmt.Println("seconds timer")
}
}
}
func main() {
q := make(chan bool)
wg := new(sync.WaitGroup)
wg.Add(1)
go TimeTick(wg,q)
time.Sleep(time.Second*10)
close(q)
wg.Wait()
}
执行结果
seconds timer
seconds timer
seconds timer
quit
很优雅的通过关闭channel退出了轮训计时器 goroutine，
到此这篇关于golang开发中select多路选择的文章就介绍到这了,更多相关golang select多路选择内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要给大家介绍了关于Golang中数据结构Queue的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。2017-09-09
这篇文章主要介绍了golang调用c实现的dll接口细节分享，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-05-05
最近会使用到sqlite3,这里作个记录,记性越来越差就是这样,下面这篇文章主要给大家介绍了关于Golang操作sqlite3数据库的详细教程,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2023-04-04
除标准库中提供的同步原语外，Go语言还在子仓库sync中提供了4种扩展原语，本文主要为大家介绍的是其中的golang/sync/errgroup.Group，感兴趣的小伙伴可以了解一下2023-07-07
这篇文章主要为大家介绍了goroutine 泄漏和避免泄漏实战示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-12-12
这篇文章主要介绍了Golang之如何读取文件内容问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-06-06
这篇文章主要为大家介绍了go语法入门any类型的使用场景示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-09-09
Go语言的XML包提供了强大的数据解析功能,让你轻松处理各种XML格式的数据,这个指南将带你深入了解如何使用Go语言的XML包,快速上手XML数据解析,准备好开启XML解析之旅了吗?Let's Go!2024-03-03
在这篇文章中,我们将详细描述如何在 Go 语言中使用 SQLite 数据库,我们将会从如何在 Go 中安装和使用 SQLite 驱动包开始讲起,然后逐步介绍如何创建数据库连接,执行 SQL 查询,处理返回的数据以及关闭数据库连接2024-01-01
这篇文章主要为大家介绍了Go 语言数据结构之双链表学习教程详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-08-08
最新评论