Files
ent/entc/integration/ent/user_update.go
Ariel Mashraki 5bc8568069 all: add license reference to README and add copyright headers to gencode
Reviewed By: alexsn

Differential Revision: D17119262

fbshipit-source-id: 046f095ca9432c920778db0edb2158dedb23c0a2
2019-08-30 08:46:03 -07:00

2438 lines
66 KiB
Go

// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
// This source code is licensed under the Apache 2.0 license found
// in the LICENSE file in the root directory of this source tree.
// Code generated (@generated) by entc, DO NOT EDIT.
package ent
import (
"context"
"errors"
"fmt"
"strconv"
"github.com/facebookincubator/ent/entc/integration/ent/card"
"github.com/facebookincubator/ent/entc/integration/ent/file"
"github.com/facebookincubator/ent/entc/integration/ent/pet"
"github.com/facebookincubator/ent/entc/integration/ent/predicate"
"github.com/facebookincubator/ent/entc/integration/ent/user"
"github.com/facebookincubator/ent/dialect"
"github.com/facebookincubator/ent/dialect/gremlin"
"github.com/facebookincubator/ent/dialect/gremlin/graph/dsl"
"github.com/facebookincubator/ent/dialect/gremlin/graph/dsl/__"
"github.com/facebookincubator/ent/dialect/gremlin/graph/dsl/g"
"github.com/facebookincubator/ent/dialect/gremlin/graph/dsl/p"
"github.com/facebookincubator/ent/dialect/sql"
)
// UserUpdate is the builder for updating User entities.
type UserUpdate struct {
config
age *int
name *string
last *string
nickname *string
phone *string
card map[string]struct{}
pets map[string]struct{}
files map[string]struct{}
groups map[string]struct{}
friends map[string]struct{}
followers map[string]struct{}
following map[string]struct{}
team map[string]struct{}
spouse map[string]struct{}
children map[string]struct{}
parent map[string]struct{}
clearedCard bool
removedPets map[string]struct{}
removedFiles map[string]struct{}
removedGroups map[string]struct{}
removedFriends map[string]struct{}
removedFollowers map[string]struct{}
removedFollowing map[string]struct{}
clearedTeam bool
clearedSpouse bool
removedChildren map[string]struct{}
clearedParent bool
predicates []predicate.User
}
// Where adds a new predicate for the builder.
func (uu *UserUpdate) Where(ps ...predicate.User) *UserUpdate {
uu.predicates = append(uu.predicates, ps...)
return uu
}
// SetAge sets the age field.
func (uu *UserUpdate) SetAge(i int) *UserUpdate {
uu.age = &i
return uu
}
// SetName sets the name field.
func (uu *UserUpdate) SetName(s string) *UserUpdate {
uu.name = &s
return uu
}
// SetLast sets the last field.
func (uu *UserUpdate) SetLast(s string) *UserUpdate {
uu.last = &s
return uu
}
// SetNillableLast sets the last field if the given value is not nil.
func (uu *UserUpdate) SetNillableLast(s *string) *UserUpdate {
if s != nil {
uu.SetLast(*s)
}
return uu
}
// SetNickname sets the nickname field.
func (uu *UserUpdate) SetNickname(s string) *UserUpdate {
uu.nickname = &s
return uu
}
// SetNillableNickname sets the nickname field if the given value is not nil.
func (uu *UserUpdate) SetNillableNickname(s *string) *UserUpdate {
if s != nil {
uu.SetNickname(*s)
}
return uu
}
// SetPhone sets the phone field.
func (uu *UserUpdate) SetPhone(s string) *UserUpdate {
uu.phone = &s
return uu
}
// SetNillablePhone sets the phone field if the given value is not nil.
func (uu *UserUpdate) SetNillablePhone(s *string) *UserUpdate {
if s != nil {
uu.SetPhone(*s)
}
return uu
}
// SetCardID sets the card edge to Card by id.
func (uu *UserUpdate) SetCardID(id string) *UserUpdate {
if uu.card == nil {
uu.card = make(map[string]struct{})
}
uu.card[id] = struct{}{}
return uu
}
// SetNillableCardID sets the card edge to Card by id if the given value is not nil.
func (uu *UserUpdate) SetNillableCardID(id *string) *UserUpdate {
if id != nil {
uu = uu.SetCardID(*id)
}
return uu
}
// SetCard sets the card edge to Card.
func (uu *UserUpdate) SetCard(c *Card) *UserUpdate {
return uu.SetCardID(c.ID)
}
// AddPetIDs adds the pets edge to Pet by ids.
func (uu *UserUpdate) AddPetIDs(ids ...string) *UserUpdate {
if uu.pets == nil {
uu.pets = make(map[string]struct{})
}
for i := range ids {
uu.pets[ids[i]] = struct{}{}
}
return uu
}
// AddPets adds the pets edges to Pet.
func (uu *UserUpdate) AddPets(p ...*Pet) *UserUpdate {
ids := make([]string, len(p))
for i := range p {
ids[i] = p[i].ID
}
return uu.AddPetIDs(ids...)
}
// AddFileIDs adds the files edge to File by ids.
func (uu *UserUpdate) AddFileIDs(ids ...string) *UserUpdate {
if uu.files == nil {
uu.files = make(map[string]struct{})
}
for i := range ids {
uu.files[ids[i]] = struct{}{}
}
return uu
}
// AddFiles adds the files edges to File.
func (uu *UserUpdate) AddFiles(f ...*File) *UserUpdate {
ids := make([]string, len(f))
for i := range f {
ids[i] = f[i].ID
}
return uu.AddFileIDs(ids...)
}
// AddGroupIDs adds the groups edge to Group by ids.
func (uu *UserUpdate) AddGroupIDs(ids ...string) *UserUpdate {
if uu.groups == nil {
uu.groups = make(map[string]struct{})
}
for i := range ids {
uu.groups[ids[i]] = struct{}{}
}
return uu
}
// AddGroups adds the groups edges to Group.
func (uu *UserUpdate) AddGroups(g ...*Group) *UserUpdate {
ids := make([]string, len(g))
for i := range g {
ids[i] = g[i].ID
}
return uu.AddGroupIDs(ids...)
}
// AddFriendIDs adds the friends edge to User by ids.
func (uu *UserUpdate) AddFriendIDs(ids ...string) *UserUpdate {
if uu.friends == nil {
uu.friends = make(map[string]struct{})
}
for i := range ids {
uu.friends[ids[i]] = struct{}{}
}
return uu
}
// AddFriends adds the friends edges to User.
func (uu *UserUpdate) AddFriends(u ...*User) *UserUpdate {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uu.AddFriendIDs(ids...)
}
// AddFollowerIDs adds the followers edge to User by ids.
func (uu *UserUpdate) AddFollowerIDs(ids ...string) *UserUpdate {
if uu.followers == nil {
uu.followers = make(map[string]struct{})
}
for i := range ids {
uu.followers[ids[i]] = struct{}{}
}
return uu
}
// AddFollowers adds the followers edges to User.
func (uu *UserUpdate) AddFollowers(u ...*User) *UserUpdate {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uu.AddFollowerIDs(ids...)
}
// AddFollowingIDs adds the following edge to User by ids.
func (uu *UserUpdate) AddFollowingIDs(ids ...string) *UserUpdate {
if uu.following == nil {
uu.following = make(map[string]struct{})
}
for i := range ids {
uu.following[ids[i]] = struct{}{}
}
return uu
}
// AddFollowing adds the following edges to User.
func (uu *UserUpdate) AddFollowing(u ...*User) *UserUpdate {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uu.AddFollowingIDs(ids...)
}
// SetTeamID sets the team edge to Pet by id.
func (uu *UserUpdate) SetTeamID(id string) *UserUpdate {
if uu.team == nil {
uu.team = make(map[string]struct{})
}
uu.team[id] = struct{}{}
return uu
}
// SetNillableTeamID sets the team edge to Pet by id if the given value is not nil.
func (uu *UserUpdate) SetNillableTeamID(id *string) *UserUpdate {
if id != nil {
uu = uu.SetTeamID(*id)
}
return uu
}
// SetTeam sets the team edge to Pet.
func (uu *UserUpdate) SetTeam(p *Pet) *UserUpdate {
return uu.SetTeamID(p.ID)
}
// SetSpouseID sets the spouse edge to User by id.
func (uu *UserUpdate) SetSpouseID(id string) *UserUpdate {
if uu.spouse == nil {
uu.spouse = make(map[string]struct{})
}
uu.spouse[id] = struct{}{}
return uu
}
// SetNillableSpouseID sets the spouse edge to User by id if the given value is not nil.
func (uu *UserUpdate) SetNillableSpouseID(id *string) *UserUpdate {
if id != nil {
uu = uu.SetSpouseID(*id)
}
return uu
}
// SetSpouse sets the spouse edge to User.
func (uu *UserUpdate) SetSpouse(u *User) *UserUpdate {
return uu.SetSpouseID(u.ID)
}
// AddChildIDs adds the children edge to User by ids.
func (uu *UserUpdate) AddChildIDs(ids ...string) *UserUpdate {
if uu.children == nil {
uu.children = make(map[string]struct{})
}
for i := range ids {
uu.children[ids[i]] = struct{}{}
}
return uu
}
// AddChildren adds the children edges to User.
func (uu *UserUpdate) AddChildren(u ...*User) *UserUpdate {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uu.AddChildIDs(ids...)
}
// SetParentID sets the parent edge to User by id.
func (uu *UserUpdate) SetParentID(id string) *UserUpdate {
if uu.parent == nil {
uu.parent = make(map[string]struct{})
}
uu.parent[id] = struct{}{}
return uu
}
// SetNillableParentID sets the parent edge to User by id if the given value is not nil.
func (uu *UserUpdate) SetNillableParentID(id *string) *UserUpdate {
if id != nil {
uu = uu.SetParentID(*id)
}
return uu
}
// SetParent sets the parent edge to User.
func (uu *UserUpdate) SetParent(u *User) *UserUpdate {
return uu.SetParentID(u.ID)
}
// ClearCard clears the card edge to Card.
func (uu *UserUpdate) ClearCard() *UserUpdate {
uu.clearedCard = true
return uu
}
// RemovePetIDs removes the pets edge to Pet by ids.
func (uu *UserUpdate) RemovePetIDs(ids ...string) *UserUpdate {
if uu.removedPets == nil {
uu.removedPets = make(map[string]struct{})
}
for i := range ids {
uu.removedPets[ids[i]] = struct{}{}
}
return uu
}
// RemovePets removes pets edges to Pet.
func (uu *UserUpdate) RemovePets(p ...*Pet) *UserUpdate {
ids := make([]string, len(p))
for i := range p {
ids[i] = p[i].ID
}
return uu.RemovePetIDs(ids...)
}
// RemoveFileIDs removes the files edge to File by ids.
func (uu *UserUpdate) RemoveFileIDs(ids ...string) *UserUpdate {
if uu.removedFiles == nil {
uu.removedFiles = make(map[string]struct{})
}
for i := range ids {
uu.removedFiles[ids[i]] = struct{}{}
}
return uu
}
// RemoveFiles removes files edges to File.
func (uu *UserUpdate) RemoveFiles(f ...*File) *UserUpdate {
ids := make([]string, len(f))
for i := range f {
ids[i] = f[i].ID
}
return uu.RemoveFileIDs(ids...)
}
// RemoveGroupIDs removes the groups edge to Group by ids.
func (uu *UserUpdate) RemoveGroupIDs(ids ...string) *UserUpdate {
if uu.removedGroups == nil {
uu.removedGroups = make(map[string]struct{})
}
for i := range ids {
uu.removedGroups[ids[i]] = struct{}{}
}
return uu
}
// RemoveGroups removes groups edges to Group.
func (uu *UserUpdate) RemoveGroups(g ...*Group) *UserUpdate {
ids := make([]string, len(g))
for i := range g {
ids[i] = g[i].ID
}
return uu.RemoveGroupIDs(ids...)
}
// RemoveFriendIDs removes the friends edge to User by ids.
func (uu *UserUpdate) RemoveFriendIDs(ids ...string) *UserUpdate {
if uu.removedFriends == nil {
uu.removedFriends = make(map[string]struct{})
}
for i := range ids {
uu.removedFriends[ids[i]] = struct{}{}
}
return uu
}
// RemoveFriends removes friends edges to User.
func (uu *UserUpdate) RemoveFriends(u ...*User) *UserUpdate {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uu.RemoveFriendIDs(ids...)
}
// RemoveFollowerIDs removes the followers edge to User by ids.
func (uu *UserUpdate) RemoveFollowerIDs(ids ...string) *UserUpdate {
if uu.removedFollowers == nil {
uu.removedFollowers = make(map[string]struct{})
}
for i := range ids {
uu.removedFollowers[ids[i]] = struct{}{}
}
return uu
}
// RemoveFollowers removes followers edges to User.
func (uu *UserUpdate) RemoveFollowers(u ...*User) *UserUpdate {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uu.RemoveFollowerIDs(ids...)
}
// RemoveFollowingIDs removes the following edge to User by ids.
func (uu *UserUpdate) RemoveFollowingIDs(ids ...string) *UserUpdate {
if uu.removedFollowing == nil {
uu.removedFollowing = make(map[string]struct{})
}
for i := range ids {
uu.removedFollowing[ids[i]] = struct{}{}
}
return uu
}
// RemoveFollowing removes following edges to User.
func (uu *UserUpdate) RemoveFollowing(u ...*User) *UserUpdate {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uu.RemoveFollowingIDs(ids...)
}
// ClearTeam clears the team edge to Pet.
func (uu *UserUpdate) ClearTeam() *UserUpdate {
uu.clearedTeam = true
return uu
}
// ClearSpouse clears the spouse edge to User.
func (uu *UserUpdate) ClearSpouse() *UserUpdate {
uu.clearedSpouse = true
return uu
}
// RemoveChildIDs removes the children edge to User by ids.
func (uu *UserUpdate) RemoveChildIDs(ids ...string) *UserUpdate {
if uu.removedChildren == nil {
uu.removedChildren = make(map[string]struct{})
}
for i := range ids {
uu.removedChildren[ids[i]] = struct{}{}
}
return uu
}
// RemoveChildren removes children edges to User.
func (uu *UserUpdate) RemoveChildren(u ...*User) *UserUpdate {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uu.RemoveChildIDs(ids...)
}
// ClearParent clears the parent edge to User.
func (uu *UserUpdate) ClearParent() *UserUpdate {
uu.clearedParent = true
return uu
}
// Save executes the query and returns the number of rows/vertices matched by this operation.
func (uu *UserUpdate) Save(ctx context.Context) (int, error) {
if len(uu.card) > 1 {
return 0, errors.New("ent: multiple assignments on a unique edge \"card\"")
}
if len(uu.team) > 1 {
return 0, errors.New("ent: multiple assignments on a unique edge \"team\"")
}
if len(uu.spouse) > 1 {
return 0, errors.New("ent: multiple assignments on a unique edge \"spouse\"")
}
if len(uu.parent) > 1 {
return 0, errors.New("ent: multiple assignments on a unique edge \"parent\"")
}
switch uu.driver.Dialect() {
case dialect.MySQL, dialect.SQLite:
return uu.sqlSave(ctx)
case dialect.Gremlin:
return uu.gremlinSave(ctx)
default:
return 0, errors.New("ent: unsupported dialect")
}
}
// SaveX is like Save, but panics if an error occurs.
func (uu *UserUpdate) SaveX(ctx context.Context) int {
affected, err := uu.Save(ctx)
if err != nil {
panic(err)
}
return affected
}
// Exec executes the query.
func (uu *UserUpdate) Exec(ctx context.Context) error {
_, err := uu.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (uu *UserUpdate) ExecX(ctx context.Context) {
if err := uu.Exec(ctx); err != nil {
panic(err)
}
}
func (uu *UserUpdate) sqlSave(ctx context.Context) (n int, err error) {
selector := sql.Select(user.FieldID).From(sql.Table(user.Table))
for _, p := range uu.predicates {
p(selector)
}
rows := &sql.Rows{}
query, args := selector.Query()
if err = uu.driver.Query(ctx, query, args, rows); err != nil {
return 0, err
}
defer rows.Close()
var ids []int
for rows.Next() {
var id int
if err := rows.Scan(&id); err != nil {
return 0, fmt.Errorf("ent: failed reading id: %v", err)
}
ids = append(ids, id)
}
if len(ids) == 0 {
return 0, nil
}
tx, err := uu.driver.Tx(ctx)
if err != nil {
return 0, err
}
var (
update bool
res sql.Result
builder = sql.Update(user.Table).Where(sql.InInts(user.FieldID, ids...))
)
if uu.age != nil {
update = true
builder.Set(user.FieldAge, *uu.age)
}
if uu.name != nil {
update = true
builder.Set(user.FieldName, *uu.name)
}
if uu.last != nil {
update = true
builder.Set(user.FieldLast, *uu.last)
}
if uu.nickname != nil {
update = true
builder.Set(user.FieldNickname, *uu.nickname)
}
if uu.phone != nil {
update = true
builder.Set(user.FieldPhone, *uu.phone)
}
if update {
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if uu.clearedCard {
query, args := sql.Update(user.CardTable).
SetNull(user.CardColumn).
Where(sql.InInts(card.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.card) > 0 {
for _, id := range ids {
eid, serr := strconv.Atoi(keys(uu.card)[0])
if serr != nil {
return 0, rollback(tx, err)
}
query, args := sql.Update(user.CardTable).
Set(user.CardColumn, id).
Where(sql.EQ(card.FieldID, eid).And().IsNull(user.CardColumn)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return 0, rollback(tx, err)
}
if int(affected) < len(uu.card) {
return 0, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"card\" %v already connected to a different \"User\"", keys(uu.card))})
}
}
}
if len(uu.removedPets) > 0 {
eids := make([]int, len(uu.removedPets))
for eid := range uu.removedPets {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Update(user.PetsTable).
SetNull(user.PetsColumn).
Where(sql.InInts(user.PetsColumn, ids...)).
Where(sql.InInts(pet.FieldID, eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.pets) > 0 {
for _, id := range ids {
p := sql.P()
for eid := range uu.pets {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
p.Or().EQ(pet.FieldID, eid)
}
query, args := sql.Update(user.PetsTable).
Set(user.PetsColumn, id).
Where(sql.And(p, sql.IsNull(user.PetsColumn))).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return 0, rollback(tx, err)
}
if int(affected) < len(uu.pets) {
return 0, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"pets\" %v already connected to a different \"User\"", keys(uu.pets))})
}
}
}
if len(uu.removedFiles) > 0 {
eids := make([]int, len(uu.removedFiles))
for eid := range uu.removedFiles {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Update(user.FilesTable).
SetNull(user.FilesColumn).
Where(sql.InInts(user.FilesColumn, ids...)).
Where(sql.InInts(file.FieldID, eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.files) > 0 {
for _, id := range ids {
p := sql.P()
for eid := range uu.files {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
p.Or().EQ(file.FieldID, eid)
}
query, args := sql.Update(user.FilesTable).
Set(user.FilesColumn, id).
Where(sql.And(p, sql.IsNull(user.FilesColumn))).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return 0, rollback(tx, err)
}
if int(affected) < len(uu.files) {
return 0, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"files\" %v already connected to a different \"User\"", keys(uu.files))})
}
}
}
if len(uu.removedGroups) > 0 {
eids := make([]int, len(uu.removedGroups))
for eid := range uu.removedGroups {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Delete(user.GroupsTable).
Where(sql.InInts(user.GroupsPrimaryKey[0], ids...)).
Where(sql.InInts(user.GroupsPrimaryKey[1], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.groups) > 0 {
values := make([][]int, 0, len(ids))
for _, id := range ids {
for eid := range uu.groups {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
values = append(values, []int{id, eid})
}
}
builder := sql.Insert(user.GroupsTable).
Columns(user.GroupsPrimaryKey[0], user.GroupsPrimaryKey[1])
for _, v := range values {
builder.Values(v[0], v[1])
}
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.removedFriends) > 0 {
eids := make([]int, len(uu.removedFriends))
for eid := range uu.removedFriends {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Delete(user.FriendsTable).
Where(sql.InInts(user.FriendsPrimaryKey[0], ids...)).
Where(sql.InInts(user.FriendsPrimaryKey[1], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
query, args = sql.Delete(user.FriendsTable).
Where(sql.InInts(user.FriendsPrimaryKey[1], ids...)).
Where(sql.InInts(user.FriendsPrimaryKey[0], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.friends) > 0 {
values := make([][]int, 0, len(ids))
for _, id := range ids {
for eid := range uu.friends {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
values = append(values, []int{id, eid}, []int{eid, id})
}
}
builder := sql.Insert(user.FriendsTable).
Columns(user.FriendsPrimaryKey[0], user.FriendsPrimaryKey[1])
for _, v := range values {
builder.Values(v[0], v[1])
}
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.removedFollowers) > 0 {
eids := make([]int, len(uu.removedFollowers))
for eid := range uu.removedFollowers {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Delete(user.FollowersTable).
Where(sql.InInts(user.FollowersPrimaryKey[1], ids...)).
Where(sql.InInts(user.FollowersPrimaryKey[0], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.followers) > 0 {
values := make([][]int, 0, len(ids))
for _, id := range ids {
for eid := range uu.followers {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
values = append(values, []int{id, eid})
}
}
builder := sql.Insert(user.FollowersTable).
Columns(user.FollowersPrimaryKey[1], user.FollowersPrimaryKey[0])
for _, v := range values {
builder.Values(v[0], v[1])
}
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.removedFollowing) > 0 {
eids := make([]int, len(uu.removedFollowing))
for eid := range uu.removedFollowing {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Delete(user.FollowingTable).
Where(sql.InInts(user.FollowingPrimaryKey[0], ids...)).
Where(sql.InInts(user.FollowingPrimaryKey[1], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.following) > 0 {
values := make([][]int, 0, len(ids))
for _, id := range ids {
for eid := range uu.following {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
values = append(values, []int{id, eid})
}
}
builder := sql.Insert(user.FollowingTable).
Columns(user.FollowingPrimaryKey[0], user.FollowingPrimaryKey[1])
for _, v := range values {
builder.Values(v[0], v[1])
}
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if uu.clearedTeam {
query, args := sql.Update(user.TeamTable).
SetNull(user.TeamColumn).
Where(sql.InInts(pet.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.team) > 0 {
for _, id := range ids {
eid, serr := strconv.Atoi(keys(uu.team)[0])
if serr != nil {
return 0, rollback(tx, err)
}
query, args := sql.Update(user.TeamTable).
Set(user.TeamColumn, id).
Where(sql.EQ(pet.FieldID, eid).And().IsNull(user.TeamColumn)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return 0, rollback(tx, err)
}
if int(affected) < len(uu.team) {
return 0, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"team\" %v already connected to a different \"User\"", keys(uu.team))})
}
}
}
if uu.clearedSpouse {
query, args := sql.Update(user.SpouseTable).
SetNull(user.SpouseColumn).
Where(sql.InInts(user.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
query, args = sql.Update(user.SpouseTable).
SetNull(user.SpouseColumn).
Where(sql.InInts(user.SpouseColumn, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.spouse) > 0 {
if n := len(ids); n > 1 {
return 0, rollback(tx, fmt.Errorf("ent: can't link O2O edge \"spouse\" to %d vertices (> 1)", n))
}
for eid := range uu.spouse {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
query, args := sql.Update(user.SpouseTable).
Set(user.SpouseColumn, eid).
Where(sql.EQ(user.FieldID, ids[0])).Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
query, args = sql.Update(user.SpouseTable).
Set(user.SpouseColumn, ids[0]).
Where(sql.EQ(user.FieldID, eid).And().IsNull(user.SpouseColumn)).Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return 0, rollback(tx, err)
}
if int(affected) < len(uu.spouse) {
return 0, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("\"spouse\" (%v) already connected to a different \"User\"", eid)})
}
}
}
if len(uu.removedChildren) > 0 {
eids := make([]int, len(uu.removedChildren))
for eid := range uu.removedChildren {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Update(user.ChildrenTable).
SetNull(user.ChildrenColumn).
Where(sql.InInts(user.ChildrenColumn, ids...)).
Where(sql.InInts(user.FieldID, eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.children) > 0 {
for _, id := range ids {
p := sql.P()
for eid := range uu.children {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
p.Or().EQ(user.FieldID, eid)
}
query, args := sql.Update(user.ChildrenTable).
Set(user.ChildrenColumn, id).
Where(sql.And(p, sql.IsNull(user.ChildrenColumn))).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return 0, rollback(tx, err)
}
if int(affected) < len(uu.children) {
return 0, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"children\" %v already connected to a different \"User\"", keys(uu.children))})
}
}
}
if uu.clearedParent {
query, args := sql.Update(user.ParentTable).
SetNull(user.ParentColumn).
Where(sql.InInts(user.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
if len(uu.parent) > 0 {
for eid := range uu.parent {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
query, args := sql.Update(user.ParentTable).
Set(user.ParentColumn, eid).
Where(sql.InInts(user.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return 0, rollback(tx, err)
}
}
}
if err = tx.Commit(); err != nil {
return 0, err
}
return len(ids), nil
}
func (uu *UserUpdate) gremlinSave(ctx context.Context) (int, error) {
res := &gremlin.Response{}
query, bindings := uu.gremlin().Query()
if err := uu.driver.Exec(ctx, query, bindings, res); err != nil {
return 0, err
}
if err, ok := isConstantError(res); ok {
return 0, err
}
return res.ReadInt()
}
func (uu *UserUpdate) gremlin() *dsl.Traversal {
type constraint struct {
pred *dsl.Traversal // constraint predicate.
test *dsl.Traversal // test matches and its constant.
}
constraints := make([]*constraint, 0, 8)
v := g.V().HasLabel(user.Label)
for _, p := range uu.predicates {
p(v)
}
var (
rv = v.Clone()
_ = rv
trs []*dsl.Traversal
)
if uu.age != nil {
v.Property(dsl.Single, user.FieldAge, *uu.age)
}
if uu.name != nil {
v.Property(dsl.Single, user.FieldName, *uu.name)
}
if uu.last != nil {
v.Property(dsl.Single, user.FieldLast, *uu.last)
}
if uu.nickname != nil {
constraints = append(constraints, &constraint{
pred: g.V().Has(user.Label, user.FieldNickname, *uu.nickname).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueField(user.Label, user.FieldNickname, *uu.nickname)),
})
v.Property(dsl.Single, user.FieldNickname, *uu.nickname)
}
if uu.phone != nil {
constraints = append(constraints, &constraint{
pred: g.V().Has(user.Label, user.FieldPhone, *uu.phone).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueField(user.Label, user.FieldPhone, *uu.phone)),
})
v.Property(dsl.Single, user.FieldPhone, *uu.phone)
}
if uu.clearedCard {
tr := rv.Clone().OutE(user.CardLabel).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.card {
v.AddE(user.CardLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.CardLabel).InV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.CardLabel, id)),
})
}
for id := range uu.removedPets {
tr := rv.Clone().OutE(user.PetsLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.pets {
v.AddE(user.PetsLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.PetsLabel).InV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.PetsLabel, id)),
})
}
for id := range uu.removedFiles {
tr := rv.Clone().OutE(user.FilesLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.files {
v.AddE(user.FilesLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.FilesLabel).InV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.FilesLabel, id)),
})
}
for id := range uu.removedGroups {
tr := rv.Clone().OutE(user.GroupsLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.groups {
v.AddE(user.GroupsLabel).To(g.V(id)).OutV()
}
for id := range uu.removedFriends {
tr := rv.Clone().BothE(user.FriendsLabel).Where(__.Or(__.InV().HasID(id), __.OutV().HasID(id))).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.friends {
v.AddE(user.FriendsLabel).To(g.V(id)).OutV()
}
for id := range uu.removedFollowers {
tr := rv.Clone().InE(user.FollowingLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.followers {
v.AddE(user.FollowingLabel).From(g.V(id)).InV()
}
for id := range uu.removedFollowing {
tr := rv.Clone().OutE(user.FollowingLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.following {
v.AddE(user.FollowingLabel).To(g.V(id)).OutV()
}
if uu.clearedTeam {
tr := rv.Clone().OutE(user.TeamLabel).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.team {
v.AddE(user.TeamLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.TeamLabel).InV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.TeamLabel, id)),
})
}
if uu.clearedSpouse {
tr := rv.Clone().BothE(user.SpouseLabel).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.spouse {
v.AddE(user.SpouseLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: rv.Clone().Both(user.SpouseLabel).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.SpouseLabel, id)),
})
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.SpouseLabel).Where(__.Or(__.InV().HasID(id), __.OutV().HasID(id))).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.SpouseLabel, id)),
})
}
for id := range uu.removedChildren {
tr := rv.Clone().InE(user.ParentLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.children {
v.AddE(user.ParentLabel).From(g.V(id)).InV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.ParentLabel).OutV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.ParentLabel, id)),
})
}
if uu.clearedParent {
tr := rv.Clone().OutE(user.ParentLabel).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uu.parent {
v.AddE(user.ParentLabel).To(g.V(id)).OutV()
}
v.Count()
if len(constraints) > 0 {
constraints = append(constraints, &constraint{
pred: rv.Count(),
test: __.Is(p.GT(1)).Constant(&ErrConstraintFailed{msg: "update traversal contains more than one vertex"}),
})
v = constraints[0].pred.Coalesce(constraints[0].test, v)
for _, cr := range constraints[1:] {
v = cr.pred.Coalesce(cr.test, v)
}
}
trs = append(trs, v)
return dsl.Join(trs...)
}
// UserUpdateOne is the builder for updating a single User entity.
type UserUpdateOne struct {
config
id string
age *int
name *string
last *string
nickname *string
phone *string
card map[string]struct{}
pets map[string]struct{}
files map[string]struct{}
groups map[string]struct{}
friends map[string]struct{}
followers map[string]struct{}
following map[string]struct{}
team map[string]struct{}
spouse map[string]struct{}
children map[string]struct{}
parent map[string]struct{}
clearedCard bool
removedPets map[string]struct{}
removedFiles map[string]struct{}
removedGroups map[string]struct{}
removedFriends map[string]struct{}
removedFollowers map[string]struct{}
removedFollowing map[string]struct{}
clearedTeam bool
clearedSpouse bool
removedChildren map[string]struct{}
clearedParent bool
}
// SetAge sets the age field.
func (uuo *UserUpdateOne) SetAge(i int) *UserUpdateOne {
uuo.age = &i
return uuo
}
// SetName sets the name field.
func (uuo *UserUpdateOne) SetName(s string) *UserUpdateOne {
uuo.name = &s
return uuo
}
// SetLast sets the last field.
func (uuo *UserUpdateOne) SetLast(s string) *UserUpdateOne {
uuo.last = &s
return uuo
}
// SetNillableLast sets the last field if the given value is not nil.
func (uuo *UserUpdateOne) SetNillableLast(s *string) *UserUpdateOne {
if s != nil {
uuo.SetLast(*s)
}
return uuo
}
// SetNickname sets the nickname field.
func (uuo *UserUpdateOne) SetNickname(s string) *UserUpdateOne {
uuo.nickname = &s
return uuo
}
// SetNillableNickname sets the nickname field if the given value is not nil.
func (uuo *UserUpdateOne) SetNillableNickname(s *string) *UserUpdateOne {
if s != nil {
uuo.SetNickname(*s)
}
return uuo
}
// SetPhone sets the phone field.
func (uuo *UserUpdateOne) SetPhone(s string) *UserUpdateOne {
uuo.phone = &s
return uuo
}
// SetNillablePhone sets the phone field if the given value is not nil.
func (uuo *UserUpdateOne) SetNillablePhone(s *string) *UserUpdateOne {
if s != nil {
uuo.SetPhone(*s)
}
return uuo
}
// SetCardID sets the card edge to Card by id.
func (uuo *UserUpdateOne) SetCardID(id string) *UserUpdateOne {
if uuo.card == nil {
uuo.card = make(map[string]struct{})
}
uuo.card[id] = struct{}{}
return uuo
}
// SetNillableCardID sets the card edge to Card by id if the given value is not nil.
func (uuo *UserUpdateOne) SetNillableCardID(id *string) *UserUpdateOne {
if id != nil {
uuo = uuo.SetCardID(*id)
}
return uuo
}
// SetCard sets the card edge to Card.
func (uuo *UserUpdateOne) SetCard(c *Card) *UserUpdateOne {
return uuo.SetCardID(c.ID)
}
// AddPetIDs adds the pets edge to Pet by ids.
func (uuo *UserUpdateOne) AddPetIDs(ids ...string) *UserUpdateOne {
if uuo.pets == nil {
uuo.pets = make(map[string]struct{})
}
for i := range ids {
uuo.pets[ids[i]] = struct{}{}
}
return uuo
}
// AddPets adds the pets edges to Pet.
func (uuo *UserUpdateOne) AddPets(p ...*Pet) *UserUpdateOne {
ids := make([]string, len(p))
for i := range p {
ids[i] = p[i].ID
}
return uuo.AddPetIDs(ids...)
}
// AddFileIDs adds the files edge to File by ids.
func (uuo *UserUpdateOne) AddFileIDs(ids ...string) *UserUpdateOne {
if uuo.files == nil {
uuo.files = make(map[string]struct{})
}
for i := range ids {
uuo.files[ids[i]] = struct{}{}
}
return uuo
}
// AddFiles adds the files edges to File.
func (uuo *UserUpdateOne) AddFiles(f ...*File) *UserUpdateOne {
ids := make([]string, len(f))
for i := range f {
ids[i] = f[i].ID
}
return uuo.AddFileIDs(ids...)
}
// AddGroupIDs adds the groups edge to Group by ids.
func (uuo *UserUpdateOne) AddGroupIDs(ids ...string) *UserUpdateOne {
if uuo.groups == nil {
uuo.groups = make(map[string]struct{})
}
for i := range ids {
uuo.groups[ids[i]] = struct{}{}
}
return uuo
}
// AddGroups adds the groups edges to Group.
func (uuo *UserUpdateOne) AddGroups(g ...*Group) *UserUpdateOne {
ids := make([]string, len(g))
for i := range g {
ids[i] = g[i].ID
}
return uuo.AddGroupIDs(ids...)
}
// AddFriendIDs adds the friends edge to User by ids.
func (uuo *UserUpdateOne) AddFriendIDs(ids ...string) *UserUpdateOne {
if uuo.friends == nil {
uuo.friends = make(map[string]struct{})
}
for i := range ids {
uuo.friends[ids[i]] = struct{}{}
}
return uuo
}
// AddFriends adds the friends edges to User.
func (uuo *UserUpdateOne) AddFriends(u ...*User) *UserUpdateOne {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uuo.AddFriendIDs(ids...)
}
// AddFollowerIDs adds the followers edge to User by ids.
func (uuo *UserUpdateOne) AddFollowerIDs(ids ...string) *UserUpdateOne {
if uuo.followers == nil {
uuo.followers = make(map[string]struct{})
}
for i := range ids {
uuo.followers[ids[i]] = struct{}{}
}
return uuo
}
// AddFollowers adds the followers edges to User.
func (uuo *UserUpdateOne) AddFollowers(u ...*User) *UserUpdateOne {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uuo.AddFollowerIDs(ids...)
}
// AddFollowingIDs adds the following edge to User by ids.
func (uuo *UserUpdateOne) AddFollowingIDs(ids ...string) *UserUpdateOne {
if uuo.following == nil {
uuo.following = make(map[string]struct{})
}
for i := range ids {
uuo.following[ids[i]] = struct{}{}
}
return uuo
}
// AddFollowing adds the following edges to User.
func (uuo *UserUpdateOne) AddFollowing(u ...*User) *UserUpdateOne {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uuo.AddFollowingIDs(ids...)
}
// SetTeamID sets the team edge to Pet by id.
func (uuo *UserUpdateOne) SetTeamID(id string) *UserUpdateOne {
if uuo.team == nil {
uuo.team = make(map[string]struct{})
}
uuo.team[id] = struct{}{}
return uuo
}
// SetNillableTeamID sets the team edge to Pet by id if the given value is not nil.
func (uuo *UserUpdateOne) SetNillableTeamID(id *string) *UserUpdateOne {
if id != nil {
uuo = uuo.SetTeamID(*id)
}
return uuo
}
// SetTeam sets the team edge to Pet.
func (uuo *UserUpdateOne) SetTeam(p *Pet) *UserUpdateOne {
return uuo.SetTeamID(p.ID)
}
// SetSpouseID sets the spouse edge to User by id.
func (uuo *UserUpdateOne) SetSpouseID(id string) *UserUpdateOne {
if uuo.spouse == nil {
uuo.spouse = make(map[string]struct{})
}
uuo.spouse[id] = struct{}{}
return uuo
}
// SetNillableSpouseID sets the spouse edge to User by id if the given value is not nil.
func (uuo *UserUpdateOne) SetNillableSpouseID(id *string) *UserUpdateOne {
if id != nil {
uuo = uuo.SetSpouseID(*id)
}
return uuo
}
// SetSpouse sets the spouse edge to User.
func (uuo *UserUpdateOne) SetSpouse(u *User) *UserUpdateOne {
return uuo.SetSpouseID(u.ID)
}
// AddChildIDs adds the children edge to User by ids.
func (uuo *UserUpdateOne) AddChildIDs(ids ...string) *UserUpdateOne {
if uuo.children == nil {
uuo.children = make(map[string]struct{})
}
for i := range ids {
uuo.children[ids[i]] = struct{}{}
}
return uuo
}
// AddChildren adds the children edges to User.
func (uuo *UserUpdateOne) AddChildren(u ...*User) *UserUpdateOne {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uuo.AddChildIDs(ids...)
}
// SetParentID sets the parent edge to User by id.
func (uuo *UserUpdateOne) SetParentID(id string) *UserUpdateOne {
if uuo.parent == nil {
uuo.parent = make(map[string]struct{})
}
uuo.parent[id] = struct{}{}
return uuo
}
// SetNillableParentID sets the parent edge to User by id if the given value is not nil.
func (uuo *UserUpdateOne) SetNillableParentID(id *string) *UserUpdateOne {
if id != nil {
uuo = uuo.SetParentID(*id)
}
return uuo
}
// SetParent sets the parent edge to User.
func (uuo *UserUpdateOne) SetParent(u *User) *UserUpdateOne {
return uuo.SetParentID(u.ID)
}
// ClearCard clears the card edge to Card.
func (uuo *UserUpdateOne) ClearCard() *UserUpdateOne {
uuo.clearedCard = true
return uuo
}
// RemovePetIDs removes the pets edge to Pet by ids.
func (uuo *UserUpdateOne) RemovePetIDs(ids ...string) *UserUpdateOne {
if uuo.removedPets == nil {
uuo.removedPets = make(map[string]struct{})
}
for i := range ids {
uuo.removedPets[ids[i]] = struct{}{}
}
return uuo
}
// RemovePets removes pets edges to Pet.
func (uuo *UserUpdateOne) RemovePets(p ...*Pet) *UserUpdateOne {
ids := make([]string, len(p))
for i := range p {
ids[i] = p[i].ID
}
return uuo.RemovePetIDs(ids...)
}
// RemoveFileIDs removes the files edge to File by ids.
func (uuo *UserUpdateOne) RemoveFileIDs(ids ...string) *UserUpdateOne {
if uuo.removedFiles == nil {
uuo.removedFiles = make(map[string]struct{})
}
for i := range ids {
uuo.removedFiles[ids[i]] = struct{}{}
}
return uuo
}
// RemoveFiles removes files edges to File.
func (uuo *UserUpdateOne) RemoveFiles(f ...*File) *UserUpdateOne {
ids := make([]string, len(f))
for i := range f {
ids[i] = f[i].ID
}
return uuo.RemoveFileIDs(ids...)
}
// RemoveGroupIDs removes the groups edge to Group by ids.
func (uuo *UserUpdateOne) RemoveGroupIDs(ids ...string) *UserUpdateOne {
if uuo.removedGroups == nil {
uuo.removedGroups = make(map[string]struct{})
}
for i := range ids {
uuo.removedGroups[ids[i]] = struct{}{}
}
return uuo
}
// RemoveGroups removes groups edges to Group.
func (uuo *UserUpdateOne) RemoveGroups(g ...*Group) *UserUpdateOne {
ids := make([]string, len(g))
for i := range g {
ids[i] = g[i].ID
}
return uuo.RemoveGroupIDs(ids...)
}
// RemoveFriendIDs removes the friends edge to User by ids.
func (uuo *UserUpdateOne) RemoveFriendIDs(ids ...string) *UserUpdateOne {
if uuo.removedFriends == nil {
uuo.removedFriends = make(map[string]struct{})
}
for i := range ids {
uuo.removedFriends[ids[i]] = struct{}{}
}
return uuo
}
// RemoveFriends removes friends edges to User.
func (uuo *UserUpdateOne) RemoveFriends(u ...*User) *UserUpdateOne {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uuo.RemoveFriendIDs(ids...)
}
// RemoveFollowerIDs removes the followers edge to User by ids.
func (uuo *UserUpdateOne) RemoveFollowerIDs(ids ...string) *UserUpdateOne {
if uuo.removedFollowers == nil {
uuo.removedFollowers = make(map[string]struct{})
}
for i := range ids {
uuo.removedFollowers[ids[i]] = struct{}{}
}
return uuo
}
// RemoveFollowers removes followers edges to User.
func (uuo *UserUpdateOne) RemoveFollowers(u ...*User) *UserUpdateOne {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uuo.RemoveFollowerIDs(ids...)
}
// RemoveFollowingIDs removes the following edge to User by ids.
func (uuo *UserUpdateOne) RemoveFollowingIDs(ids ...string) *UserUpdateOne {
if uuo.removedFollowing == nil {
uuo.removedFollowing = make(map[string]struct{})
}
for i := range ids {
uuo.removedFollowing[ids[i]] = struct{}{}
}
return uuo
}
// RemoveFollowing removes following edges to User.
func (uuo *UserUpdateOne) RemoveFollowing(u ...*User) *UserUpdateOne {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uuo.RemoveFollowingIDs(ids...)
}
// ClearTeam clears the team edge to Pet.
func (uuo *UserUpdateOne) ClearTeam() *UserUpdateOne {
uuo.clearedTeam = true
return uuo
}
// ClearSpouse clears the spouse edge to User.
func (uuo *UserUpdateOne) ClearSpouse() *UserUpdateOne {
uuo.clearedSpouse = true
return uuo
}
// RemoveChildIDs removes the children edge to User by ids.
func (uuo *UserUpdateOne) RemoveChildIDs(ids ...string) *UserUpdateOne {
if uuo.removedChildren == nil {
uuo.removedChildren = make(map[string]struct{})
}
for i := range ids {
uuo.removedChildren[ids[i]] = struct{}{}
}
return uuo
}
// RemoveChildren removes children edges to User.
func (uuo *UserUpdateOne) RemoveChildren(u ...*User) *UserUpdateOne {
ids := make([]string, len(u))
for i := range u {
ids[i] = u[i].ID
}
return uuo.RemoveChildIDs(ids...)
}
// ClearParent clears the parent edge to User.
func (uuo *UserUpdateOne) ClearParent() *UserUpdateOne {
uuo.clearedParent = true
return uuo
}
// Save executes the query and returns the updated entity.
func (uuo *UserUpdateOne) Save(ctx context.Context) (*User, error) {
if len(uuo.card) > 1 {
return nil, errors.New("ent: multiple assignments on a unique edge \"card\"")
}
if len(uuo.team) > 1 {
return nil, errors.New("ent: multiple assignments on a unique edge \"team\"")
}
if len(uuo.spouse) > 1 {
return nil, errors.New("ent: multiple assignments on a unique edge \"spouse\"")
}
if len(uuo.parent) > 1 {
return nil, errors.New("ent: multiple assignments on a unique edge \"parent\"")
}
switch uuo.driver.Dialect() {
case dialect.MySQL, dialect.SQLite:
return uuo.sqlSave(ctx)
case dialect.Gremlin:
return uuo.gremlinSave(ctx)
default:
return nil, errors.New("ent: unsupported dialect")
}
}
// SaveX is like Save, but panics if an error occurs.
func (uuo *UserUpdateOne) SaveX(ctx context.Context) *User {
u, err := uuo.Save(ctx)
if err != nil {
panic(err)
}
return u
}
// Exec executes the query on the entity.
func (uuo *UserUpdateOne) Exec(ctx context.Context) error {
_, err := uuo.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (uuo *UserUpdateOne) ExecX(ctx context.Context) {
if err := uuo.Exec(ctx); err != nil {
panic(err)
}
}
func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (u *User, err error) {
selector := sql.Select(user.Columns...).From(sql.Table(user.Table))
user.ID(uuo.id)(selector)
rows := &sql.Rows{}
query, args := selector.Query()
if err = uuo.driver.Query(ctx, query, args, rows); err != nil {
return nil, err
}
defer rows.Close()
var ids []int
for rows.Next() {
var id int
u = &User{config: uuo.config}
if err := u.FromRows(rows); err != nil {
return nil, fmt.Errorf("ent: failed scanning row into User: %v", err)
}
id = u.id()
ids = append(ids, id)
}
switch n := len(ids); {
case n == 0:
return nil, fmt.Errorf("ent: User not found with id: %v", uuo.id)
case n > 1:
return nil, fmt.Errorf("ent: more than one User with the same id: %v", uuo.id)
}
tx, err := uuo.driver.Tx(ctx)
if err != nil {
return nil, err
}
var (
update bool
res sql.Result
builder = sql.Update(user.Table).Where(sql.InInts(user.FieldID, ids...))
)
if uuo.age != nil {
update = true
builder.Set(user.FieldAge, *uuo.age)
u.Age = *uuo.age
}
if uuo.name != nil {
update = true
builder.Set(user.FieldName, *uuo.name)
u.Name = *uuo.name
}
if uuo.last != nil {
update = true
builder.Set(user.FieldLast, *uuo.last)
u.Last = *uuo.last
}
if uuo.nickname != nil {
update = true
builder.Set(user.FieldNickname, *uuo.nickname)
u.Nickname = *uuo.nickname
}
if uuo.phone != nil {
update = true
builder.Set(user.FieldPhone, *uuo.phone)
u.Phone = *uuo.phone
}
if update {
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if uuo.clearedCard {
query, args := sql.Update(user.CardTable).
SetNull(user.CardColumn).
Where(sql.InInts(card.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.card) > 0 {
for _, id := range ids {
eid, serr := strconv.Atoi(keys(uuo.card)[0])
if serr != nil {
return nil, rollback(tx, err)
}
query, args := sql.Update(user.CardTable).
Set(user.CardColumn, id).
Where(sql.EQ(card.FieldID, eid).And().IsNull(user.CardColumn)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len(uuo.card) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"card\" %v already connected to a different \"User\"", keys(uuo.card))})
}
}
}
if len(uuo.removedPets) > 0 {
eids := make([]int, len(uuo.removedPets))
for eid := range uuo.removedPets {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Update(user.PetsTable).
SetNull(user.PetsColumn).
Where(sql.InInts(user.PetsColumn, ids...)).
Where(sql.InInts(pet.FieldID, eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.pets) > 0 {
for _, id := range ids {
p := sql.P()
for eid := range uuo.pets {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
p.Or().EQ(pet.FieldID, eid)
}
query, args := sql.Update(user.PetsTable).
Set(user.PetsColumn, id).
Where(sql.And(p, sql.IsNull(user.PetsColumn))).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len(uuo.pets) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"pets\" %v already connected to a different \"User\"", keys(uuo.pets))})
}
}
}
if len(uuo.removedFiles) > 0 {
eids := make([]int, len(uuo.removedFiles))
for eid := range uuo.removedFiles {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Update(user.FilesTable).
SetNull(user.FilesColumn).
Where(sql.InInts(user.FilesColumn, ids...)).
Where(sql.InInts(file.FieldID, eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.files) > 0 {
for _, id := range ids {
p := sql.P()
for eid := range uuo.files {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
p.Or().EQ(file.FieldID, eid)
}
query, args := sql.Update(user.FilesTable).
Set(user.FilesColumn, id).
Where(sql.And(p, sql.IsNull(user.FilesColumn))).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len(uuo.files) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"files\" %v already connected to a different \"User\"", keys(uuo.files))})
}
}
}
if len(uuo.removedGroups) > 0 {
eids := make([]int, len(uuo.removedGroups))
for eid := range uuo.removedGroups {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Delete(user.GroupsTable).
Where(sql.InInts(user.GroupsPrimaryKey[0], ids...)).
Where(sql.InInts(user.GroupsPrimaryKey[1], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.groups) > 0 {
values := make([][]int, 0, len(ids))
for _, id := range ids {
for eid := range uuo.groups {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
values = append(values, []int{id, eid})
}
}
builder := sql.Insert(user.GroupsTable).
Columns(user.GroupsPrimaryKey[0], user.GroupsPrimaryKey[1])
for _, v := range values {
builder.Values(v[0], v[1])
}
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.removedFriends) > 0 {
eids := make([]int, len(uuo.removedFriends))
for eid := range uuo.removedFriends {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Delete(user.FriendsTable).
Where(sql.InInts(user.FriendsPrimaryKey[0], ids...)).
Where(sql.InInts(user.FriendsPrimaryKey[1], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
query, args = sql.Delete(user.FriendsTable).
Where(sql.InInts(user.FriendsPrimaryKey[1], ids...)).
Where(sql.InInts(user.FriendsPrimaryKey[0], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.friends) > 0 {
values := make([][]int, 0, len(ids))
for _, id := range ids {
for eid := range uuo.friends {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
values = append(values, []int{id, eid}, []int{eid, id})
}
}
builder := sql.Insert(user.FriendsTable).
Columns(user.FriendsPrimaryKey[0], user.FriendsPrimaryKey[1])
for _, v := range values {
builder.Values(v[0], v[1])
}
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.removedFollowers) > 0 {
eids := make([]int, len(uuo.removedFollowers))
for eid := range uuo.removedFollowers {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Delete(user.FollowersTable).
Where(sql.InInts(user.FollowersPrimaryKey[1], ids...)).
Where(sql.InInts(user.FollowersPrimaryKey[0], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.followers) > 0 {
values := make([][]int, 0, len(ids))
for _, id := range ids {
for eid := range uuo.followers {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
values = append(values, []int{id, eid})
}
}
builder := sql.Insert(user.FollowersTable).
Columns(user.FollowersPrimaryKey[1], user.FollowersPrimaryKey[0])
for _, v := range values {
builder.Values(v[0], v[1])
}
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.removedFollowing) > 0 {
eids := make([]int, len(uuo.removedFollowing))
for eid := range uuo.removedFollowing {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Delete(user.FollowingTable).
Where(sql.InInts(user.FollowingPrimaryKey[0], ids...)).
Where(sql.InInts(user.FollowingPrimaryKey[1], eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.following) > 0 {
values := make([][]int, 0, len(ids))
for _, id := range ids {
for eid := range uuo.following {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
values = append(values, []int{id, eid})
}
}
builder := sql.Insert(user.FollowingTable).
Columns(user.FollowingPrimaryKey[0], user.FollowingPrimaryKey[1])
for _, v := range values {
builder.Values(v[0], v[1])
}
query, args := builder.Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if uuo.clearedTeam {
query, args := sql.Update(user.TeamTable).
SetNull(user.TeamColumn).
Where(sql.InInts(pet.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.team) > 0 {
for _, id := range ids {
eid, serr := strconv.Atoi(keys(uuo.team)[0])
if serr != nil {
return nil, rollback(tx, err)
}
query, args := sql.Update(user.TeamTable).
Set(user.TeamColumn, id).
Where(sql.EQ(pet.FieldID, eid).And().IsNull(user.TeamColumn)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len(uuo.team) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"team\" %v already connected to a different \"User\"", keys(uuo.team))})
}
}
}
if uuo.clearedSpouse {
query, args := sql.Update(user.SpouseTable).
SetNull(user.SpouseColumn).
Where(sql.InInts(user.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
query, args = sql.Update(user.SpouseTable).
SetNull(user.SpouseColumn).
Where(sql.InInts(user.SpouseColumn, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.spouse) > 0 {
if n := len(ids); n > 1 {
return nil, rollback(tx, fmt.Errorf("ent: can't link O2O edge \"spouse\" to %d vertices (> 1)", n))
}
for eid := range uuo.spouse {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
query, args := sql.Update(user.SpouseTable).
Set(user.SpouseColumn, eid).
Where(sql.EQ(user.FieldID, ids[0])).Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
query, args = sql.Update(user.SpouseTable).
Set(user.SpouseColumn, ids[0]).
Where(sql.EQ(user.FieldID, eid).And().IsNull(user.SpouseColumn)).Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len(uuo.spouse) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("\"spouse\" (%v) already connected to a different \"User\"", eid)})
}
}
}
if len(uuo.removedChildren) > 0 {
eids := make([]int, len(uuo.removedChildren))
for eid := range uuo.removedChildren {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
eids = append(eids, eid)
}
query, args := sql.Update(user.ChildrenTable).
SetNull(user.ChildrenColumn).
Where(sql.InInts(user.ChildrenColumn, ids...)).
Where(sql.InInts(user.FieldID, eids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.children) > 0 {
for _, id := range ids {
p := sql.P()
for eid := range uuo.children {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
p.Or().EQ(user.FieldID, eid)
}
query, args := sql.Update(user.ChildrenTable).
Set(user.ChildrenColumn, id).
Where(sql.And(p, sql.IsNull(user.ChildrenColumn))).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
affected, err := res.RowsAffected()
if err != nil {
return nil, rollback(tx, err)
}
if int(affected) < len(uuo.children) {
return nil, rollback(tx, &ErrConstraintFailed{msg: fmt.Sprintf("one of \"children\" %v already connected to a different \"User\"", keys(uuo.children))})
}
}
}
if uuo.clearedParent {
query, args := sql.Update(user.ParentTable).
SetNull(user.ParentColumn).
Where(sql.InInts(user.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
if len(uuo.parent) > 0 {
for eid := range uuo.parent {
eid, serr := strconv.Atoi(eid)
if serr != nil {
err = rollback(tx, serr)
return
}
query, args := sql.Update(user.ParentTable).
Set(user.ParentColumn, eid).
Where(sql.InInts(user.FieldID, ids...)).
Query()
if err := tx.Exec(ctx, query, args, &res); err != nil {
return nil, rollback(tx, err)
}
}
}
if err = tx.Commit(); err != nil {
return nil, err
}
return u, nil
}
func (uuo *UserUpdateOne) gremlinSave(ctx context.Context) (*User, error) {
res := &gremlin.Response{}
query, bindings := uuo.gremlin(uuo.id).Query()
if err := uuo.driver.Exec(ctx, query, bindings, res); err != nil {
return nil, err
}
if err, ok := isConstantError(res); ok {
return nil, err
}
u := &User{config: uuo.config}
if err := u.FromResponse(res); err != nil {
return nil, err
}
return u, nil
}
func (uuo *UserUpdateOne) gremlin(id string) *dsl.Traversal {
type constraint struct {
pred *dsl.Traversal // constraint predicate.
test *dsl.Traversal // test matches and its constant.
}
constraints := make([]*constraint, 0, 8)
v := g.V(id)
var (
rv = v.Clone()
_ = rv
trs []*dsl.Traversal
)
if uuo.age != nil {
v.Property(dsl.Single, user.FieldAge, *uuo.age)
}
if uuo.name != nil {
v.Property(dsl.Single, user.FieldName, *uuo.name)
}
if uuo.last != nil {
v.Property(dsl.Single, user.FieldLast, *uuo.last)
}
if uuo.nickname != nil {
constraints = append(constraints, &constraint{
pred: g.V().Has(user.Label, user.FieldNickname, *uuo.nickname).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueField(user.Label, user.FieldNickname, *uuo.nickname)),
})
v.Property(dsl.Single, user.FieldNickname, *uuo.nickname)
}
if uuo.phone != nil {
constraints = append(constraints, &constraint{
pred: g.V().Has(user.Label, user.FieldPhone, *uuo.phone).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueField(user.Label, user.FieldPhone, *uuo.phone)),
})
v.Property(dsl.Single, user.FieldPhone, *uuo.phone)
}
if uuo.clearedCard {
tr := rv.Clone().OutE(user.CardLabel).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.card {
v.AddE(user.CardLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.CardLabel).InV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.CardLabel, id)),
})
}
for id := range uuo.removedPets {
tr := rv.Clone().OutE(user.PetsLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.pets {
v.AddE(user.PetsLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.PetsLabel).InV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.PetsLabel, id)),
})
}
for id := range uuo.removedFiles {
tr := rv.Clone().OutE(user.FilesLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.files {
v.AddE(user.FilesLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.FilesLabel).InV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.FilesLabel, id)),
})
}
for id := range uuo.removedGroups {
tr := rv.Clone().OutE(user.GroupsLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.groups {
v.AddE(user.GroupsLabel).To(g.V(id)).OutV()
}
for id := range uuo.removedFriends {
tr := rv.Clone().BothE(user.FriendsLabel).Where(__.Or(__.InV().HasID(id), __.OutV().HasID(id))).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.friends {
v.AddE(user.FriendsLabel).To(g.V(id)).OutV()
}
for id := range uuo.removedFollowers {
tr := rv.Clone().InE(user.FollowingLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.followers {
v.AddE(user.FollowingLabel).From(g.V(id)).InV()
}
for id := range uuo.removedFollowing {
tr := rv.Clone().OutE(user.FollowingLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.following {
v.AddE(user.FollowingLabel).To(g.V(id)).OutV()
}
if uuo.clearedTeam {
tr := rv.Clone().OutE(user.TeamLabel).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.team {
v.AddE(user.TeamLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.TeamLabel).InV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.TeamLabel, id)),
})
}
if uuo.clearedSpouse {
tr := rv.Clone().BothE(user.SpouseLabel).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.spouse {
v.AddE(user.SpouseLabel).To(g.V(id)).OutV()
constraints = append(constraints, &constraint{
pred: rv.Clone().Both(user.SpouseLabel).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.SpouseLabel, id)),
})
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.SpouseLabel).Where(__.Or(__.InV().HasID(id), __.OutV().HasID(id))).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.SpouseLabel, id)),
})
}
for id := range uuo.removedChildren {
tr := rv.Clone().InE(user.ParentLabel).Where(__.OtherV().HasID(id)).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.children {
v.AddE(user.ParentLabel).From(g.V(id)).InV()
constraints = append(constraints, &constraint{
pred: g.E().HasLabel(user.ParentLabel).OutV().HasID(id).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge(user.Label, user.ParentLabel, id)),
})
}
if uuo.clearedParent {
tr := rv.Clone().OutE(user.ParentLabel).Drop().Iterate()
trs = append(trs, tr)
}
for id := range uuo.parent {
v.AddE(user.ParentLabel).To(g.V(id)).OutV()
}
v.ValueMap(true)
if len(constraints) > 0 {
v = constraints[0].pred.Coalesce(constraints[0].test, v)
for _, cr := range constraints[1:] {
v = cr.pred.Coalesce(cr.test, v)
}
}
trs = append(trs, v)
return dsl.Join(trs...)
}