详解golang开发中select多路选择
select是Golang中的一个控制结构,语法上类似于switch语句,只不过select是用于goroutine间通信的,每个case必须是一个通信操作,要么是发送要么是接收,select会随机执行一个可运行的case。如果没有case可运行,goroutine将阻塞,直到有case可运行。
select多路选择
select写法上跟switchcase的写法基本一致,只不过golang的select是通信控制语句。select的执行必须有通信的发送或者接受,如果没有就一直阻塞。
ch:=make(chanbool,0) ch1:=make(chanbool,0) select{ caseret:=<-ch: fmt.Println(ret) caseret:=<-ch1: fmt.Println(ret) }
如果ch和ch1都没有通信数据发送,select就一直阻塞,直到ch或者ch1有数据发送,select就执行相应的case来接受数据。
select实现超时控制
我们可以利用select机制实现一种简单的超时控制。
先看下程序完整执行的代码
funcservice(chchanbool){ time.Sleep(time.Second*3) ch<-true } funcmain(){ ch:=make(chanbool,0) goservice(ch) select{ caseret:=<-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,再看下执行效果
funcservice(chchanbool){ time.Sleep(time.Second*6) ch<-true } funcmain(){ ch:=make(chanbool,0) goservice(ch) select{ caseret:=<-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 oktrue正常接收数据 okfalse通道关闭
可以看到接受数据其实有两个参数,第二个bool值会反应channel是否关闭,是否可以正常接受数据。
看下测试代码
我们写了一个数据发送者,两个数据接收者,当发送者关闭channel的时候,两个接收者的goroutine可以通过以上的语法判断channel是否关闭,决定自己的goroutine是否结束。
funcsender(chchanint,wg*sync.WaitGroup){ fori:=0;i<10;i++{ ch<-i } close(ch) wg.Done() } funcreceiver(chchanint,wg*sync.WaitGroup){ for{ ifval,ok:=<-ch;ok{ fmt.Println(fmt.Sprintf("%d,%s",val,"revevier")) }else{ fmt.Println("quitrecevier") break; } } wg.Done() } funcreceiver2(chchanint,wg*sync.WaitGroup){ for{ ifval,ok:=<-ch;ok{ fmt.Println(fmt.Sprintf("%d,%s",val,"revevier2")) }else{ fmt.Println("quitrecevier2") break; } } wg.Done() } funcmain(){ ch:=make(chanint,0) wg:=&sync.WaitGroup{} wg.Add(1) gosender(ch,wg) wg.Add(1) goreceiver(ch,wg) wg.Add(1) goreceiver2(ch,wg) wg.Wait() }
执行结果
0,revevier2
2,revevier2
3,revevier2
4,revevier2
5,revevier2
6,revevier2
7,revevier2
1,revevier
9,revevier
quitrecevier
8,revevier2
quitrecevier2
可以看到一个数据发送者,两个数据接收者,当channel关闭的时候,两个数据接收者都收到了channel关闭的通知。
需要注意的是,给一个已经关闭的channel发送数据,程序会panic,从一个已经关闭的channel接收数据,会接收到没有参考意义的channel类型的0值数据,Int是0,string是空...
select退出计时器等程序
开发中经常会经常会使用轮训计时器,但是当程序退出时,轮训计时器无法关闭的问题。其实select是可以解决这个问题的。
如果我们有一个轮训任务,需要一个timer,每隔3S执行逻辑,过完10S之后关闭这个timer。
看下代码
funcTimeTick(wg*sync.WaitGroup,qchanbool){ deferwg.Done() t:=time.NewTicker(time.Second*3) defert.Stop() for{ select{ case<-q: fmt.Println("quit") return case<-t.C: fmt.Println("secondstimer") } } } funcmain(){ q:=make(chanbool) wg:=new(sync.WaitGroup) wg.Add(1) goTimeTick(wg,q) time.Sleep(time.Second*10) close(q) wg.Wait() }
执行结果
secondstimer
secondstimer
secondstimer
quit
很优雅的通过关闭channel退出了轮训计时器goroutine,
到此这篇关于golang开发中select多路选择的文章就介绍到这了,更多相关golangselect多路选择内容请搜索毛票票以前的文章或继续浏览下面的相关文章希望大家以后多多支持毛票票!