关闭完成通道后 goroutine 缺少打印

Question

当我注意到并非管道中的所有关闭打印都被打印时，我正在根据“Go 中的并发”一书中的示例运行以下代码。看到“完成倍增！” 不见了。另一方面，NumGoroutine() 只显示 main 函数正在运行。以下代码有什么问题？（https://play.golang.org/p/tkFgvKboVgS）package mainimport (    "fmt"    "runtime"    "time")func main() {    generator := func(done <-chan struct{}) <-chan int {        intStream := make(chan int)        i:=0        go func() {            defer close(intStream)            for {                select {                case <-done:                    fmt.Println("done generator!")                    return                case intStream <- i:                    time.Sleep(1 * time.Second)                    i++                }                fmt.Println("generator after select")            }        }()        return intStream    }    multiply := func(        done <-chan struct{},        intStream <-chan int,        multiplier int,    ) <-chan int {        multipliedStream := make(chan int)        go func() {            defer close(multipliedStream)            for i := range intStream {                select {                case <-done:                    fmt.Println("done multiply !")                    return                case multipliedStream <- i * multiplier:                }                fmt.Println("multiply after select")            }        }()        return multipliedStream    }    add := func(        done <-chan struct{},        intStream <-chan int,        additive int,    ) <-chan int {        addedStream := make(chan int)        go func() {            defer close(addedStream)            for i := range intStream {                select {                case <-done:                    fmt.Println("done add !")                    return                case addedStream <- i + additive:                }                fmt.Println("add after select")            }        }()        return addedStream    }

Answer 1

有些代码路径不会打印某些done消息。调度程序碰巧选择了一个不为multiply. 如果您稍微更改代码（例如，在与现在不同的实例上登录），您会发现它也可能会丢失add done消息。（https://play.golang.org/p/meEPM5GR9Rr）。原因如下：

如果done消息在生成器将数字写入通道并且乘法器读取它之后立即到达，那么乘法器会看到done可用并选择它。multiplier打印done消息时就是这种情况。如果done消息在 multiplier 在 for 循环中等待时到达，则 multiplier 将接收输入通道（而不是done通道）的关闭，导致 for 循环终止而不打印done消息。

出现问题是因为您正在从 for 循环中的通道读取，然后进行选择。在等待 for 循环从通道中读取数据时，不会评估与选择相关的任何事件。

解决这个问题的更好方法是不使用 for 循环从通道中读取。例如：

for {

     select {

        case <-done:

           return

        case i, ok:= <-intstream:

           if !ok {

              return

           }

           select {

               case <- done:

                    return

               case addedStream <- i + additive:

           }

     }

}

Answer 2

你的add和multiply例程不是永远的循环，而是for ... range循环。因此，在每个循环的顶部，它们等待下一个整数，而不是等待select接收关闭done或将结果发送到它们的流。这不是问题，但这意味着如果它们的输入流关闭，它们将返回而不进入循环本身。

如果我添加fmt.Println调用以暴露由于到达输入流的末尾而退出的点，则行为会略有变化（可能是由于时间的原因；我没有费心去解释它，而Burak Serdar已经发布了他的答案我正在输入这个），输出变为：

add after select

2

multiply after select

generator after select

multiply after select

add after select

4

generator after select

multiply after select

add after select

6

generator after select

Closed done

done multiply !

add got end of stream - done!

finished iterating pipeline

generator after select

done generator!

ramaining goroutines: 1

finished!

通常更合理的是只让生成器本身接收done信号，并使流水线函数始终写入所有结果，这使得它们更可预测。当然，读取每个管道的人必须读到最后——但你已经在主 goroutine 中这样做了，所以我们只是在整个过程中传播它。 这是以这种方式执行的代码的简化版本；它输出：

2

generator after select

4

generator after select

6

generator after select

Closed done

8

generator after select

done generator!

multiply got end of stream - done!

add got end of stream - done!

finished iterating pipeline

remaining goroutines: 1

请注意，这一次，我们从最终生成的值 (3) 中得到最终的计算值 (8)。