Go中select多路复用的实现示例

select {
case val := <-ch1:
    // ch1 可读时执行
case ch2 <- data:
    // ch2 可写时执行
default:
    // 所有通道都不准备好时执行（可选）
}

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

    go func() {
        time.Sleep(1 * time.Second)
        ch1 <- "消息来自 ch1"
    }()

    go func() {
        time.Sleep(2 * time.Second)
        ch2 <- "消息来自 ch2"
    }()

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

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

select {
case msg := <-ch:
    fmt.Println("收到：", msg)
default:
    fmt.Println("没有收到任何数据")
}

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

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
}