diff --git a/entc/gen/template/meta.tmpl b/entc/gen/template/meta.tmpl index 9a9c2be23..38b7146ab 100644 --- a/entc/gen/template/meta.tmpl +++ b/entc/gen/template/meta.tmpl @@ -124,13 +124,14 @@ var ( } {{ $name := $f.Validator -}} - // {{ $name }} is a validator for the "{{ $f.Name }}" field enum values. It is called by the builders before save. - func {{ $name }}({{ $f.Name }} {{ $enum }}) error { - switch {{ $f.Name }} { + {{ $arg := receiver $f.BuilderField }} + // {{ $name }} is a validator for the "{{ $arg }}" field enum values. It is called by the builders before save. + func {{ $name }}({{ $arg }} {{ $enum }}) error { + switch {{ $arg }} { case {{ range $i, $e := $f.Enums }}{{ if ne $i 0 }},{{ end }}{{ $f.StructField }}{{ pascal $e }}{{ end }}: return nil default: - return fmt.Errorf("{{ $.Package }}: invalid enum value for {{ $f.Name }} field: %q", {{ $f.Name }}) + return fmt.Errorf("{{ $.Package }}: invalid enum value for {{ $f.Name }} field: %q", {{ $arg }}) } } {{ end }}