Return null if invalid enum is encountered

99designs · Dec 18, 2023 · d916f91 · d916f91
1 parent 7dd971c
commit d916f91
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/plugin/modelgen/models.gotpl b/plugin/modelgen/models.gotpl
@@ -98,7 +98,12 @@
 	}
 
 	func (e {{ goModelName .Name }}) MarshalGQL(w io.Writer) {
-		fmt.Fprint(w, strconv.Quote(e.String()))
+		switch e.IsValid() {
+		case true:
+			fmt.Fprint(w, strconv.Quote(e.String()))
+		case false:
+			graphql.Null.MarshalGQL(w)
+		}
 	}
 
 {{- end }}