Go中的protobuf数组元素的范围

Question

我在.proto文件中定义了一个Protobuf结构：

message Msg{
  message SubMsg {
    string SubVariable1 = 1;
    int32 SubVariable2 = 2;
    ...
  }
  string Variable1 = 1;
  repeated SubMsg Variable2 = 2;
  ...
}

在使用来自JSON API的数据时，我使用https://godoc.org/google.golang.org/protobuf/encoding/protojson包将数据拉入到此结构中，如下所示：

Response, err := Client.Do(Request)
if err != nil {
    log.Error(err)
}
DataByte, err := ioutil.ReadAll(Response.Body)
if err != nil {
    log.Error(err)
}
DataProto := Msg{}
err = protojson.Unmarshal(DataByte, &DataProto)
if err != nil {
    log.Error(err)
}

我想要做的是遍历Variable2的元素，以便能够使用protoreflect API访问SubVariables，我已经尝试了这两种方法：

Array := DataProto.GetVariable2()
for i := range Array {
  Element := Array[i]
}

还有：

DataProto.GetVariable2().ProtoReflect().Range(func(fd protoreflect.FieldDescriptor, v protoreflect.Value) {
      …
      return true})

第一个命令失败，并显示错误消息：

cannot range over DataProto.GetVariable2() (type *SubMsg)

尽管如此，DataProto.GetVariable2()返回一个[]*Msg_SubMsg类型的变量。

第二个失败，出现以下错误：

DataProto.GetVariable2.ProtoReflect undefined (type []*SubMsg has no field or method ProtoReflect)

这表明DataProto.GetVariable2()确实返回了一个数组，而不是我第一种方法返回的错误中所建议的数组。这对我来说很有意义，因为protoreflect只允许在定义的消息上调用此方法，而不是那些消息的数组。因此，必须有另一种方法来访问这些数组的元素才能使用protoreflect (到目前为止，我还没能在网上找到并回答这个问题)。

有人能帮我弄明白这些看似矛盾的错误消息吗？有没有人自己成功地迭代过Protobuf数组？

提前谢谢。

Answer 1

您需要将Array变量视为List，这意味着您不能像在第二次尝试时那样使用Range()。虽然很接近。下面是一个遍历和检查嵌套消息的函数示例：

import (
    "testing"

    "google.golang.org/protobuf/reflect/protoreflect"
)

func TestVariable2(t *testing.T) {
    pb := &Msg{
        Variable2: []*Msg_SubMsg{
            {
                SubVariable1: "string",
                SubVariable2: 1,
            },
        },
    }

    pbreflect := pb.ProtoReflect()
    fd := pbreflect.Descriptor().Fields().ByJSONName("Variable2")
    if !fd.IsList() {
        t.Fatal("expected a list")
    }
    l := pbreflect.Get(fd).List()
    for i := 0; i < l.Len(); i++ {
        // should test that we are now inspecting a message type
        li := l.Get(i).Message()
        li.Range(func(lifd protoreflect.FieldDescriptor, liv protoreflect.Value) bool {
            t.Logf("%v: %v", lifd.Name(), liv)
            return true
        })
    }
}

如果想要查看输出，请使用go test -v ./...运行