add list of enums

99designs · Jan 3, 2019 · 354c321 · 354c321
1 parent c033f5f
commit 354c321
Show file tree

Hide file tree

Showing 6 changed files with 42 additions and 1 deletion.
diff --git a/codegen/templates/data.go b/codegen/templates/data.go
diff --git a/codegen/templates/models.gotpl b/codegen/templates/models.gotpl
@@ -59,6 +59,12 @@ import (
 	{{- end }}
 	)
 
+	var All{{.GoType}} = []{{.GoType}}{
+	{{- range $value := .Values}}
+		{{$enum.GoType}}{{ .Name|toCamel }},
+	{{- end }}
+	}
+
 	func (e {{.GoType}}) IsValid() bool {
 		switch e {
 		case {{ range $index, $element := .Values}}{{if $index}},{{end}}{{ $enum.GoType }}{{ $element.Name|toCamel }}{{end}}:

diff --git a/example/starwars/models_gen.go b/example/starwars/models_gen.go