快捷導航

詳解Go語言中select語句的常見用法

更新時間：2023年07月19日 08:30:37 作者：AlwaysBeta

這篇文章主要是來和大家介紹一下Go語言中select?語句的常見用法，以及在使用過程中的注意事項，文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下

基本語法

select 語句的基本語法如下：

select?{
case?<-channel1:
????//?通道?channel1?就緒時的處理邏輯
case?data?:=?<-channel2:
????//?通道?channel2?就緒時的處理邏輯
default:
????//?當沒有任何通道就緒時的默認處理邏輯
}

看到這個語法，很容易想到 switch 語句。

雖然 select 語句和 switch 語句在表面上有些相似，但它們的用途和功能是不同的。

switch 用于條件判斷，而 select 用于通道操作。不能在 select 語句中使用任意類型的條件表達式，只能對通道進行操作。

使用規(guī)則

雖然語法簡單，但是在使用過程中，還是有一些地方需要注意，我總結(jié)了如下四點：

select 語句只能用于通道操作，用于在多個通道之間進行選擇，以監(jiān)聽通道的就緒狀態(tài)，而不是用于其他類型的條件判斷。
select 語句可以包含多個 case 子句，每個 case 子句對應(yīng)一個通道操作。當其中任意一個通道就緒時，相應(yīng)的 case 子句會被執(zhí)行。
如果多個通道都已經(jīng)就緒，select 語句會隨機選擇一個通道來執(zhí)行。這樣確保了多個通道之間的公平競爭。
select 語句的執(zhí)行可能是阻塞的，也可能是非阻塞的。如果沒有任何一個通道就緒且沒有默認的 default 子句，select 語句會阻塞，直到有一個通道就緒。如果有 default 子句，且沒有任何通道就緒，那么 select 語句會執(zhí)行 default 子句，從而避免阻塞。

多路復用

select 最常見的用途之一，同時監(jiān)聽多個通道，并根據(jù)它們的就緒狀態(tài)執(zhí)行不同的操作。

package?main
import?(
????"fmt"
????"time"
)
func?main()?{
????c1?:=?make(chan?string)
????c2?:=?make(chan?string)
????go?func()?{
????????time.Sleep(3?*?time.Second)
????????c1?<-?"one"
????}()
????go?func()?{
????????time.Sleep(3?*?time.Second)
????????c2?<-?"two"
????}()
????select?{
????case?msg?:=?<-c1:
????????fmt.Println(msg)
????case?msg?:=?<-c2:
????????fmt.Println(msg)
????}
}

執(zhí)行上面的代碼，程序會隨機打印 one 或者 two，如果通道為空的話，程序就會一直阻塞在那里。

非阻塞通信

當通道中沒有數(shù)據(jù)可讀或者沒有緩沖空間可寫時，普通的讀寫操作將會阻塞。

但通過 select 語句，我們可以在沒有數(shù)據(jù)就緒時執(zhí)行默認的邏輯，避免程序陷入無限等待狀態(tài)。

package?main
import?(
????"fmt"
)
func?main()?{
????channel?:=?make(chan?int)
????select?{
????case?data?:=?<-channel:
????????fmt.Println("Received:",?data)
????default:
????????fmt.Println("No?data?available.")
????}
}

執(zhí)行上面代碼，程序會執(zhí)行 default 分支。

輸出：

No data available.

超時處理

通過結(jié)合 select 和 time.After 函數(shù)，我們可以在指定時間內(nèi)等待通道就緒，超過時間后執(zhí)行相應(yīng)的邏輯。

package?main
import?(
????"fmt"
????"time"
)
func?main()?{
????channel?:=?make(chan?int)
????select?{
????case?data?:=?<-channel:
????????fmt.Println("Received:",?data)
????case?<-time.After(3?*?time.Second):
????????fmt.Println("Timeout?occurred.")
????}
}

執(zhí)行上面代碼，如果 channel 在 3 秒內(nèi)沒有數(shù)據(jù)可讀，select 會執(zhí)行 time.After 分支。

輸出：