正确地关闭 Goroutine 中的共享通道

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《正确地关闭 Goroutine 中的共享通道》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

我正在尝试找到使用工作程序 go 例程生成的结果的正确方法，同时在所有工作完成后优雅地退出结果循环。为了说明这一点，我做了以下例子。我的真实情况与此示例略有不同，因为我不知道每个工作程序 go 例程将返回多少“工作”，显然这些 for 循环正在执行固定数量的结果 (5)。

我对 goroutine 和通道很陌生，但以下是我所理解的基本租户；

• 只有发件人才能关闭频道
• 在通道上执行 range 将继续，直到通道关闭

package main

import (
    "fmt"
    "sync"
)

func worker1(r chan string, wg *sync.WaitGroup) {
    for i := 0; i < 5; i++ {
        r <- fmt.Sprintf("1.%d", i)
    }

    wg.Done()
}

func worker2(r chan string, wg *sync.WaitGroup) {
    for i := 0; i < 5; i++ {
        r <- fmt.Sprintf("2.%d", i)
    }

    wg.Done()
}

func main() {
    var wg sync.WaitGroup

    r := make(chan string)
    wg.Add(2)
    go worker1(r, &wg)
    go worker2(r, &wg)

    for i := range r {
        fmt.Printf("Got job result: %s\n", i)
    }

    wg.Wait()
}

这个例子是死锁的，因为范围循环永远不会退出，因为通道永远不会关闭。工作完成后，我可以在通道上执行关闭操作（即将 wg.done() 替换为 close(r)），但是当其他工作 goroutine 尝试将进一步结果发送到已关闭的通道时，我会感到恐慌渠道。

最后，我认为我可以将 wg.wait() 移至结果循环上方，完成后关闭通道，然后开始打印结果，但这意味着在所有工作完成之前我无法打印任何结果在所有线程上完成。

在所有工作线程完成后优雅地退出结果循环，同时也不等到所有工作完成才开始打印结果的正确方法是什么？

解决方案

我已经编辑了您的代码，使其可以无死锁地工作。问题在于通道上的接收正在阻塞主线程，并且您的两个 goroutines 都不再发送任何数据。

此解决方案运行一个新的 goroutine，一旦 waitgroup 完成，它就会关闭结果通道。

package main

import (
    "fmt"
    "sync"
)

func worker1(r chan string, wg *sync.WaitGroup) {
    for i := 0; i < 5; i++ {
        r <- fmt.Sprintf("1.%d", i)
    }

    wg.Done()
}

func worker2(r chan string, wg *sync.WaitGroup) {
    for i := 0; i < 5; i++ {
        r <- fmt.Sprintf("2.%d", i)
    }

    wg.Done()
}

func main() {
    var wg sync.WaitGroup

    r := make(chan string)
    wg.Add(2)
    go worker1(r, &wg)
    go worker2(r, &wg)

    go func() {
        defer close(r)
        wg.Wait()
    }()

    for i := range r {
        fmt.Printf("Got job result: %s\n", i)
    }
}

(Go Playground)

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《正确地关闭 Goroutine 中的共享通道》文章吧，也可关注golang学习网公众号了解相关技术文章。