1

我想知道如何动态设置枚举值。

我有以下.proto文件:

syntax = "proto3";
package garden;

option go_package = "proto/garden";

message Garden {
  enum Flower {
    Rose = 0;
    Acacia = 1;
    Firethorn = 2;
    Nemophila = 3;
  }
  Flower flower = 1;
}

设置枚举的一种方法如下:

garden := pb.Garden{}
garden.Flower = pb.Garden_Rose

我想动态生成我的花园,并且只有值“Rose”。有值映射,但以下不起作用:

garden := pb.Garden{}
garden.Flower = pb.Garden_Flower_value["Rose"] // gives error: cannot use garden.Garden_Flower_value[flower] (type int32) as type garden.Garden_Flower in assignment

garden.Flower = 0 // works somehow??

我假设我可以使用 protoreflect 包来设置值。不幸的是,我还不清楚它是如何工作的。

garden.Flower = protoreflect.EnumNumber(pb.Garden_Rose)

此外,我想动态设置“.Flower”。我想出了如何动态设置结构中的字段,但缺乏如何将 aprotoreflect.Value转换为reflect.Value类型的知识。

4

2 回答 2

1

您是否已经尝试过这种方法?

flower := "Rose"
garden := pb.Garden{}
garden.Flower = pb.Garden_Flower(pb.Garden_Flower_value[flower])
fmt.Println(garden.Flower.String()) // Rose
于 2020-10-21T10:24:36.090 回答
0

如果有人想知道如何动态地完成所有这些。我终于想通了:

func main() {
    var garden = pb.Garden{}
    var gardenProto = garden.ProtoReflect()
    var fields = gardenProto.Descriptor().Fields()
    var flower = fields.ByName(protoreflect.Name("flower"))
    var rose = protoreflect.Name("Rose")
    var enumValue = protoreflect.ValueOfEnum(protoreflect.EnumNumber(flower.Enum().Values().ByName(rose).Number()))
    gardenProto.Set(flower, enumValue)
    fmt.Print(garden.GetFlower())
}

当将字符串“Rose”更改为任何其他有效的 Flowers 时,枚举会自动更新。此外,请确保字段名称是.proto文件中指定的名称。在这个例子中是flower.

于 2020-10-21T16:02:17.787 回答