Go协程通信：如何实现通道切片([]chan{})的消息接收？

给定一个通道切片([]chan{})，需要实现从该通道数组中接收消息，但遇到以下两个问题：

• 使用select语句，但切片长度不固定，无法确定case的数量
• 遍历切片逐个接收，无消息时会造成阻塞

可以通过goroutine + 统一接收通道的方式解决，把每个通道的接收操作放到独立goroutine中，将所有消息转发到一个统一通道，既能动态适配任意数量的通道，又能避免阻塞。

持续接收所有消息（直到通道关闭） #

import "sync"

func receiveFromChannels(channels []chan interface{}) <-chan interface{} {
    result := make(chan interface{})
    var wg sync.WaitGroup
    wg.Add(len(channels))

    for _, ch := range channels {
        // 启动goroutine处理单个通道
        go func(c chan interface{}) {
            defer wg.Done()
            // 持续接收直到通道关闭
            for msg := range c {
                result <- msg
            }
        }(ch)
    }

    // 所有通道处理完毕后关闭统一接收通道
    go func() {
        wg.Wait()
        close(result)
    }()

    return result
}

使用示例 #

// 外部调用时，只需从返回的通道接收消息
msgChan := receiveFromChannels(yourChannelSlice)
for msg := range msgChan {
    // 处理接收到的消息
    println(msg)
}

单次接收每个通道的一条消息 #

如果只需要从每个通道接收一次消息，可修改goroutine内的逻辑：

go func(c chan interface{}) {
    defer wg.Done()
    msg, ok := <-c
    if ok {
        result <- msg
    }
}(ch)

核心逻辑说明 #

• 每个通道对应一个goroutine，独立处理接收，不会互相阻塞
• 统一接收通道对外屏蔽了原通道的数量差异，外部只需处理这一个通道
• sync.WaitGroup用于跟踪所有goroutine的完成状态，确保所有原通道处理完后关闭统一通道，避免外部接收时永久阻塞

内容的提问来源于stack exchange，提问作者Shin-chan Loves Elephants