Golang中switch語句和select語句的用法教程

本文主要給大家介紹了關于Golang中switch和select用法的相關內容，分享出來供大家參考學習，下面來一起看看詳細的介紹：

一、switch語句

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

1、表達式switch語句

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

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

names := []string{"Golang","java","PHP"} 
switch name:=names[0];name { 
case "Golang": 
 fmt.Println("Golang") 
... 
default: 
 fmt.Println("Unknown") 
} 

2、類型switch語句

類型switch語句與一般形式有兩點差別。第一點，緊隨case關鍵字的不是表達式，而是類型說明符。類型說明符由若干個類型字面量組成，且多個類型字面量之間由英文逗號分隔。第二點，它的switch表達式是非常特殊的。這種特殊的表達式也起到了類型斷言的作用，但其表現形式很特殊，如：v.(type) , 其中v必須代表一個接口類型的值。該類表達式只能出現在類型switch語句中，且只能充當switch表達式。一個類型switch語句的示例如下：

v := 11 
switch i := interface{}(v).(type) { 
case int, int8, int16, int32, int64: 
 fmt.Println("A signed integer:%d. The type is %T. \n", v, i) 
case uint, uint8, uint16, uint32, uint64: 
 fmt.Println("A unsigned integer: %d. The type is %T. \n", v, i) 
default: 
 fmt.Println("Unknown!") 
} 

我們這里把switch表達式的結果賦給了一個變量。如此以來，我們就可以在該switch語句中使用這個結果了。這段代碼被執行后,輸出:"A signed integer:11. The type is int. "

最后說一下fallthrough。它既是一個關鍵字，又可以代表一條語句。fallthrough語句可被包含在表達式switch語句中的case語句中。它的作用是使控制權流轉到下一個case。不過要注意fallthrough語句僅能作為case語句中的最后一條語句出現。并且，包含它的case語句不是其所屬switch語句的最后一條case語句。

二、select語句

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

示例：

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

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

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

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

【使用 select 實現 timeout 機制】

如下：

timeout := make(chan bool, 1) 
go func() { 
 time.Sleep(time.Second * 10) 
 timeout <- true 
}() 
select { 
case <-pssScanResponseChan: 
 
case <-timeout: 
 fmt.PrintIn("timeout!") 
} 

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

下面這個更有意思一點。

當 select 語句帶有 default 的時候：

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

此時因為 ch2 和 ch3 都為空，所以 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 實現該需求。

此外，利用default特性，我們可以使用select語句將chan清空，如下：

flag := false 
for { 
 select { 
 case <-pssScanResponseChan: 
 continue 
 default: 
 flag = true 
 } 
 if true == flag { 
 break 
 } 
} 

總結

以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作能帶來一定的幫助，如果有疑問大家可以留言交流，謝謝大家對億速云的支持。