快捷導(dǎo)航

Go中select多路復(fù)用的實(shí)現(xiàn)示例

更新時(shí)間：2025年08月14日 11:20:42 作者：程序員愛(ài)釣魚(yú)

Go的select用于多通道通信,實(shí)現(xiàn)多路復(fù)用,支持隨機(jī)選擇、超時(shí)控制及非阻塞操作,建議合理使用以避免協(xié)程泄漏和死循環(huán),感興趣的可以了解一下

select 是 Go 并發(fā)編程中非常強(qiáng)大的語(yǔ)法結(jié)構(gòu)，它允許程序同時(shí)等待多個(gè)通道操作的完成，從而實(shí)現(xiàn)多路復(fù)用機(jī)制，是協(xié)程調(diào)度、超時(shí)控制、通道競(jìng)爭(zhēng)等場(chǎng)景的核心工具。

一、什么是select

select 類(lèi)似于 switch 語(yǔ)句，但它用于監(jiān)聽(tīng)多個(gè)通道的發(fā)送/接收操作。一旦其中任意一個(gè)通道準(zhǔn)備就緒，select 就會(huì)執(zhí)行相應(yīng)的語(yǔ)句塊。

基本語(yǔ)法：

select {
case val := <-ch1:
    // ch1 可讀時(shí)執(zhí)行
case ch2 <- data:
    // ch2 可寫(xiě)時(shí)執(zhí)行
default:
    // 所有通道都不準(zhǔn)備好時(shí)執(zhí)行（可選）
}

二、select 使用示例

示例1：監(jiān)聽(tīng)多個(gè)通道輸入

func main() {
    ch1 := make(chan string)
    ch2 := make(chan string)

    go func() {
        time.Sleep(1 * time.Second)
        ch1 <- "消息來(lái)自 ch1"
    }()

    go func() {
        time.Sleep(2 * time.Second)
        ch2 <- "消息來(lái)自 ch2"
    }()

    select {
    case msg1 := <-ch1:
        fmt.Println(msg1)
    case msg2 := <-ch2:
        fmt.Println(msg2)
    }
}

輸出（大概率）：

消息來(lái)自 ch1

誰(shuí)先準(zhǔn)備好，誰(shuí)被執(zhí)行。

三、select的特性

隨機(jī)選擇：如果多個(gè) case 同時(shí)滿(mǎn)足，Go 會(huì)隨機(jī)選擇一個(gè)執(zhí)行，避免饑餓。
阻塞行為：當(dāng)沒(méi)有任何 case 可以運(yùn)行時(shí)，select 會(huì)阻塞，除非有 default。
可配合 for 使用：實(shí)現(xiàn)多路輪詢(xún)、協(xié)程調(diào)度等高級(jí)用法。

四、使用select實(shí)現(xiàn)超時(shí)機(jī)制

select 搭配 time.After() 可實(shí)現(xiàn)通道的超時(shí)控制：

select {
case msg := <-ch:
    fmt.Println("收到消息：", msg)
case <-time.After(2 * time.Second):
    fmt.Println("超時(shí)未收到")
}

實(shí)用場(chǎng)景：

網(wǎng)絡(luò)請(qǐng)求超時(shí)
等待任務(wù)執(zhí)行完成
控制并發(fā)阻塞時(shí)間

五、非阻塞通信：使用default

select {
case msg := <-ch:
    fmt.Println("收到：", msg)
default:
    fmt.Println("沒(méi)有收到任何數(shù)據(jù)")
}

不等待，立即返回默認(rèn)分支。

六、監(jiān)聽(tīng)通道關(guān)閉

配合 range 和 select，可以?xún)?yōu)雅處理通道關(guān)閉：

for {
    select {
    case msg, ok := <-ch:
        if !ok {
            fmt.Println("通道已關(guān)閉")
            return
        }
        fmt.Println("收到：", msg)
    }
}

七、實(shí)踐：合并多個(gè)輸入通道

func merge(ch1, ch2 <-chan string) <-chan string {
    out := make(chan string)
    go func() {
        for {
            select {
            case msg := <-ch1:
                out <- msg
            case msg := <-ch2:
                out <- msg
            }
        }
    }()
    return out
}

? 實(shí)現(xiàn)“扇入”（fan-in）模式，將多個(gè)輸入流合并成一個(gè)輸出。

八、小結(jié)

功能	是否支持
同時(shí)監(jiān)聽(tīng)多個(gè)通道	?
隨機(jī)選擇就緒的通道執(zhí)行	?
支持默認(rèn)分支防止阻塞	?
可實(shí)現(xiàn)超時(shí)控制與輪詢(xún)	?
實(shí)現(xiàn)非阻塞收發(fā)或關(guān)閉判斷	?