当您使用范围通道中断 for 语句时会发生什么

Question

我正在关注此代码以获取具有 channels

的惰性数字范围

// iterator
func iterator(n int, c chan int) {
    for i := 0; i < n; i++ {
        c <- i
    }
    close(c)
    fmt.Println("iterator End")
}

c := make(chan int)
go iterator(5, c)
for i := range c {
    fmt.Println(i)
}

这将按预期打印

0
1
2
3
4
fmt.Println("iterator End")

但是当我像这样打破 for 循环时发生了什么

c := make(chan int)
go getNumbers(5, c)
for i := range c {
    if i == 2 {
        break
    }
    fmt.Println(i)
}

goroutine 似乎被阻塞了，因为从不打印iterator End（我也尝试让主线程休眠）。 我想知道如何处理这种情况？ 我需要使用select 来解决这个问题吗？ 有什么安全的方法可以检查范围是否 break 并停止迭代器中的 for-loop？

Answer 1

如果一个 goroutine 写入一个无缓冲的通道并且没有其他 goroutine 正在从该通道读取 - 那么写入将永远阻塞。这将导致 goroutine 泄漏。这就是你正在经历的。

如果你有一个写入通道的“生产者”goroutine，你需要一种方法来通知它停止。关闭通道在这里不是关键部分 - 因为通道超出范围时会被垃圾收集。阻塞的 goroutine（永远不会解除阻塞）被认为是泄漏，因为它永远不会被回收，所以你真的需要结束 goroutine。

您可以通过多种方式表示退出意图 - 最受欢迎的两种方式是：

• done 频道；或
• context.Context取消

信号：完成通道

func iterator(n int, c chan int, done <-chan struct{}) {
    for i := 0; i < n; i++ {
        select {
        case c <- i:
        case <-done:
            break
        }
    }
    close(c)
    fmt.Println("iterator End")
}

阅读器协程：

c := make(chan int)
done := make(chan struct{})
go iterator(5, c, done)
for i := range c {
    if i == 2 {
        break
    }
    fmt.Println(i)
}
close(done) // signal writer goroutine to quit

信号：context.Context

func iterator(ctx context.Context, n int, c chan int) {
        defer close(c)
        defer fmt.Println("iterator End")

        for i := 0; i < n; i++ {
                select {
                case c <- i:
                case <-ctx.Done():
                        fmt.Println("canceled. Reason:", ctx.Err())
                        return
                }
        }
}

读取 goroutine：

func run(ctx context.Context) {
        ctx, cancel := context.WithCancel(ctx)
        defer cancel()  // call this regardless - avoid context leaks - but signals producer your intent to stop
        c := make(chan int)
        go iterator(ctx, 5, c)
        for i := range c {
                if i == 2 {
                        break
                }
                fmt.Println(i)
        }
}

https://play.golang.org/p/4-fDyCurB7t