63 lines
1.8 KiB
Smarty
63 lines
1.8 KiB
Smarty
|
{{- $type := .Name -}}
|
||
|
|
{{- $short := (shortname $type "enumVal" "text" "buf" "ok" "src") -}}
|
||
|
|
{{- $reverseNames := .ReverseConstNames -}}
|
||
|
|
// {{ $type }} is the '{{ .Enum.EnumName }}' enum type from schema '{{ .Schema }}'.
|
||
|
|
type {{ $type }} uint16
|
||
|
|
|
||
|
|
const (
|
||
|
|
{{- range .Values }}
|
||
|
|
// {{ if $reverseNames }}{{ .Name }}{{ $type }}{{ else }}{{ $type }}{{ .Name }}{{ end }} is the '{{ .Val.EnumValue }}' {{ $type }}.
|
||
|
|
{{ if $reverseNames }}{{ .Name }}{{ $type }}{{ else }}{{ $type }}{{ .Name }}{{ end }} = {{ $type }}({{ .Val.ConstValue }})
|
||
|
|
{{ end -}}
|
||
|
|
)
|
||
|
|
|
||
|
|
// String returns the string value of the {{ $type }}.
|
||
|
|
func ({{ $short }} {{ $type }}) String() string {
|
||
|
|
var enumVal string
|
||
|
|
|
||
|
|
switch {{ $short }} {
|
||
|
|
{{- range .Values }}
|
||
|
|
case {{ if $reverseNames }}{{ .Name }}{{ $type }}{{ else }}{{ $type }}{{ .Name }}{{ end }}:
|
||
|
|
enumVal = "{{ .Val.EnumValue }}"
|
||
|
|
{{ end -}}
|
||
|
|
}
|
||
|
|
|
||
|
|
return enumVal
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalText marshals {{ $type }} into text.
|
||
|
|
func ({{ $short }} {{ $type }}) MarshalText() ([]byte, error) {
|
||
|
|
return []byte({{ $short }}.String()), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// UnmarshalText unmarshals {{ $type }} from text.
|
||
|
|
func ({{ $short }} *{{ $type }}) UnmarshalText(text []byte) error {
|
||
|
|
switch string(text) {
|
||
|
|
{{- range .Values }}
|
||
|
|
case "{{ .Val.EnumValue }}":
|
||
|
|
*{{ $short }} = {{ if $reverseNames }}{{ .Name }}{{ $type }}{{ else }}{{ $type }}{{ .Name }}{{ end }}
|
||
|
|
{{ end }}
|
||
|
|
|
||
|
|
default:
|
||
|
|
return errors.New("invalid {{ $type }}")
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Value satisfies the sql/driver.Valuer interface for {{ $type }}.
|
||
|
|
func ({{ $short }} {{ $type }}) Value() (driver.Value, error) {
|
||
|
|
return {{ $short }}.String(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Scan satisfies the database/sql.Scanner interface for {{ $type }}.
|
||
|
|
func ({{ $short }} *{{ $type }}) Scan(src interface{}) error {
|
||
|
|
buf, ok := src.([]byte)
|
||
|
|
if !ok {
|
||
|
|
return errors.New("invalid {{ $type }}")
|
||
|
|
}
|
||
|
|
|
||
|
|
return {{ $short }}.UnmarshalText(buf)
|
||
|
|
}
|
||
|
|
|