Add ordinal function to enums

volatiletech · Sep 12, 2023 · dd364f2 · dd364f2
1 parent cc34619
commit dd364f2
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/templates/main/singleton/boil_types.go.tpl b/templates/main/singleton/boil_types.go.tpl
@@ -138,6 +138,18 @@ It only titlecases the EnumValue portion if it's snake-cased.
 						func (e {{$enumName}}) String() string {
 							return string(e)
 						}
+
+						func (e {{$enumName}}) Ordinal() int {
+							switch e {
+							{{range $idx, $val := $vals -}}
+								{{- $enumValue := titleCase $val -}}
+								case {{$enumName}}{{$enumValue}}:
+									return {{$idx}}
+							{{end}}
+							default:
+								panic(errors.New("enum is not valid"))
+							}
+						}
 					{{- end -}}
 
 					{{ if and