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

Question

我正在用 golang 实现 websocket 客户端。
我必须在一个 websocket session 中发送多条消息。
为了处理网络问题，每当连接意外关闭时，我都需要重新连接到 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 消息？

Answer 1

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

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

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

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

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

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

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

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

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

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

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

---

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

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