Protobuffer API для динамического доступа к перечислениям

Я хочу знать, как динамически устанавливать значение Enum.

У меня есть следующие .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)

Кроме того, я хочу установить ".Flower" динамически. Я понял, как динамически устанавливать поля в структуре, но не знаю, как преобразовать protoreflect.Value к reflect.Value тип.

Источник

2 ответа

Решение

Вы уже пробовали этот подход?

flower := "Rose"
garden := pb.Garden{}
garden.Flower = pb.Garden_Flower(pb.Garden_Flower_value[flower])
fmt.Println(garden.Flower.String()) // Rose
Источник

Если кто-то хочет знать, как все это делать динамически. Я наконец-то понял:

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

При изменении строки "Роза" на любой другой допустимый цвет, перечисление автоматически обновляется. Кроме того, убедитесь, что имя поля совпадает с тем, которое указано в .protoфайл. В этом примере это было flower.

Источник
Другие вопросы по тегам