中午好欧美黄色片,久久亚洲精品无码系列客服,天天操人人添天天爱

主頁 > 知識庫 > 詳解golang開發中select多路選擇

詳解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;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()
}

執行結果