增加enumvalue

coderyw · Oct 23, 2024 · 68a8d0d · 68a8d0d
1 parent 0eaceac
commit 68a8d0d
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/protoc-gen-gogo/generator/generator.go b/protoc-gen-gogo/generator/generator.go
@@ -1571,6 +1571,28 @@ func (g *Generator) generateEnum(enum *EnumDescriptor) {
 		}
 		g.Out()
 		g.P("}")
+
+		g.P("var ", ccTypeName, "AliasValue = map[string]int32{")
+		g.In()
+		generatedStr := make(map[string]bool) // avoid duplicate values
+		for _, e := range enum.Value {
+			duplicate := ""
+			str := ""
+
+			if gogoproto.IsEnumAliasCustomName(e) {
+				str = strconv.Quote(gogoproto.GetEnumAliasCustomName(e))
+			} else {
+				str = strconv.Quote(*e.Name)
+			}
+
+			if _, present := generatedStr[str]; present {
+				duplicate = "// Duplicate value: "
+			}
+			g.P(duplicate, str, ": ", e.Number, ",")
+			generatedStr[str] = true
+		}
+		g.Out()
+		g.P("}")
 	}
 
 	if gogoproto.EnabledGoEnumEqual(enum.file.FileDescriptorProto, enum.EnumDescriptorProto) {