快捷導(dǎo)航

詳解golang開發(fā)中select多路選擇

更新時(shí)間：2020年09月27日 09:02:48 作者：二十四橋明月夜33

這篇文章主要介紹了golang開發(fā)中select多路選擇,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

select 是 Golang 中的一個(gè)控制結(jié)構(gòu)，語法上類似于switch 語句，只不過select是用于 goroutine 間通信的，每個(gè) case 必須是一個(gè)通信操作，要么是發(fā)送要么是接收，select 會(huì)隨機(jī)執(zhí)行一個(gè)可運(yùn)行的 case。如果沒有 case 可運(yùn)行，goroutine 將阻塞，直到有 case 可運(yùn)行。

select 多路選擇

select寫法上跟switch case的寫法基本一致，只不過golang的select是通信控制語句。select的執(zhí)行必須有通信的發(fā)送或者接受，如果沒有就一直阻塞。

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都沒有通信數(shù)據(jù)發(fā)送，select就一直阻塞，直到ch或者ch1有數(shù)據(jù)發(fā)送，select就執(zhí)行相應(yīng)的case來接受數(shù)據(jù)。

select 實(shí)現(xiàn)超時(shí)控制

我們可以利用select機(jī)制實(shí)現(xiàn)一種簡(jiǎn)單的超時(shí)控制。
先看下程序完整執(zhí)行的代碼

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超時(shí)定義了5S，service程序執(zhí)行3S，所以肯定沒有超時(shí)，跟預(yù)想的一致。
我們?cè)倏纯闯瑫r(shí)的執(zhí)行，我們將service程序執(zhí)行時(shí)間該為6S。超時(shí)控制繼續(xù)是5S，再看下執(zhí)行效果

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

執(zhí)行到了超時(shí)的case，跟預(yù)想的其實(shí)是一致的。

select 判斷channel是否關(guān)閉

先看下接受數(shù)據(jù)的語法

val,ok <- ch
ok true 正常接收數(shù)據(jù)
ok false 通道關(guān)閉

可以看到接受數(shù)據(jù)其實(shí)有兩個(gè)參數(shù)，第二個(gè)bool值會(huì)反應(yīng)channel是否關(guān)閉，是否可以正常接受數(shù)據(jù)。

看下測(cè)試代碼
我們寫了一個(gè)數(shù)據(jù)發(fā)送者，兩個(gè)數(shù)據(jù)接收者，當(dāng)發(fā)送者關(guān)閉channel的時(shí)候，兩個(gè)接收者的 goroutine 可以通過以上的語法判斷channel是否關(guān)閉，決定自己的 goroutine 是否結(jié)束。

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()
}

執(zhí)行結(jié)果

0,revevier2
2,revevier2
3,revevier2
4,revevier2
5,revevier2
6,revevier2
7,revevier2
1,revevier
9,revevier
quit recevier
8,revevier2
quit recevier2

可以看到一個(gè)數(shù)據(jù)發(fā)送者，兩個(gè)數(shù)據(jù)接收者，當(dāng)channel關(guān)閉的時(shí)候，兩個(gè)數(shù)據(jù)接收者都收到了channel關(guān)閉的通知。
需要注意的是，給一個(gè)已經(jīng)關(guān)閉的channel發(fā)送數(shù)據(jù)，程序會(huì)panic，從一個(gè)已經(jīng)關(guān)閉的channel接收數(shù)據(jù)，會(huì)接收到?jīng)]有參考意義的channel類型的0值數(shù)據(jù)，Int是0，string是空...

select 退出計(jì)時(shí)器等程序

開發(fā)中經(jīng)常會(huì)經(jīng)常會(huì)使用輪訓(xùn)計(jì)時(shí)器，但是當(dāng)程序退出時(shí)，輪訓(xùn)計(jì)時(shí)器無法關(guān)閉的問題。其實(shí)select是可以解決這個(gè)問題的。
如果我們有一個(gè)輪訓(xùn)任務(wù)，需要一個(gè)timer，每隔3S執(zhí)行邏輯，過完10S之后關(guān)閉這個(gè)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()
}

執(zhí)行結(jié)果