接收可写入部分的 Protobuf 编码消息？

在IT行业这个发展更新速度很快的行业，只有不停止的学习，才不会被行业所淘汰。如果你是Golang学习者，那么本文《接收可写入部分的 Protobuf 编码消息？》就很适合你！本篇内容主要包括##content_title##，希望对大家的知识积累有所帮助，助力实战开发！

我正在尝试通过 tcp 在 golang 中发送和接收 protobuff 编码的消息，其中发送方可以在操作中途取消 write()，并且接收方可以正确接收部分消息。

请注意，我使用单个 tcp 连接无限地发送不同用户定义类型的消息（这不是每个连接消息的情况）

为了具体解释我的问题，首先我将介绍如何在不进行部分写入的情况下实现发送/接收。

在我的程序中，有多种类型的消息，在 .proto 文件中定义。我将解释一种此类消息类型的机制。

message messagetype {
  int64 sender = 1;
  int64 receiver = 2;
  int64 operation = 3;
  string message = 4;
}

然后我使用 golang protobuf 插件来生成存根。

然后在发送方，以下是我发送的方式。

func send(w *bufio.writer, code uint8, orimsg messagetype) {
    err := w.writebyte(code)
    data, err := proto.marshal(orimsg)
    lengthwritten := len(data)
    var b [8]byte
    bs := b[:8]
    binary.littleendian.putuint64(bs, uint64(lengthwritten))
    _, err = w.write(bs)
    _, err = w.write(data)
    w.flush()
}

然后在接收方，以下是我的接收方式。

reader *bufio.Reader
for true {
        if msgType, err = reader.ReadByte(); err != nil {
            panic()
        }
        if msgType == 1 || msgType == 2{
            var b [8]byte
            bs := b[:8]

            _, err := io.ReadFull(reader, bs) 
            numBytes := binary.LittleEndian.Uint64(bs)
            data := make([]byte, numBytes)
            length, err := io.ReadFull(reader, data) 
            msg *MessageType = new(GenericConsensus) // an empty message 
            err = proto.Unmarshal(data[:length], msg) 
            // do something with the message 
            
        } else {
            // unknown message type handler
        }
    }

现在我的问题是，如果发件人在中间中止他的写入怎么办：更具体地说，

1. 情况1：如果发送方写入消息类型字节，然后中止怎么办？在这种情况下，接收方将读取消息类型字节，并等待接收8字节消息长度，但发送方不发送它。

2. 情况 2：这是情况 1 的扩展版本，其中发送方首先仅发送消息类型字节，然后中止发送消息长度和编组消息，然后发送下一条消息：类型字节、长度和编码消息。现在在接收方，一切都会出错，因为消息的顺序（类型、长度和编码消息）被违反。

所以我的问题是，如何修改接收方，使其可以继续运行，尽管发送方违反了预先商定的 type:length:encoded-message 顺序？

谢谢

正确答案

为什么发送者会中止一条消息，然后发送另一条消息？你的意思是它是一个完全拜占庭式的发件人？或者您正在准备模糊测试吗？

如果您的 API 合约规定发送者始终需要发送正确的消息，那么接收者可以简单地忽略错误消息，甚至在发现违反 API 合约的情况下关闭连接。

如果您确实需要它，这里有一些关于如何使其发挥作用的想法：

• 从一个独特的序言开始 - 但随后您必须确保该序言永远不会出现在数据中
• 在将消息发送到解码器之前添加校验和。因此完整的数据包将是：[msg_type : msg_len : msg : chksum ]。这允许接收者检查它是正确的消息还是错误的消息。

此外，就目前的代码而言，发送最大 64 位的大小很容易崩溃。因此，您还应该检查尺寸是否在有用的范围内。我会将其限制为 32 位...

好了，本文到此结束，带大家了解了《接收可写入部分的 Protobuf 编码消息？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！