mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
2216 lines
58 KiB
Go
2216 lines
58 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/dialect/sql"
|
|
"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"
|
|
)
|
|
|
|
// UserUpdate is the builder for updating User entities.
|
|
type UserUpdate struct {
|
|
config
|
|
age *int
|
|
addage *int
|
|
name *string
|
|
last *string
|
|
nickname *string
|
|
clearnickname bool
|
|
phone *string
|
|
clearphone bool
|
|
password *string
|
|
clearpassword bool
|
|
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
|
|
uu.addage = nil
|
|
return uu
|
|
}
|
|
|
|
// AddAge adds i to age.
|
|
func (uu *UserUpdate) AddAge(i int) *UserUpdate {
|
|
if uu.addage == nil {
|
|
uu.addage = &i
|
|
} else {
|
|
*uu.addage += 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
|
|
}
|
|
|
|
// ClearNickname clears the value of nickname.
|
|
func (uu *UserUpdate) ClearNickname() *UserUpdate {
|
|
uu.nickname = nil
|
|
uu.clearnickname = true
|
|
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
|
|
}
|
|
|
|
// ClearPhone clears the value of phone.
|
|
func (uu *UserUpdate) ClearPhone() *UserUpdate {
|
|
uu.phone = nil
|
|
uu.clearphone = true
|
|
return uu
|
|
}
|
|
|
|
// SetPassword sets the password field.
|
|
func (uu *UserUpdate) SetPassword(s string) *UserUpdate {
|
|
uu.password = &s
|
|
return uu
|
|
}
|
|
|
|
// SetNillablePassword sets the password field if the given value is not nil.
|
|
func (uu *UserUpdate) SetNillablePassword(s *string) *UserUpdate {
|
|
if s != nil {
|
|
uu.SetPassword(*s)
|
|
}
|
|
return uu
|
|
}
|
|
|
|
// ClearPassword clears the value of password.
|
|
func (uu *UserUpdate) ClearPassword() *UserUpdate {
|
|
uu.password = nil
|
|
uu.clearpassword = true
|
|
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\"")
|
|
}
|
|
return uu.sqlSave(ctx)
|
|
}
|
|
|
|
// 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) {
|
|
var (
|
|
builder = sql.Dialect(uu.driver.Dialect())
|
|
selector = builder.Select(user.FieldID).From(builder.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 (
|
|
res sql.Result
|
|
updater = builder.Update(user.Table)
|
|
)
|
|
updater = updater.Where(sql.InInts(user.FieldID, ids...))
|
|
if value := uu.age; value != nil {
|
|
updater.Set(user.FieldAge, *value)
|
|
}
|
|
if value := uu.addage; value != nil {
|
|
updater.Add(user.FieldAge, *value)
|
|
}
|
|
if value := uu.name; value != nil {
|
|
updater.Set(user.FieldName, *value)
|
|
}
|
|
if value := uu.last; value != nil {
|
|
updater.Set(user.FieldLast, *value)
|
|
}
|
|
if value := uu.nickname; value != nil {
|
|
updater.Set(user.FieldNickname, *value)
|
|
}
|
|
if uu.clearnickname {
|
|
updater.SetNull(user.FieldNickname)
|
|
}
|
|
if value := uu.phone; value != nil {
|
|
updater.Set(user.FieldPhone, *value)
|
|
}
|
|
if uu.clearphone {
|
|
updater.SetNull(user.FieldPhone)
|
|
}
|
|
if value := uu.password; value != nil {
|
|
updater.Set(user.FieldPassword, *value)
|
|
}
|
|
if uu.clearpassword {
|
|
updater.SetNull(user.FieldPassword)
|
|
}
|
|
if !updater.Empty() {
|
|
query, args := updater.Query()
|
|
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
|
return 0, rollback(tx, err)
|
|
}
|
|
}
|
|
if uu.clearedCard {
|
|
query, args := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 = builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 = builder.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 := builder.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 = builder.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 := builder.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 := builder.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 := builder.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 := builder.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
|
|
}
|
|
|
|
// UserUpdateOne is the builder for updating a single User entity.
|
|
type UserUpdateOne struct {
|
|
config
|
|
id string
|
|
age *int
|
|
addage *int
|
|
name *string
|
|
last *string
|
|
nickname *string
|
|
clearnickname bool
|
|
phone *string
|
|
clearphone bool
|
|
password *string
|
|
clearpassword bool
|
|
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
|
|
uuo.addage = nil
|
|
return uuo
|
|
}
|
|
|
|
// AddAge adds i to age.
|
|
func (uuo *UserUpdateOne) AddAge(i int) *UserUpdateOne {
|
|
if uuo.addage == nil {
|
|
uuo.addage = &i
|
|
} else {
|
|
*uuo.addage += 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
|
|
}
|
|
|
|
// ClearNickname clears the value of nickname.
|
|
func (uuo *UserUpdateOne) ClearNickname() *UserUpdateOne {
|
|
uuo.nickname = nil
|
|
uuo.clearnickname = true
|
|
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
|
|
}
|
|
|
|
// ClearPhone clears the value of phone.
|
|
func (uuo *UserUpdateOne) ClearPhone() *UserUpdateOne {
|
|
uuo.phone = nil
|
|
uuo.clearphone = true
|
|
return uuo
|
|
}
|
|
|
|
// SetPassword sets the password field.
|
|
func (uuo *UserUpdateOne) SetPassword(s string) *UserUpdateOne {
|
|
uuo.password = &s
|
|
return uuo
|
|
}
|
|
|
|
// SetNillablePassword sets the password field if the given value is not nil.
|
|
func (uuo *UserUpdateOne) SetNillablePassword(s *string) *UserUpdateOne {
|
|
if s != nil {
|
|
uuo.SetPassword(*s)
|
|
}
|
|
return uuo
|
|
}
|
|
|
|
// ClearPassword clears the value of password.
|
|
func (uuo *UserUpdateOne) ClearPassword() *UserUpdateOne {
|
|
uuo.password = nil
|
|
uuo.clearpassword = true
|
|
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\"")
|
|
}
|
|
return uuo.sqlSave(ctx)
|
|
}
|
|
|
|
// 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) {
|
|
var (
|
|
builder = sql.Dialect(uuo.driver.Dialect())
|
|
selector = builder.Select(user.Columns...).From(builder.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, &ErrNotFound{fmt.Sprintf("User 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 (
|
|
res sql.Result
|
|
updater = builder.Update(user.Table)
|
|
)
|
|
updater = updater.Where(sql.InInts(user.FieldID, ids...))
|
|
if value := uuo.age; value != nil {
|
|
updater.Set(user.FieldAge, *value)
|
|
u.Age = *value
|
|
}
|
|
if value := uuo.addage; value != nil {
|
|
updater.Add(user.FieldAge, *value)
|
|
u.Age += *value
|
|
}
|
|
if value := uuo.name; value != nil {
|
|
updater.Set(user.FieldName, *value)
|
|
u.Name = *value
|
|
}
|
|
if value := uuo.last; value != nil {
|
|
updater.Set(user.FieldLast, *value)
|
|
u.Last = *value
|
|
}
|
|
if value := uuo.nickname; value != nil {
|
|
updater.Set(user.FieldNickname, *value)
|
|
u.Nickname = *value
|
|
}
|
|
if uuo.clearnickname {
|
|
var value string
|
|
u.Nickname = value
|
|
updater.SetNull(user.FieldNickname)
|
|
}
|
|
if value := uuo.phone; value != nil {
|
|
updater.Set(user.FieldPhone, *value)
|
|
u.Phone = *value
|
|
}
|
|
if uuo.clearphone {
|
|
var value string
|
|
u.Phone = value
|
|
updater.SetNull(user.FieldPhone)
|
|
}
|
|
if value := uuo.password; value != nil {
|
|
updater.Set(user.FieldPassword, *value)
|
|
u.Password = *value
|
|
}
|
|
if uuo.clearpassword {
|
|
var value string
|
|
u.Password = value
|
|
updater.SetNull(user.FieldPassword)
|
|
}
|
|
if !updater.Empty() {
|
|
query, args := updater.Query()
|
|
if err := tx.Exec(ctx, query, args, &res); err != nil {
|
|
return nil, rollback(tx, err)
|
|
}
|
|
}
|
|
if uuo.clearedCard {
|
|
query, args := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 = builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 := builder.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 = builder.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 := builder.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 = builder.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 := builder.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 := builder.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 := builder.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 := builder.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
|
|
}
|