golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-28 09:49:08 +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

9
entc/gen/template/dialect/gremlin/delete.tmpl
View File

@@ -8,10 +8,13 @@ in the LICENSE file in the root directory of this source tree.
{{ $builder := pascal $.Scope.Builder }}
{{ $receiver := receiver $builder }}
func ({{ $receiver}} *{{ $builder }}) gremlinExec(ctx context.Context) error {
func ({{ $receiver}} *{{ $builder }}) gremlinExec(ctx context.Context) (int, error) {
res := &gremlin.Response{}
query, bindings := {{ $receiver }}.gremlin().Query()
return {{ $receiver }}.driver.Exec(ctx, query, bindings, res)
if err := {{ $receiver }}.driver.Exec(ctx, query, bindings, res); err != nil {
return 0, err
}
return res.ReadInt()
}
@@ -20,6 +23,6 @@ func ({{ $receiver }} *{{ $builder }}) gremlin() *dsl.Traversal {
for _, p := range {{ $receiver }}.predicates {
p(t)
}
return t.Drop()
return t.SideEffect(__.Drop()).Count()
}
{{ end }}

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 }}