mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
72 lines
2.3 KiB
Cheetah
72 lines
2.3 KiB
Cheetah
{{/*
|
|
Copyright 2019-present Facebook Inc. All rights reserved.
|
|
This source code is licensed under the Apache 2.0 license found
|
|
in the LICENSE file in the root directory of this source tree.
|
|
*/}}
|
|
|
|
{{/* gotype: entgo.io/ent/entc/gen.typeScope */}}
|
|
|
|
{{ define "dialect/sql/delete" }}
|
|
{{ $builder := pascal $.Scope.Builder }}
|
|
{{ $receiver := $.Scope.Receiver }}
|
|
{{ $mutation := print $receiver ".mutation" }}
|
|
|
|
func ({{ $receiver}} *{{ $builder }}) sqlExec(ctx context.Context) (int, error) {
|
|
_spec := sqlgraph.NewDeleteSpec({{ $.Package }}.Table, {{ if $.HasOneFieldID }}sqlgraph.NewFieldSpec({{ $.Package }}.{{ $.ID.Constant }}, field.{{ $.ID.Type.ConstName }}){{ else }}nil{{ end }})
|
|
{{- /* Allow mutating the sqlgraph.DeleteSpec by ent extensions or user templates.*/}}
|
|
{{- with $tmpls := matchTemplate "dialect/sql/delete/spec/*" }}
|
|
{{- range $tmpl := $tmpls }}
|
|
{{- xtemplate $tmpl $ }}
|
|
{{- end }}
|
|
{{- end }}
|
|
if ps := {{ $mutation }}.Predicates(); len(ps) > 0 {
|
|
_spec.Predicate = func(selector *sql.Selector) {
|
|
for i := range ps {
|
|
ps[i](selector)
|
|
}
|
|
}
|
|
}
|
|
{{- if $.HasBlobFields }}
|
|
// Collect blob keys before deleting rows so we can remove blobs from storage afterward.
|
|
_blobCleanup, _blobErr := ent.NewBlobs({{ $mutation }}.blobOpeners.{{ $.Name }}).Delete(ctx, &sqlgraph.BlobSpec{
|
|
Driver: {{ $receiver }}.driver,
|
|
Predicate: _spec.Predicate,
|
|
Table: {{ $.Package }}.Table,
|
|
Columns: map[string]string{
|
|
{{- range $f := $.BlobFields }}
|
|
{{ $.Package }}.{{ $f.Constant }}: "{{ $f.BlobKeyColumn }}",
|
|
{{- end }}
|
|
},
|
|
})
|
|
if _blobErr != nil {
|
|
return 0, _blobErr
|
|
}
|
|
{{- end }}
|
|
affected, err := sqlgraph.DeleteNodes(ctx, {{ $receiver}}.driver, _spec)
|
|
if err != nil && sqlgraph.IsConstraintError(err) {
|
|
err = &ConstraintError{msg: err.Error(), wrap: err}
|
|
}
|
|
{{- if $.HasBlobFields }}
|
|
if err == nil {
|
|
if txd, ok := {{ $receiver }}.driver.(*txDriver); ok {
|
|
txd.mu.Lock()
|
|
txd.onCommit = append(txd.onCommit, func(next Committer) Committer {
|
|
return CommitFunc(func(ctx context.Context, tx *Tx) error {
|
|
if err := next.Commit(ctx, tx); err != nil {
|
|
return err
|
|
}
|
|
return _blobCleanup(ctx)
|
|
})
|
|
})
|
|
txd.mu.Unlock()
|
|
} else {
|
|
err = _blobCleanup(ctx)
|
|
}
|
|
}
|
|
{{- end }}
|
|
{{ $mutation }}.done = true
|
|
return affected, err
|
|
}
|
|
|
|
{{ end }}
|