Go 中如何正确判断输出通道已关闭并安全结束协程处理

花韻仙語

发布时间：2026-01-02 13:49:25

377人浏览过

来源于php中文网

原创

Go 中如何正确判断输出通道已关闭并安全结束协程处理

在 go 并发编程中，当多个 worker 协程向同一输出通道（output channel）发送结果时，主协程需可靠感知“所有 worker 已完成且无更多数据”，从而安全退出 `for range` 循环。`sync.waitgroup` 是最简洁、标准且符合 go idioms 的解决方案。

在典型的生产者-消费者并发模型中（如 Rob Pike 提倡的“扇入”模式），我们常启动多个 worker 协程从输入通道读取任务、处理后将结果写入共享的输出通道。此时，主协程通过 for range outchan 消费结果——但该循环仅在通道被显式关闭后才会自动退出；若无人关闭通道，循环将永久阻塞，导致程序无法正常终止。

因此，关键在于：何时、由谁来关闭输出通道？ 答案是：由一个“协调协程”在确认所有 worker 全部完成工作后执行 close(outchan)。而 sync.WaitGroup 正是为此类场景设计的同步原语——它提供线程安全的计数器，支持 Add()、Done() 和阻塞式 Wait()，完美匹配“启动 N 个 worker → 等待全部结束 → 关闭通道”的控制流。

以下是推荐实现方式：

与光AI

一站式AI视频工作流创作平台

下载

package main

import (
    "fmt"
    "sync"
    "time"
)

type Result struct {
    ID    int
    Value string
}

func worker(wg *sync.WaitGroup, in <-chan int, out chan<- Result) {
    defer wg.Done() // 确保无论何种退出路径都调用 Done()

    for task := range in {
        // 模拟耗时处理
        time.Sleep(100 * time.Millisecond)
        result := Result{
            ID:    task,
            Value: fmt.Sprintf("processed-%d", task),
        }
        out <- result
    }
}

func main() {
    const numWorkers = 3

    // 输入通道（可由生产者协程填充）
    in := make(chan int, 10)
    // 输出通道（无缓冲，避免阻塞；也可根据需要设缓冲）
    out := make(chan Result, 10)

    var wg sync.WaitGroup

    // 启动 worker 协程
    for i := 0; i < numWorkers; i++ {
        wg.Add(1)
        go worker(&wg, in, out)
    }

    // 启动协调协程：等待所有 worker 完成后关闭输出通道
    go func() {
        wg.Wait()
        close(out) // 关键：关闭后，for range 将自然退出
    }()

    // 主协程：发送任务到输入通道（模拟生产者）
    go func() {
        for i := 1; i <= 5; i++ {
            in <- i
        }
        close(in) // 关闭输入通道，通知 worker 无新任务
    }()

    // 消费所有输出结果（自动在 out 关闭后退出）
    for res := range out {
        fmt.Printf("Received: %+v\n", res)
    }

    fmt.Println("All done.")
}

✅ 关键要点说明：