多路复用多个 Go 通道到一个通道

package main

import (
    "fmt"
    "sync"
)

func combine(inputs []<-chan int, output chan<- int) {
    var group sync.WaitGroup

    // 为每个输入通道启动一个 goroutine
    for i := range inputs {
        group.Add(1) // 增加 WaitGroup 的计数器
        go func(input <-chan int) {
            defer group.Done() // goroutine 结束时减少计数器
            for val := range input {
                output <- val // 将数据发送到输出通道
            }
            // 输入通道关闭时，range 循环结束，goroutine 退出
        }(inputs[i])
    }

    // 启动一个 goroutine 等待所有输入通道完成
    go func() {
        group.Wait()      // 阻塞直到 WaitGroup 的计数器为 0
        close(output) // 关闭输出通道，通知接收者没有更多数据
    }()
}

func main() {
    // 创建一些示例输入通道
    input1 := make(chan int)
    input2 := make(chan int)
    input3 := make(chan int)

    // 创建输出通道
    output := make(chan int)

    // 将输入通道放入切片
    inputs := []<-chan int{input1, input2, input3}

    // 启动 combine 函数
    go combine(inputs, output)

    // 向输入通道发送数据
    go func() {
        input1 <- 1
        input1 <- 2
        close(input1)
    }()

    go func() {
        input2 <- 3
        input2 <- 4
        input2 <- 5
        close(input2)
    }()

    go func() {
        input3 <- 6
        close(input3)
    }()

    // 从输出通道接收数据并打印
    for val := range output {
        fmt.Println(val)
    }

    fmt.Println("Done")
}

1
2
3
4
5
6
Done