用于动态枚举访问的Protobabuffer API



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

我有以下.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

我想动态地生成我的花园,并且只有价值";玫瑰";。有值映射,但以下不起作用:

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)

此外,我想设置";。花"动态地。我知道如何在结构中动态设置字段,但不知道如何将protoreflect.Value强制转换为reflect.Value类型。

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

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())
}

当改变字符串";玫瑰"对于任何其他有效的Flowers,枚举将自动更新。此外,请确保字段名是.proto文件中指定的字段名。在本例中为flower

您已经尝试过这种方法吗?

flower := "Rose"
garden := pb.Garden{}
garden.Flower = pb.Garden_Flower(pb.Garden_Flower_value[flower])
fmt.Println(garden.Flower.String()) // Rose

最新更新