Merge pull request #1305 from EmiPhil/add_ordinal_to_enum

Add ordinal function to enums
volatiletech · Sep 13, 2023 · 59e63d5 · 59e63d5
2 parents cc34619 + dd364f2
commit 59e63d5
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