接收来自默认阻塞通道的数据

大家好，我们又见面了啊~本文《接收来自默认阻塞通道的数据》的内容中将会涉及到等等。如果你正在学习Golang相关知识，欢迎关注我，以后会给大家带来更多Golang相关文章，希望我们能一起进步！下面就开始本文的正式内容~

我有一个使用 goroutine 从网络读取数据的函数。这个 go 例程有 2 个选择

1. 外部选择，带有计时器，如果读取器失败则重新开始读取
2. 实际读取数据的内部select 所有这些都包含在 waitgroup 中

stop := make(chan struct{})
wg := &sync.WaitGroup{}
wg.Add(1)

go func((){
defer wg.Done()
        ticker := time.NewTicker(restartDuration)
        defer ticker.Stop()
        for {
            listener, err := net.ListenUDP()
             if err != nil {
                return err)
            }

          L: // Labelled for loop 

                for {
                    select {
                    case <-stop:
                        return
                    default:
                        n, _, err := listener.ReadFrom(inboundRTPPacket)
                        if err != nil {
                            break L
                        }

                    }
                }
                if err = listener.Close(); err != nil {
                    logrus.Warn("Could not close udp listener")
                }


            select {
            case <-stop:
                return
            case <-ticker.C:
            }
        }
}()

如果内部选择已删除但不适用于内部选择，则外部选择会收到 close(stop) 调用并执行操作。

我不确定我在这里错过了什么，

正确答案

您的代码在这里是读取阻塞的：

n, _, err := listener.readfrom(inboundrtppacket)

即使稍后执行 close(stop) - 它也不会解锁上面的代码 - 因为 net.listener 不知道您的取消机制。

context.Context 专为此特定目的而设计 - 将 context 传递给任何可能阻塞的调用。如果调用阻塞并且 context 被取消，则调用将取消阻塞。

io.reader 的读取操作不附带上下文支持。这是一个非常困难的问题：

• 马特·赖尔的 blog
• Cap’n Proto

为您提供一个快速但肮脏的解决方案，用于演示目的：

// watches for cancelation:
go func() {
    <-stop
    listener.close()
}()

// ...

// this will unblock upon `close(stop)` with an error
// as the connection is closed
n, _, err := listener.readfrom(inboundrtppacket)

注意：在每个连接处理程序之后必须 close(stop) 以避免 goroutine 泄漏。

由于您使用的是 *net.udpconn，因此更强大的技术是利用 setreaddeadline 方法来取消阻止的读取调用，例如

go func() {

    <-stop // watch for cancelations

    var err error

    // ensure 'listener' has the `SetReadDeadline` method
    // (which *net.UDPConn does)
    if rd, ok := listener.(interface {
        SetReadDeadline(time.Time) error
    }); ok {
        err = rd.SetReadDeadline(time.Now()) // unblock any blocked reads

    } else {
        err = listener.Close() // no deadline method, so default to just closing
    }

    if err != nil {
        log.Println(err)
    }
}()

再次确保 close(stop) 以避免 goroutine 泄漏。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《接收来自默认阻塞通道的数据》文章吧，也可关注golang学习网公众号了解相关技术文章。