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

千瑶同学_3559

发布时间：2025-09-06 17:35:02

748人浏览过

来源于php中文网

原创

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

本教程介绍如何将多个只读的 Go 通道（

在并发编程中，经常会遇到需要将多个数据源合并到一个数据流中的情况。在 Go 语言中，通道（channel）是实现并发的重要工具，因此将多个通道的数据合并到一个通道的需求也十分常见。本文将介绍如何使用 sync.WaitGroup 将多个只读通道多路复用到一个只写通道。

使用 sync.WaitGroup 实现通道多路复用

以下是一个通用的 combine 函数，它接受一个只读通道切片 inputs 和一个只写通道 output 作为参数，并将所有输入通道的数据合并到输出通道中。

AI舌诊面诊问诊医疗系统+诊疗商城系统源码

AI 舌诊面诊问诊医疗系统 + 诊疗商城系统源码是一套结合 AI 辅助诊断与商品转化的健康类程序，基于浩创云、ThinkPHP（PHP）开发，支持微信服务号与 H5 端，适合做中医舌诊面诊问答、健康建议推送以及配套诊疗商品的运营。

下载

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

代码解释：

sync.WaitGroup： sync.WaitGroup 用于等待一组 goroutine 完成。Add(1) 增加计数器，Done() 减少计数器，Wait() 阻塞直到计数器为 0。
combine 函数：
- group.Add(1): 为每个输入通道创建一个新的 goroutine，因此需要增加 WaitGroup 的计数器。
- go func(input
- go func() { group.Wait(); close(output) }()：启动一个新的 goroutine，该 goroutine 等待所有输入通道的 goroutine 完成（即 WaitGroup 的计数器为 0），然后关闭输出通道。
关闭输出通道： 只有当所有输入通道都关闭后，才能关闭输出通道。否则，接收者可能会在发送者完成发送之前就关闭通道，导致数据丢失。

运行结果：

1
2
3
4
5
6
Done

注意事项

通道方向：
错误处理： 在实际应用中，需要考虑错误处理。例如，如果某个输入通道发生错误，可能需要关闭输出通道并返回错误信息。
死锁： 在使用通道时，需要避免死锁。例如，如果一个 goroutine 试图从一个空的通道接收数据，并且没有其他 goroutine 向该通道发送数据，则会发生死锁。
性能： 对于高并发的场景，可能需要考虑性能优化。例如，可以使用缓冲通道来提高吞吐量。