重新连接时丢弃消息的可重新连接的 websocket

reconnectable websocket which drops message while reconnecting

我正在用 golang 实现 websocket 客户端。
我必须在一个 websocket 会话中发送多条消息。
为了处理网络问题,每当连接意外关闭时,我需要重新连接到 websocket 服务器。
目前我正在考虑这样的实施。

for {
    select {
        case t := <-message:
            err := connection.WriteMessage(websocket.TextMessage, []byte(t.String()))
            if err != nil {
               // If session is disconnected.
               // Try to reconnect session here.
               connection.reconnect()
            }
        case t := <- errSignal:
            panic()
    }
}

在上面的示例中,消息在重新连接时堆叠。
这不是我的目的。
重新连接时如何删除 websocket 消息?

messages stacks while reconnecting. This is not preferable for my purpose. How can I drop websocket messages while reconnecting?

我认为message是一个缓冲通道。我不清楚您在重新连接时删除 websocket 消息时究竟要求什么行为,或者为什么要这样做,但是您有一些选项可以跟踪与重新连接相关的消息并根据需要处理它们。

首先,缓冲通道的行为类似于队列:先进先出 (FIFO)。您始终可以使用接收操作从通道中弹出一个元素。您不需要将其传递给变量或使用它。所以说你只是想从重新连接周围的队列中删除前两条消息而不对它们做任何事情(不知道为什么),你可以:

if err != nil {
     // If session is disconnected.
     // Try to reconnect session here.
     connection.reconnect()
     // drop the next two messages
     <-message
     <-message
}

但是这将从队列的前面删除消息。如果在您开始重新连接时队列不为空,它不会专门删除在重新连接期间添加的消息。

如果要将删除的消息数与重新连接期间添加的消息数相关联,可以使用通道长度:

before := len(message)
connection.reconnect()
after := len(message)
for x := 0; x < after - before; x++ {
    <-message
}

同样,这将从队列的前面移除,我不知道你为什么要这样做,除非你只是想清空频道。

并且如果频道在重新连接开始时为 non-empty,并且您确实想要删除在重新连接期间添加的消息,则可以使用 the time package。可以为任何 Go 类型定义通道。因此,创建一个包含消息字段和时间戳的结构,将缓冲的 message 通道重新定义为结构类型,并在发送消息之前设置时间戳。保存重新连接之前的 "before" 时间戳和之后的 "after"。然后在处理收到的消息之前,您可以检查它是否在 after/before window 中,如果是,则将其删除(而不是写入)。您可以创建一个新的数据结构来保存类型上的几个 before/after windows 方法,以检查给定时间是否在任何范围内。同样,我不知道你为什么要这样做。

也许更好的解决方案是只限制通道的缓冲区大小,然后当通道已满时无法向通道添加新消息。那能满足你的需求吗?如果您有理由丢弃消息,也许您可​​以解释更多关于您的目标和设计的信息——尤其是解释您想要丢弃哪些消息。

如果您包含更多相关代码,例如消息通道的声明,它也可能会澄清您的问题。


编辑:提问者在对此答案的评论中添加了信息,并对问题发表评论。

缓冲通道和非缓冲通道之间的选择部分取决于您是否希望发送方在接收方不可用时阻塞。如果这是您唯一的接收器,则在重新连接期间它将不可用。因此,如果它适合您的发件人阻止设计,您可以使用无缓冲通道而不是时间戳,并且在重新连接期间不会向通道添加任何消息。但是,阻塞在发送通道的发送者将等待接收者使用他们的旧消息,只有发送成功后,他们才会发送包含当前数据的新消息。如果这对您不起作用,缓冲通道可能是更好的选择。