Protobuf API 用于动态枚举访问

Question

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

我有以下 .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”。我想出了如何在结构中动态设置字段，但不知道如何将 protoreflect.Value 转换为 reflect.Value 类型。

Answer 1

你试过这种方法了吗？

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

Answer 2

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

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.

Protobuf API 用于动态枚举访问

Protobuffer API for dynamic Enum Access

enums

go

protocol-buffers

protobuf-go