原文鏈接： Go 語言 select 都能做什么炫乓？

在 Go 語言中，select 是一個關鍵字，用于監(jiān)聽和 channel 有關的 IO 操作。

通過 select 語句，我們可以同時監(jiān)聽多個 channel衫画，并在其中任意一個 channel 就緒時進行相應的處理。

本文將總結一下 select 語句的常見用法府瞄，以及在使用過程中的注意事項碧磅。

基本語法

select 語句的基本語法如下：

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

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

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

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

使用規(guī)則

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

1. select 語句只能用于通道操作舒裤，用于在多個通道之間進行選擇喳资，以監(jiān)聽通道的就緒狀態(tài)，而不是用于其他類型的條件判斷腾供。
2. select 語句可以包含多個 case 子句仆邓，每個 case 子句對應一個通道操作。當其中任意一個通道就緒時伴鳖，相應的 case 子句會被執(zhí)行节值。
3. 如果多個通道都已經(jīng)就緒，select 語句會隨機選擇一個通道來執(zhí)行榜聂。這樣確保了多個通道之間的公平競爭搞疗。
4. 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.

超時處理

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

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 分支箱蟆。

輸出：

Timeout occurred.

以上就是本文的全部內(nèi)容沟绪，如果覺得還不錯的話歡迎點贊，轉(zhuǎn)發(fā)和關注空猜，感謝支持近零。

推薦閱讀：

• Go 語言 context 都能做什么？