entc/gen: returns affected rows in delete

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

Reviewed By: alexsn

Differential Revision: D17396880

fbshipit-source-id: 3a3a9849b816777bd0ef94cf55b2291705a37df6

examples/m2m2types/ent/client.go

@@ -130,6 +130,20 @@ func (c *GroupClient) Query() *GroupQuery {
 	return &GroupQuery{config: c.config}
 }
 
+// Get returns a Group entity by its id.
+func (c *GroupClient) Get(ctx context.Context, id int) (*Group, error) {
+	return c.Query().Where(group.ID(id)).Only(ctx)
+}
+
+// GetX is like Get, but panics if an error occurs.
+func (c *GroupClient) GetX(ctx context.Context, id int) *Group {
+	gr, err := c.Get(ctx, id)
+	if err != nil {
+		panic(err)
+	}
+	return gr
+}
+
 // QueryUsers queries the users edge of a Group.
 func (c *GroupClient) QueryUsers(gr *Group) *UserQuery {
 	query := &UserQuery{config: c.config}
@@ -200,6 +214,20 @@ func (c *UserClient) Query() *UserQuery {
 	return &UserQuery{config: c.config}
 }
 
+// Get returns a User entity by its id.
+func (c *UserClient) Get(ctx context.Context, id int) (*User, error) {
+	return c.Query().Where(user.ID(id)).Only(ctx)
+}
+
+// GetX is like Get, but panics if an error occurs.
+func (c *UserClient) GetX(ctx context.Context, id int) *User {
+	u, err := c.Get(ctx, id)
+	if err != nil {
+		panic(err)
+	}
+	return u
+}
+
 // QueryGroups queries the groups edge of a User.
 func (c *UserClient) QueryGroups(u *User) *GroupQuery {
 	query := &GroupQuery{config: c.config}

examples/m2m2types/ent/group_delete.go

@@ -21,32 +21,41 @@ type GroupDelete struct {
 	predicates []predicate.Group
 }
 
-// Where adds a new predicate for the builder.
+// Where adds a new predicate to the delete builder.
 func (gd *GroupDelete) Where(ps ...predicate.Group) *GroupDelete {
 	gd.predicates = append(gd.predicates, ps...)
 	return gd
 }
 
-// Exec executes the deletion query.
-func (gd *GroupDelete) Exec(ctx context.Context) error {
+// Exec executes the deletion query and returns how many vertices were deleted.
+func (gd *GroupDelete) Exec(ctx context.Context) (int, error) {
 	return gd.sqlExec(ctx)
 }
 
 // ExecX is like Exec, but panics if an error occurs.
-func (gd *GroupDelete) ExecX(ctx context.Context) {
-	if err := gd.Exec(ctx); err != nil {
+func (gd *GroupDelete) ExecX(ctx context.Context) int {
+	n, err := gd.Exec(ctx)
+	if err != nil {
 		panic(err)
 	}
+	return n
 }
 
-func (gd *GroupDelete) sqlExec(ctx context.Context) error {
+func (gd *GroupDelete) sqlExec(ctx context.Context) (int, error) {
 	var res sql.Result
 	selector := sql.Select().From(sql.Table(group.Table))
 	for _, p := range gd.predicates {
 		p(selector)
 	}
 	query, args := sql.Delete(group.Table).FromSelect(selector).Query()
-	return gd.driver.Exec(ctx, query, args, &res)
+	if err := gd.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
 }
 
 // GroupDeleteOne is the builder for deleting a single Group entity.
@@ -56,7 +65,15 @@ type GroupDeleteOne struct {
 
 // Exec executes the deletion query.
 func (gdo *GroupDeleteOne) Exec(ctx context.Context) error {
-	return gdo.gd.Exec(ctx)
+	n, err := gdo.gd.Exec(ctx)
+	switch {
+	case err != nil:
+		return err
+	case n == 0:
+		return &ErrNotFound{group.Label}
+	default:
+		return nil
+	}
 }
 
 // ExecX is like Exec, but panics if an error occurs.

examples/m2m2types/ent/group_query.go

@@ -73,20 +73,6 @@ func (gq *GroupQuery) QueryUsers() *UserQuery {
 	return query
 }
 
-// Get returns a Group entity by its id.
-func (gq *GroupQuery) Get(ctx context.Context, id int) (*Group, error) {
-	return gq.Where(group.ID(id)).Only(ctx)
-}
-
-// GetX is like Get, but panics if an error occurs.
-func (gq *GroupQuery) GetX(ctx context.Context, id int) *Group {
-	gr, err := gq.Get(ctx, id)
-	if err != nil {
-		panic(err)
-	}
-	return gr
-}
-
 // First returns the first Group entity in the query. Returns *ErrNotFound when no group was found.
 func (gq *GroupQuery) First(ctx context.Context) (*Group, error) {
 	grs, err := gq.Limit(1).All(ctx)

examples/m2m2types/ent/group_update.go

@@ -136,9 +136,9 @@ func (gu *GroupUpdate) sqlSave(ctx context.Context) (n int, err error) {
 		res     sql.Result
 		builder = sql.Update(group.Table).Where(sql.InInts(group.FieldID, ids...))
 	)
-	if gu.name != nil {
+	if value := gu.name; value != nil {
 		update = true
-		builder.Set(group.FieldName, *gu.name)
+		builder.Set(group.FieldName, *value)
 	}
 	if update {
 		query, args := builder.Query()
@@ -299,10 +299,10 @@ func (guo *GroupUpdateOne) sqlSave(ctx context.Context) (gr *Group, err error) {
 		res     sql.Result
 		builder = sql.Update(group.Table).Where(sql.InInts(group.FieldID, ids...))
 	)
-	if guo.name != nil {
+	if value := guo.name; value != nil {
 		update = true
-		builder.Set(group.FieldName, *guo.name)
-		gr.Name = *guo.name
+		builder.Set(group.FieldName, *value)
+		gr.Name = *value
 	}
 	if update {
 		query, args := builder.Query()

examples/m2m2types/ent/user_delete.go

@@ -21,32 +21,41 @@ type UserDelete struct {
 	predicates []predicate.User
 }
 
-// Where adds a new predicate for the builder.
+// Where adds a new predicate to the delete builder.
 func (ud *UserDelete) Where(ps ...predicate.User) *UserDelete {
 	ud.predicates = append(ud.predicates, ps...)
 	return ud
 }
 
-// Exec executes the deletion query.
-func (ud *UserDelete) Exec(ctx context.Context) error {
+// Exec executes the deletion query and returns how many vertices were deleted.
+func (ud *UserDelete) Exec(ctx context.Context) (int, error) {
 	return ud.sqlExec(ctx)
 }
 
 // ExecX is like Exec, but panics if an error occurs.
-func (ud *UserDelete) ExecX(ctx context.Context) {
-	if err := ud.Exec(ctx); err != nil {
+func (ud *UserDelete) ExecX(ctx context.Context) int {
+	n, err := ud.Exec(ctx)
+	if err != nil {
 		panic(err)
 	}
+	return n
 }
 
-func (ud *UserDelete) sqlExec(ctx context.Context) error {
+func (ud *UserDelete) sqlExec(ctx context.Context) (int, error) {
 	var res sql.Result
 	selector := sql.Select().From(sql.Table(user.Table))
 	for _, p := range ud.predicates {
 		p(selector)
 	}
 	query, args := sql.Delete(user.Table).FromSelect(selector).Query()
-	return ud.driver.Exec(ctx, query, args, &res)
+	if err := ud.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
 }
 
 // UserDeleteOne is the builder for deleting a single User entity.
@@ -56,7 +65,15 @@ type UserDeleteOne struct {
 
 // Exec executes the deletion query.
 func (udo *UserDeleteOne) Exec(ctx context.Context) error {
-	return udo.ud.Exec(ctx)
+	n, err := udo.ud.Exec(ctx)
+	switch {
+	case err != nil:
+		return err
+	case n == 0:
+		return &ErrNotFound{user.Label}
+	default:
+		return nil
+	}
 }
 
 // ExecX is like Exec, but panics if an error occurs.

examples/m2m2types/ent/user_query.go

@@ -73,20 +73,6 @@ func (uq *UserQuery) QueryGroups() *GroupQuery {
 	return query
 }
 
-// Get returns a User entity by its id.
-func (uq *UserQuery) Get(ctx context.Context, id int) (*User, error) {
-	return uq.Where(user.ID(id)).Only(ctx)
-}
-
-// GetX is like Get, but panics if an error occurs.
-func (uq *UserQuery) GetX(ctx context.Context, id int) *User {
-	u, err := uq.Get(ctx, id)
-	if err != nil {
-		panic(err)
-	}
-	return u
-}
-
 // First returns the first User entity in the query. Returns *ErrNotFound when no user was found.
 func (uq *UserQuery) First(ctx context.Context) (*User, error) {
 	us, err := uq.Limit(1).All(ctx)

examples/m2m2types/ent/user_update.go

@@ -20,6 +20,7 @@ import (
 type UserUpdate struct {
 	config
 	age           *int
+	addage        *int
 	name          *string
 	groups        map[int]struct{}
 	removedGroups map[int]struct{}
@@ -38,6 +39,12 @@ func (uu *UserUpdate) SetAge(i int) *UserUpdate {
 	return uu
 }
 
+// AddAge adds i to age.
+func (uu *UserUpdate) AddAge(i int) *UserUpdate {
+	uu.addage = &i
+	return uu
+}
+
 // SetName sets the name field.
 func (uu *UserUpdate) SetName(s string) *UserUpdate {
 	uu.name = &s
@@ -143,13 +150,17 @@ func (uu *UserUpdate) sqlSave(ctx context.Context) (n int, err error) {
 		res     sql.Result
 		builder = sql.Update(user.Table).Where(sql.InInts(user.FieldID, ids...))
 	)
-	if uu.age != nil {
+	if value := uu.age; value != nil {
 		update = true
-		builder.Set(user.FieldAge, *uu.age)
+		builder.Set(user.FieldAge, *value)
 	}
-	if uu.name != nil {
+	if value := uu.addage; value != nil {
 		update = true
-		builder.Set(user.FieldName, *uu.name)
+		builder.Add(user.FieldAge, *value)
+	}
+	if value := uu.name; value != nil {
+		update = true
+		builder.Set(user.FieldName, *value)
 	}
 	if update {
 		query, args := builder.Query()
@@ -198,6 +209,7 @@ type UserUpdateOne struct {
 	config
 	id            int
 	age           *int
+	addage        *int
 	name          *string
 	groups        map[int]struct{}
 	removedGroups map[int]struct{}
@@ -209,6 +221,12 @@ func (uuo *UserUpdateOne) SetAge(i int) *UserUpdateOne {
 	return uuo
 }
 
+// AddAge adds i to age.
+func (uuo *UserUpdateOne) AddAge(i int) *UserUpdateOne {
+	uuo.addage = &i
+	return uuo
+}
+
 // SetName sets the name field.
 func (uuo *UserUpdateOne) SetName(s string) *UserUpdateOne {
 	uuo.name = &s
@@ -317,15 +335,20 @@ func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (u *User, err error) {
 		res     sql.Result
 		builder = sql.Update(user.Table).Where(sql.InInts(user.FieldID, ids...))
 	)
-	if uuo.age != nil {
+	if value := uuo.age; value != nil {
 		update = true
-		builder.Set(user.FieldAge, *uuo.age)
-		u.Age = *uuo.age
+		builder.Set(user.FieldAge, *value)
+		u.Age = *value
 	}
-	if uuo.name != nil {
+	if value := uuo.addage; value != nil {
 		update = true
-		builder.Set(user.FieldName, *uuo.name)
-		u.Name = *uuo.name
+		builder.Add(user.FieldAge, *value)
+		u.Age += *value
+	}
+	if value := uuo.name; value != nil {
+		update = true
+		builder.Set(user.FieldName, *value)
+		u.Name = *value
 	}
 	if update {
 		query, args := builder.Query()