Files
ent/entc/gen/template/builder/delete.tmpl
Alex Snast dfadb3e27b entc: drop switch statements on single storage (#12)
Summary:
Pull Request resolved: https://github.com/facebookincubator/ent/pull/12

Pull Request resolved: https://github.com/facebookexternal/fbc/pull/1256

Reviewed By: a8m

Differential Revision: D16667212

fbshipit-source-id: e9f8a51986de5fe97356285e23946a61578d8164
2019-08-06 07:52:21 -07:00

72 lines
2.0 KiB
Cheetah

{{ define "delete" }}
{{ $pkg := base $.Config.Package }}
{{ template "header" $ }}
{{ template "import" $ }}
{{ $builder := print (pascal $.Name) "Delete" }}
{{ $receiver := receiver $builder }}
// {{ $builder }} is the builder for deleting a {{ pascal $.Name }} entity.
type {{ $builder }} struct {
config
predicates []predicate.{{ $.Name }}
}
// Where adds a new predicate for the builder.
func ({{ $receiver}} *{{ $builder }}) Where(ps ...predicate.{{ $.Name }}) *{{ $builder }} {
{{ $receiver}}.predicates = append({{ $receiver}}.predicates, ps...)
return {{ $receiver }}
}
// Exec executes the deletion query.
func ({{ $receiver}} *{{ $builder }}) Exec(ctx context.Context) error {
{{- if gt (len $.Storage) 1 -}}
switch {{ $receiver }}.driver.Dialect() {
{{- range $_, $storage := $.Storage }}
case {{ join $storage.Dialects ", " }}:
return {{ $receiver }}.{{ $storage }}Exec(ctx)
{{- end }}
default:
return errors.New("{{ $pkg }}: unsupported dialect")
}
{{- else -}}
return {{ $receiver }}.{{ index $.Storage 0 }}Exec(ctx)
{{- end }}
}
// ExecX is like Exec, but panics if an error occurs.
func ({{ $receiver }} *{{ $builder }}) ExecX(ctx context.Context) {
if err := {{ $receiver }}.Exec(ctx); err != nil {
panic(err)
}
}
{{- range $_, $storage := $.Storage }}
{{ with extend $ "Builder" $builder }}
{{ $tmpl := printf "dialect/%s/delete" $storage }}
{{ xtemplate $tmpl . }}
{{ end }}
{{ end }}
{{ $onebuilder := print $builder "One" }}
{{ $oneReceiver := receiver $onebuilder }}
// {{ $onebuilder }} is the builder for deleting a single {{ $.Name }} entity.
type {{ $onebuilder }} struct {
{{ $receiver }} *{{ $builder }}
}
// Exec executes the deletion query.
func ({{ $oneReceiver }} *{{ $onebuilder }}) Exec(ctx context.Context) error {
return {{ $oneReceiver }}.{{ $receiver }}.Exec(ctx)
}
// ExecX is like Exec, but panics if an error occurs.
func ({{ $oneReceiver }} *{{ $onebuilder }}) ExecX(ctx context.Context) {
{{ $oneReceiver }}.{{ $receiver }}.ExecX(ctx)
}
{{ end }}