缓冲通道并关闭它

Question

我有一个 sn-p 代码，我试图根据我如何放置 close 调用和位置来理解它

func main() {
    ch := make(chan int, 2)

    go func(ch chan int) {
        for i := 1; i <= 5; i++ {
            ch <- i
            fmt.Println("Func goroutine sends data: ", i)
        }
        //Pos1 - Works perfectly
        //close(ch)
    }(ch)

    fmt.Println("Main goroutine sleeps 2 seconds")
    time.Sleep(time.Second * 2)

    fmt.Println("Main goroutine begins receiving data")
    //Pos2 - Puts in only 2 ints 1 and 2 and then prints only that
    //close(ch)
    for d := range ch {
        fmt.Println("Main goroutine received data:", d)
    }
    //Pos3 - Throws fatal error
    close(ch)
}

我一直在尝试理解和阅读这方面的博客，但仍然无法理解一些东西

1. 当我在 Pos1 关闭时，它工作正常。但我不知道为什么 有用。缓冲区不能容纳超过 2 个元素在任何给定 时间，所以当写入 2 个元素时，循环将阻塞，直到 主路由进行读取。但我想在一个范围内做一个 缓冲通道，范围函数必须事先知道有多少 必须关闭该通道的迭代元素。为什么 close 在这个职位上工作吗？
2. 当我把它作为位置 2 时，它只打印 2 个元素，这是有道理的，但为什么 for loop 在尝试将更多元素写入关闭的通道时没有抛出异常？
3. 当我在 Pos3 关闭时，我得到一个异常 fatal error: all goroutines are asleep - deadlock! 尽管所有 5 个整数都被打印出来。这是为什么呢？

Answer 1

但我想在缓冲通道上做一个范围，范围函数必须事先知道要迭代多少元素，并且必须关闭该通道。

这种假设是错误的，是所有误解的根源。

Go 规范中描述了在通道上测距的行为：https://golang.org/ref/spec#For_statements

对于通道，生成的迭代值是通道上发送的连续值，直到通道关闭。如果通道为 nil，则范围表达式将永远阻塞。

for 语句求值时不需要关闭通道，语句不需要知道元素个数。

所以，在您的代码中，当您将close 放入Pos1 时，确实是正确的做法。当你把它放在Pos3时，for循环等待通道关闭，这只能发生在for循环本身之后，所以它是一个死锁。

将close 放入Pos2 是有问题的，而且行为有点棘手。 可能会引发错误，但也可以只输出两个数字。这是因为在for循环之前关闭通道时，循环可以无阻塞运行，然后main()返回。当main() 返回时，Go 程序结束。是否报错，完全取决于调度器是否在进程间切换到goroutine，这并不能保证。