如何在处理结果时正确关闭Goroutines中的共享通道

Question

我试图找到正确的方法来使用工人围棋例程产生的结果，同时在所有工作完成后优雅地退出结果循环。为了说明，我做了以下示例。我的真实世界案例与这个例子略有不同，因为我不知道每个工人围棋例程将返回多少“工作”，显然这些for循环执行固定数量的结果（5）。

我是goroutines和频道的新手，但以下是我所理解的基本租户；

• 只有发送者才能关闭频道
• 在通道上执行范围将继续，直到通道关闭

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

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

最后我想我可以移动wg了。Wait（）在结果循环上方，完成后关闭通道，然后开始打印结果，但这意味着在所有线程上的所有工作完成之前，我无法打印任何结果。

在所有工作线程完成后，如何正确地优雅地退出结果循环，同时也不要等到所有工作完成后才开始打印结果？

Answer 1

我已经编辑了你的代码，可以在没有死锁的情况下工作。问题是，通道上的接收阻塞了主线程，而您的两个goroutine都没有发送更多数据。

此解决方案运行一个新的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)
    }
}

（去游乐场）