Go—— 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")
        }
}

 

一、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 機制】

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

從而實現了對 ch 讀取操作的超時設置。

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!")
}

 

當 select 語句帶有 default 的時候：

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

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

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

 

因為 ch 插入 1 的時候已經滿了， 當 ch 要插入 2 的時候，發現 ch 已經滿了（case1 阻塞住），

則 select 執行 default 語句。 這樣就可以實現對 channel 是否已滿的檢測， 而不是一直等待。

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