首页 > 后端开发 > Golang > 如何在 Go 中解组未知的 Protobuf 消息?

如何在 Go 中解组未知的 Protobuf 消息?

Linda Hamilton
发布: 2024-12-03 02:15:10
原创
728 人浏览过

How to Unmarshal Unknown Protobuf Messages in Go?

在 Go 中解组未知的 Protobuf 消息

将 protobuf 消息解组到接口{}以促进类型转换是不可行的,因为 proto.Unmarshal 函数需要一个类型的参数proto.Message,具体protobuf类型的接口

已知元数据的替代方法

如果至少有一些上下文信息(例如字符串或数字)伴随字节切片,您可以使用该信息来映射到预期的protobuf 类型并在将其传递给 proto.Unmarshal 之前实例化它:

func decodeWithMetadata(data []byte, identifier string) error {
    var message proto.Message
    switch identifier {
        case "foo":
            message = &mypb.Foo{}
        case "bar":
            message = &mypb.Bar{}
    }
    if err := proto.Unmarshal(data, message); err != nil {
        return err
    }
    log.Printf("%v\n", data)
    return nil
}
登录后复制

解析完全未知的有效负载

但是,如果字节有效负载完全未知,请考虑 protowire 包,它允许从有线格式的 protobuf 消息中提取有限的信息。请记住,protobuf 消息的线路表示不明确,导致语义较弱。

解析器实现

以下代码解析未知的 protobuf 消息:

type Field struct {
    Tag Tag
    Val Val
}

type Tag struct {
    Num int32
    Type protowire.Type
}

type Val struct {
    Payload interface{}
    Length int
}

func parseUnknown(b []byte) []Field {
    // ... implementation to parse unknown protobuf messages as described in the provided answer ...
}
登录后复制

示例用法

给定一个 protobuf 模式像:

message Foo {
  string a = 1;
  string b = 2;
  Bar bar = 3;
}

message Bar {
  string c = 1;
}
登录后复制

以下代码演示了如何使用解析器:

// Parse the protobuf message as byte slice
b := []byte{0x0a, 0x01, 0x41, 0x12, 0x01, 0x42, 0x1a, 0x03, 0x43}
fields := parseUnknown(b)
for _, field := range fields {
    fmt.Printf("%#v\n", field)
}
登录后复制

输出:

main.Field{Tag:main.Tag{Num:1, Type:2}, Val:main.Val{Payload:[]uint8{0x41}, Length:1}}
main.Field{Tag:main.Tag{Num:2, Type:2}, Val:main.Val{Payload:[]uint8{0x42}, Length:1}}
main.Field{Tag:main.Tag{Num:1, Type:2}, Val:main.Val{Payload:[]uint8{0x43}, Length:1}}
main.Field{Tag:main.Tag{Num:3, Type:2}, Val:main.Val{Payload:[]main.Field{main.Field{Tag:main.Tag{Num:1, Type:2}, Val:main.Val{Payload:[]uint8{0x43}, Length:1}}}, Length:3}}
登录后复制

以上是如何在 Go 中解组未知的 Protobuf 消息?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板