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

33
examples/o2mrecur/ent/node_update.go
View File

@@ -21,6 +21,7 @@ import (
type NodeUpdate struct {
config
value *int
addvalue *int
parent map[int]struct{}
children map[int]struct{}
clearedParent bool
@@ -40,6 +41,12 @@ func (nu *NodeUpdate) SetValue(i int) *NodeUpdate {
return nu
}
// AddValue adds i to value.
func (nu *NodeUpdate) AddValue(i int) *NodeUpdate {
nu.addvalue = &i
return nu
}
// SetParentID sets the parent edge to Node by id.
func (nu *NodeUpdate) SetParentID(id int) *NodeUpdate {
if nu.parent == nil {
@@ -170,9 +177,13 @@ func (nu *NodeUpdate) sqlSave(ctx context.Context) (n int, err error) {
res sql.Result
builder = sql.Update(node.Table).Where(sql.InInts(node.FieldID, ids...))
)
if nu.value != nil {
if value := nu.value; value != nil {
update = true
builder.Set(node.FieldValue, *nu.value)
builder.Set(node.FieldValue, *value)
}
if value := nu.addvalue; value != nil {
update = true
builder.Add(node.FieldValue, *value)
}
if update {
query, args := builder.Query()
@@ -247,6 +258,7 @@ type NodeUpdateOne struct {
config
id int
value *int
addvalue *int
parent map[int]struct{}
children map[int]struct{}
clearedParent bool
@@ -259,6 +271,12 @@ func (nuo *NodeUpdateOne) SetValue(i int) *NodeUpdateOne {
return nuo
}
// AddValue adds i to value.
func (nuo *NodeUpdateOne) AddValue(i int) *NodeUpdateOne {
nuo.addvalue = &i
return nuo
}
// SetParentID sets the parent edge to Node by id.
func (nuo *NodeUpdateOne) SetParentID(id int) *NodeUpdateOne {
if nuo.parent == nil {
@@ -392,10 +410,15 @@ func (nuo *NodeUpdateOne) sqlSave(ctx context.Context) (n *Node, err error) {
res sql.Result
builder = sql.Update(node.Table).Where(sql.InInts(node.FieldID, ids...))
)
if nuo.value != nil {
if value := nuo.value; value != nil {
update = true
builder.Set(node.FieldValue, *nuo.value)
n.Value = *nuo.value
builder.Set(node.FieldValue, *value)
n.Value = *value
}
if value := nuo.addvalue; value != nil {
update = true
builder.Add(node.FieldValue, *value)
n.Value += *value
}
if update {
query, args := builder.Query()