golang select case 用法

 

package main
import {

        "fmt"

        "time"
)
 
func main() {
        ch1 := make(chan int)
        ch2 := make(chan int)
 
        go func(ch chan int) { <-ch }(ch1)
        go func(ch chan int) { ch <- 2 }(ch2)
 
        time.Sleep(time.Millisecond)
 
        for {
                select {
                case ch1 <- 1:
                        fmt.Println("Send operation on ch1 works!")
                        //continue
                case <-ch2:
                        fmt.Println("Receive operation on ch2 works!")
                        continue
                default:
                        fmt.Println("Exit now!")
                        return
                }
              fmt.Println("bbbbbbb")
        }
        //i := 1;
        //for {
        //    i++
        //    fmt.Println(i)
        //    if i > 3 {
        //      break
        //    }

        //}
}

　　

一、switch語句

switch語句提供了一個多分支條件執行的方法。每一個case可以攜帶一個表達式或一個類型說明符。前者又可被簡稱為case表達式。因此，Go語言的switch語句又分為表達式switch語句和類型switch語句。

Go會依照從上至下的順序對每一條case語句中case表達式進行求值，只要被發現其表達式與switch表達式的結果相同，該case語句就會被選中。其余的case語句會被忽略。   與if相同，switch語句還可以包含初始化字句，且其出現位置和寫法如出一轍：

var name string ... switch name { case "Golang": fmt.Println("Golang") case "Rust": fmt.Println("Rust") default: fmt.Println("PHP是世界上最好的語言") } 

 

二、select語句

注意到 select 的代碼形式和 switch 非常相似， 不過 select 的 case 里的操作語句只能是【IO 操作】 。

 

break語句也可以被包含在select語句中的case語句中。它的作用是立即結束當前的select語句的執行。不論其所屬的case語句中是否還有未被執行的語句。

---------------------

 

golang 的 select 的功能和 select, poll, epoll 相似， 就是監聽 IO 操作，當 IO 操作發生時，觸發相應的動作。

示例：

ch1 := make (chan int, 1) ch2 := make (chan int, 1) ... select { case <-ch1: fmt.Println("ch1 pop one element") case <-ch2: fmt.Println("ch2 pop one element") } 

注意到 select 的代碼形式和 switch 非常相似， 不過 select 的 case 里的操作語句只能是【IO 操作】 。

此示例里面 select 會一直等待等到某個 case 語句完成， 也就是等到成功從 ch1 或者 ch2 中讀到數據。 則 select 語句結束。

【使用 select 實現 timeout 機制】

如下：

timeout := make (chan bool, 1) go func() { time.Sleep(1e9) // sleep one second timeout <- true }() ch := make (chan int) select { case <- ch: case <- timeout: fmt.Println("timeout!") } 

當超時時間到的時候，case2 會操作成功。 所以 select 語句則會退出。 而不是一直阻塞在 ch 的讀取操作上。 從而實現了對 ch 讀取操作的超時設置。

下面這個更有意思一點。

當 select 語句帶有 default 的時候：

ch1 := make (chan int, 1) ch2 := make (chan int, 1) select { case <-ch1: fmt.Println("ch1 pop one element") case <-ch2: fmt.Println("ch2 pop one element") default: fmt.Println("default") } 

此時因為 ch1 和 ch2 都為空，所以 case1 和 case2 都不會讀取成功。 則 select 執行 default 語句。

就是因為這個 default 特性， 我們可以使用 select 語句來檢測 chan 是否已經滿了。

如下：

ch := make (chan int, 1) ch <- 1 select { case ch <- 2: default: fmt.Println("channel is full !") } 

因為 ch 插入 1 的時候已經滿了， 當 ch 要插入 2 的時候，發現 ch 已經滿了（case1 阻塞住）， 則 select 執行 default 語句。 這樣就可以實現對 channel 是否已滿的檢測， 而不是一直等待。

比如我們有一個服務， 當請求進來的時候我們會生成一個 job 扔進 channel， 由其他協程從 channel 中獲取 job 去執行。 但是我們希望當 channel 瞞了的時候， 將該 job 拋棄並回復 【服務繁忙，請稍微再試。】 就可以用 select 實現該需求。