add mode unmarshaller

Recidiviz · Apr 11, 2024 · a813184 · a813184
1 parent 2afed39
commit a813184
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/types/types.go b/types/types.go
@@ -79,6 +79,18 @@ func (m *Mode) UnmarshalYAML(b []byte) error {
 	return nil
 }
 
+func (m *Mode) UnmarshalJSON(b []byte) error {
+	switch strings.ToLower(strings.Trim(string(b), `"`)) {
+	case strings.ToLower(string(NullableMode)):
+		*m = NullableMode
+	case strings.ToLower(string(RequiredMode)):
+		*m = RequiredMode
+	case strings.ToLower(string(RepeatedMode)):
+		*m = RepeatedMode
+	}
+	return nil
+}
+
 const (
 	NullableMode Mode = "NULLABLE"
 	RequiredMode Mode = "REQUIRED"