entc: improve numeric addition api

Reviewed By: alexsn

Differential Revision: D17474547

fbshipit-source-id: 08e367793faac76eb0cece880259a49a8bcbc95b
This commit is contained in:
Ariel Mashraki
2019-09-19 04:58:21 -07:00
committed by Facebook Github Bot
parent bc5e8316dd
commit bf913111fd
14 changed files with 351 additions and 59 deletions

View File

@@ -45,6 +45,7 @@ func (nu *NodeUpdate) Where(ps ...predicate.Node) *NodeUpdate {
// SetValue sets the value field.
func (nu *NodeUpdate) SetValue(i int) *NodeUpdate {
nu.value = &i
nu.addvalue = nil
return nu
}
@@ -58,7 +59,11 @@ func (nu *NodeUpdate) SetNillableValue(i *int) *NodeUpdate {
// AddValue adds i to value.
func (nu *NodeUpdate) AddValue(i int) *NodeUpdate {
nu.addvalue = &i
if nu.addvalue == nil {
nu.addvalue = &i
} else {
*nu.addvalue += i
}
return nu
}
@@ -373,6 +378,7 @@ type NodeUpdateOne struct {
// SetValue sets the value field.
func (nuo *NodeUpdateOne) SetValue(i int) *NodeUpdateOne {
nuo.value = &i
nuo.addvalue = nil
return nuo
}
@@ -386,7 +392,11 @@ func (nuo *NodeUpdateOne) SetNillableValue(i *int) *NodeUpdateOne {
// AddValue adds i to value.
func (nuo *NodeUpdateOne) AddValue(i int) *NodeUpdateOne {
nuo.addvalue = &i
if nuo.addvalue == nil {
nuo.addvalue = &i
} else {
*nuo.addvalue += i
}
return nuo
}