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;i10;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多路選擇內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持腳本之家!
您可能感興趣的文章:- matplotlib之多邊形選區(PolygonSelector)的使用
- Golang的select多路復用及channel使用操作
- golang中的select關鍵字用法總結
- Go語言中Select語句用法實例
- Go語言使用select{}阻塞main函數介紹