博客 / 詳情

返回

Go語言select都能做什麼?

在Go語言中,select是一個關鍵字,用於監聽和channel有關的IO操作。

通過select語句,我們可以同時監聽channel,並在其中任意一個channel就緒多個時進行相應的處理。

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

基本語法

select語句的基本語法如下:

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

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

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

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

使用規則

雖然語法簡單,但是在使用過程中,還是有一些地方需要注意,我總結了以下四點:

  1. select該語句僅用於通道操作選擇,用於在多個通道之間進行,以監聽通道的就緒狀態,而不用於其他類型的條件判斷。
  2. select語句可以包含多個case子句,每個case子句對應一個通道操作。當其中任意一個通道適合時,相應的case子句就會被執行。
  3. 如果多個通道都已經就緒,select語句會隨機選擇一個通道來執行。這樣保證了多個通道之間的公平競爭。
  4. select語句的執行可能是阻塞的,也可能是非阻塞的。如果沒有任何一個通道就緒且沒有默認的default子句,select語句會阻塞,直到有一個通道就緒。如果有default子句,且沒有任何通道就緒,則語句會阻塞,直到有一個通道就緒select。會執行default子句,從而避免阻塞。

多路複用

select最常見的用途之一,同時監聽通道,並根據它們的多個就緒狀態執行不同的操作。

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

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

阻礙非通信

當通道中沒有數據中斷或者沒有緩衝空間可寫入時,普通的讀寫器操作將會阻塞。

但通過select語句,我們可以在沒有數據完整性時默認執行的邏輯,避免程序陷入無限等待狀態。

package main

import (
    "fmt"
)

func main() {
    channel := make(chan int)

    select {
    case data := <-channel:
        fmt.Println("Received:", data)
    default:
        fmt.Println("No data available.")
    }
}

執行上面的代碼,程序會執行default分支。

輸出:
No data available.

超時處理

通過結合selecttime.After函數,我們可以在指定期限等待通道就緒,超過時間後執行相應的邏輯。

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

執行上面代碼,如果channel3秒內沒有數據裁決,select會執行time.After分支。

輸出:

Timeout occurred.

關於Masutaa

Masutaa是個互聯網從業者自由協作交流平台,鏈接行業內TOP10%人才!目前平台上已經有將近400名互聯網尖端人才,其中近70%的從業者從業年限超3年。加入Masutaa,加入自由生活!感興趣的朋友可以搜索masutaa官網!

user avatar kuaishoutech 頭像 panshenlian 頭像 yaoyaolx_wiki 頭像
3 位用戶收藏了這個故事!

發佈 評論

Some HTML is okay.