SEBA-667 Implement model filtering
Change-Id: Idfb15da5092e33833e7930dc2325dab2d6d71025
diff --git a/commands/funcmap.go b/commands/funcmap.go
index 0342653..7a2bdf2 100644
--- a/commands/funcmap.go
+++ b/commands/funcmap.go
@@ -69,7 +69,12 @@
return desc, name, nil
}
-func GetEnumValue(val *dynamic.Message, name string) string {
- return val.FindFieldDescriptorByName(name).GetEnumType().
- FindValueByNumber(val.GetFieldByName(name).(int32)).GetName()
+func GetEnumValue(msg *dynamic.Message, name string) string {
+ return msg.FindFieldDescriptorByName(name).GetEnumType().
+ FindValueByNumber(msg.GetFieldByName(name).(int32)).GetName()
+}
+
+func SetEnumValue(msg *dynamic.Message, name string, value string) {
+ eValue := msg.FindFieldDescriptorByName(name).GetEnumType().FindValueByName(value)
+ msg.SetFieldByName(name, eValue.GetNumber())
}