golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/gen: returns affected rows in delete

Browse Source 
Summary: Pull Request resolved: https://github.com/facebookexternal/fbc/pull/1455

Reviewed By: alexsn

Differential Revision: D17396880

fbshipit-source-id: 3a3a9849b816777bd0ef94cf55b2291705a37df6
This commit is contained in:
Ariel Mashraki
2019-09-16 08:45:45 -07:00
committed by Facebook Github Bot
parent 2b1ff377b5
commit 83d0063437
82 changed files with 1566 additions and 712 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
entc/gen/template/dialect/sql/delete.tmpl
View File

@@ -8,14 +8,21 @@ in the LICENSE file in the root directory of this source tree.
{{ $builder := pascal $.Scope.Builder }}
{{ $receiver := receiver $builder }}
func ({{ $receiver}} *{{ $builder }}) sqlExec(ctx context.Context) error {
func ({{ $receiver}} *{{ $builder }}) sqlExec(ctx context.Context) (int, error) {
var res sql.Result
selector := sql.Select().From(sql.Table({{ $.Package }}.Table))
for _, p := range {{ $receiver }}.predicates {
p(selector)
}
query, args := sql.Delete({{ $.Package }}.Table).FromSelect(selector).Query()
return {{ $receiver }}.driver.Exec(ctx, query, args, &res)
if err := {{ $receiver }}.driver.Exec(ctx, query, args, &res); err != nil {
return 0, err
}
affected, err := res.RowsAffected()
if err != nil {
return 0, err
}
return int(affected), nil
}
{{ end }}