internal/codegen: add Enum.Valid and AllEnumValues

This commit adds a Valid method for enum types and an All...Values function to get all enum values. This makes it easier to work with enums. Fixes #1607
sqlc-dev · May 11, 2022 · 3dcc55e · 3dcc55e
1 parent 52b6bb4
commit 3dcc55e
Show file tree

Hide file tree

Showing 25 changed files with 336 additions and 0 deletions.
diff --git a/examples/batch/postgresql/models.go b/examples/batch/postgresql/models.go
diff --git a/examples/booktest/mysql/models.go b/examples/booktest/mysql/models.go
diff --git a/examples/booktest/postgresql/models.go b/examples/booktest/postgresql/models.go
diff --git a/examples/ondeck/mysql/models.go b/examples/ondeck/mysql/models.go
diff --git a/examples/ondeck/postgresql/models.go b/examples/ondeck/postgresql/models.go
diff --git a/internal/codegen/golang/templates/template.tmpl b/internal/codegen/golang/templates/template.tmpl
@@ -86,6 +86,18 @@ func (e *{{.Name}}) Scan(src interface{}) error {
 	}
 	return nil
 }
+
+func (e {{.Name}}) Valid() bool {
+  switch e {
+  case {{ range $idx, $name := .Constants}}{{ if gt $idx 0 }},{{ end }}{{.Name}}{{ end }}:
+    return true
+  }
+  return false
+}
+
+func All{{ .Name }}Values() []{{ .Name }} {
+	return []{{ .Name }}{ {{ range .Constants}}{{ .Name }}, {{ end }} }
+}
 {{end}}
 
 {{range .Structs}}

diff --git a/internal/endtoend/testdata/comment_on/postgresql/pgx/go/models.go b/internal/endtoend/testdata/comment_on/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/comment_on/postgresql/stdlib/go/models.go b/internal/endtoend/testdata/comment_on/postgresql/stdlib/go/models.go
diff --git a/internal/endtoend/testdata/ddl_alter_type_add_value/postgresql/pgx/go/models.go b/internal/endtoend/testdata/ddl_alter_type_add_value/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/ddl_alter_type_add_value/postgresql/stdlib/go/models.go b/internal/endtoend/testdata/ddl_alter_type_add_value/postgresql/stdlib/go/models.go
diff --git a/internal/endtoend/testdata/ddl_alter_type_rename/postgresql/pgx/go/models.go b/internal/endtoend/testdata/ddl_alter_type_rename/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/ddl_alter_type_rename/postgresql/stdlib/go/models.go b/internal/endtoend/testdata/ddl_alter_type_rename/postgresql/stdlib/go/models.go
diff --git a/...al/endtoend/testdata/ddl_alter_type_rename_and_update_columns/postgresql/pgx/go/models.go b/...al/endtoend/testdata/ddl_alter_type_rename_and_update_columns/postgresql/pgx/go/models.go
diff --git a/...endtoend/testdata/ddl_alter_type_rename_and_update_columns/postgresql/stdlib/go/models.go b/...endtoend/testdata/ddl_alter_type_rename_and_update_columns/postgresql/stdlib/go/models.go
diff --git a/internal/endtoend/testdata/ddl_alter_type_rename_value/postgresql/pgx/go/models.go b/internal/endtoend/testdata/ddl_alter_type_rename_value/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/ddl_alter_type_rename_value/postgresql/stdlib/go/models.go b/internal/endtoend/testdata/ddl_alter_type_rename_value/postgresql/stdlib/go/models.go
diff --git a/internal/endtoend/testdata/ddl_comment/postgresql/pgx/go/models.go b/internal/endtoend/testdata/ddl_comment/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/ddl_comment/postgresql/stdlib/go/models.go b/internal/endtoend/testdata/ddl_comment/postgresql/stdlib/go/models.go
diff --git a/internal/endtoend/testdata/ddl_create_enum/mysql/go/models.go b/internal/endtoend/testdata/ddl_create_enum/mysql/go/models.go