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

目前golang学习网上已经有很多关于Golang的文章了，自己在初次阅读这些文章中，也见识到了很多学习思路；那么本文《当您使用范围通道中断 for 语句时会发生什么》，也希望能帮助到大家，如果阅读完后真的对你学习Golang有帮助，欢迎动动手指，评论留言并分享~

我正在按照此代码获取具有渠道的惰性数字范围

// 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 来解决此问题吗？ 有没有安全的方法来检查范围是否中断并停止迭代器中的for循环？

解决方案

如果一个 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

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

读取协程：

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

本篇关于《当您使用范围通道中断 for 语句时会发生什么》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！