> 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機制實現一種簡單的超時控制。 先看下程式完整執行的程式碼 ```bash 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，再看下執行效果 ```bash 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 是否結束。 ```bash 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() } ``` 執行結果 ```bash 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。 看下程式碼 ```bash 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退出了輪訓計時器 goro