我想知道如何动态设置枚举值。
我有以下.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