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

Question

当我注意到并非管道中的所有关闭打印都被打印时，我根据《Go 中的并发》一书中的示例运行以下代码。
看到“完成倍增！” 不见了。
另一方面，NumGoroutine() 显示仅主函数正在运行。
下面的代码有什么问题？

package main

import (

    "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。原因如下：如果done消息在生成器将数字写入通道并且乘法器读取它之后立即到达，则乘法器会看到该数字done可用并选择该数字。multiplier打印消息时就是这种情况done。如果当donemultiplier 在 for 循环中等待时消息到达，则 multiplier 将接收输入通道（而不是通道）上的关闭消息done，从而导致 for 循环终止而不打印done消息。出现问题的原因是您正在 for 循环中读取通道，然后进行选择。在等待 for 循环从通道读取数据时，不会评估与 select 相关的任何事件。处理此问题的更好方法是不使用 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调用来公开它们由于到达输入流末尾而退出的点，则行为会略有变化（可能是由于时间原因；同时我正在输入这个），输出变成：add after select2multiply after selectgenerator after selectmultiply after selectadd after select4generator after selectmultiply after selectadd after select6generator after selectClosed donedone multiply !add got end of stream - done!finished iterating pipelinegenerator after selectdone generator!ramaining goroutines: 1finished!通常更合理的做法是仅让生成器本身接收done信号，并让管道函数始终写入其所有结果，这使它们更可预测。当然，无论谁正在读取每个管道，都必须读到最后——但是您已经在主 goroutine 中执行了此操作，因此我们只是将其传播到整个管道。 这是您的代码的简化版本，以这种方式执行此操作；它输出：2generator after select4generator after select6generator after selectClosed done8generator after selectdone generator!multiply got end of stream - done!add got end of stream - done!finished iterating pipelineremaining goroutines: 1请注意，这一次，我们从最终生成值 (3) 中得到最终计算值 (8)。