mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
877 lines
21 KiB
Go
877 lines
21 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 by entc, DO NOT EDIT.
|
|
|
|
package ent
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/facebookincubator/ent/dialect/sql"
|
|
"github.com/facebookincubator/ent/dialect/sql/sqlgraph"
|
|
"github.com/facebookincubator/ent/examples/traversal/ent/group"
|
|
"github.com/facebookincubator/ent/examples/traversal/ent/pet"
|
|
"github.com/facebookincubator/ent/examples/traversal/ent/predicate"
|
|
"github.com/facebookincubator/ent/examples/traversal/ent/user"
|
|
"github.com/facebookincubator/ent/schema/field"
|
|
)
|
|
|
|
// UserUpdate is the builder for updating User entities.
|
|
type UserUpdate struct {
|
|
config
|
|
age *int
|
|
addage *int
|
|
name *string
|
|
pets map[int]struct{}
|
|
friends map[int]struct{}
|
|
groups map[int]struct{}
|
|
manage map[int]struct{}
|
|
removedPets map[int]struct{}
|
|
removedFriends map[int]struct{}
|
|
removedGroups map[int]struct{}
|
|
removedManage map[int]struct{}
|
|
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
|
|
}
|
|
|
|
// AddPetIDs adds the pets edge to Pet by ids.
|
|
func (uu *UserUpdate) AddPetIDs(ids ...int) *UserUpdate {
|
|
if uu.pets == nil {
|
|
uu.pets = make(map[int]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([]int, len(p))
|
|
for i := range p {
|
|
ids[i] = p[i].ID
|
|
}
|
|
return uu.AddPetIDs(ids...)
|
|
}
|
|
|
|
// AddFriendIDs adds the friends edge to User by ids.
|
|
func (uu *UserUpdate) AddFriendIDs(ids ...int) *UserUpdate {
|
|
if uu.friends == nil {
|
|
uu.friends = make(map[int]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([]int, len(u))
|
|
for i := range u {
|
|
ids[i] = u[i].ID
|
|
}
|
|
return uu.AddFriendIDs(ids...)
|
|
}
|
|
|
|
// AddGroupIDs adds the groups edge to Group by ids.
|
|
func (uu *UserUpdate) AddGroupIDs(ids ...int) *UserUpdate {
|
|
if uu.groups == nil {
|
|
uu.groups = make(map[int]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([]int, len(g))
|
|
for i := range g {
|
|
ids[i] = g[i].ID
|
|
}
|
|
return uu.AddGroupIDs(ids...)
|
|
}
|
|
|
|
// AddManageIDs adds the manage edge to Group by ids.
|
|
func (uu *UserUpdate) AddManageIDs(ids ...int) *UserUpdate {
|
|
if uu.manage == nil {
|
|
uu.manage = make(map[int]struct{})
|
|
}
|
|
for i := range ids {
|
|
uu.manage[ids[i]] = struct{}{}
|
|
}
|
|
return uu
|
|
}
|
|
|
|
// AddManage adds the manage edges to Group.
|
|
func (uu *UserUpdate) AddManage(g ...*Group) *UserUpdate {
|
|
ids := make([]int, len(g))
|
|
for i := range g {
|
|
ids[i] = g[i].ID
|
|
}
|
|
return uu.AddManageIDs(ids...)
|
|
}
|
|
|
|
// RemovePetIDs removes the pets edge to Pet by ids.
|
|
func (uu *UserUpdate) RemovePetIDs(ids ...int) *UserUpdate {
|
|
if uu.removedPets == nil {
|
|
uu.removedPets = make(map[int]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([]int, len(p))
|
|
for i := range p {
|
|
ids[i] = p[i].ID
|
|
}
|
|
return uu.RemovePetIDs(ids...)
|
|
}
|
|
|
|
// RemoveFriendIDs removes the friends edge to User by ids.
|
|
func (uu *UserUpdate) RemoveFriendIDs(ids ...int) *UserUpdate {
|
|
if uu.removedFriends == nil {
|
|
uu.removedFriends = make(map[int]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([]int, len(u))
|
|
for i := range u {
|
|
ids[i] = u[i].ID
|
|
}
|
|
return uu.RemoveFriendIDs(ids...)
|
|
}
|
|
|
|
// RemoveGroupIDs removes the groups edge to Group by ids.
|
|
func (uu *UserUpdate) RemoveGroupIDs(ids ...int) *UserUpdate {
|
|
if uu.removedGroups == nil {
|
|
uu.removedGroups = make(map[int]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([]int, len(g))
|
|
for i := range g {
|
|
ids[i] = g[i].ID
|
|
}
|
|
return uu.RemoveGroupIDs(ids...)
|
|
}
|
|
|
|
// RemoveManageIDs removes the manage edge to Group by ids.
|
|
func (uu *UserUpdate) RemoveManageIDs(ids ...int) *UserUpdate {
|
|
if uu.removedManage == nil {
|
|
uu.removedManage = make(map[int]struct{})
|
|
}
|
|
for i := range ids {
|
|
uu.removedManage[ids[i]] = struct{}{}
|
|
}
|
|
return uu
|
|
}
|
|
|
|
// RemoveManage removes manage edges to Group.
|
|
func (uu *UserUpdate) RemoveManage(g ...*Group) *UserUpdate {
|
|
ids := make([]int, len(g))
|
|
for i := range g {
|
|
ids[i] = g[i].ID
|
|
}
|
|
return uu.RemoveManageIDs(ids...)
|
|
}
|
|
|
|
// Save executes the query and returns the number of rows/vertices matched by this operation.
|
|
func (uu *UserUpdate) Save(ctx context.Context) (int, error) {
|
|
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) {
|
|
spec := &sqlgraph.UpdateSpec{
|
|
Node: &sqlgraph.NodeSpec{
|
|
Table: user.Table,
|
|
Columns: user.Columns,
|
|
ID: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
if ps := uu.predicates; len(ps) > 0 {
|
|
spec.Predicate = func(selector *sql.Selector) {
|
|
for i := range ps {
|
|
ps[i](selector)
|
|
}
|
|
}
|
|
}
|
|
if value := uu.age; value != nil {
|
|
spec.Fields.Set = append(spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Value: *value,
|
|
Column: user.FieldAge,
|
|
})
|
|
}
|
|
if value := uu.addage; value != nil {
|
|
spec.Fields.Add = append(spec.Fields.Add, &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Value: *value,
|
|
Column: user.FieldAge,
|
|
})
|
|
}
|
|
if value := uu.name; value != nil {
|
|
spec.Fields.Set = append(spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeString,
|
|
Value: *value,
|
|
Column: user.FieldName,
|
|
})
|
|
}
|
|
if nodes := uu.removedPets; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.PetsTable,
|
|
Columns: []string{user.PetsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: pet.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uu.pets; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.PetsTable,
|
|
Columns: []string{user.PetsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: pet.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Add = append(spec.Edges.Add, edge)
|
|
}
|
|
if nodes := uu.removedFriends; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.FriendsTable,
|
|
Columns: user.FriendsPrimaryKey,
|
|
Bidi: true,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uu.friends; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.FriendsTable,
|
|
Columns: user.FriendsPrimaryKey,
|
|
Bidi: true,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Add = append(spec.Edges.Add, edge)
|
|
}
|
|
if nodes := uu.removedGroups; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: true,
|
|
Table: user.GroupsTable,
|
|
Columns: user.GroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: group.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uu.groups; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: true,
|
|
Table: user.GroupsTable,
|
|
Columns: user.GroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: group.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Add = append(spec.Edges.Add, edge)
|
|
}
|
|
if nodes := uu.removedManage; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: true,
|
|
Table: user.ManageTable,
|
|
Columns: []string{user.ManageColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: group.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uu.manage; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: true,
|
|
Table: user.ManageTable,
|
|
Columns: []string{user.ManageColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: group.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Add = append(spec.Edges.Add, edge)
|
|
}
|
|
if n, err = sqlgraph.UpdateNodes(ctx, uu.driver, spec); err != nil {
|
|
if cerr, ok := isSQLConstraintError(err); ok {
|
|
err = cerr
|
|
}
|
|
return 0, err
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
// UserUpdateOne is the builder for updating a single User entity.
|
|
type UserUpdateOne struct {
|
|
config
|
|
id int
|
|
age *int
|
|
addage *int
|
|
name *string
|
|
pets map[int]struct{}
|
|
friends map[int]struct{}
|
|
groups map[int]struct{}
|
|
manage map[int]struct{}
|
|
removedPets map[int]struct{}
|
|
removedFriends map[int]struct{}
|
|
removedGroups map[int]struct{}
|
|
removedManage map[int]struct{}
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
// AddPetIDs adds the pets edge to Pet by ids.
|
|
func (uuo *UserUpdateOne) AddPetIDs(ids ...int) *UserUpdateOne {
|
|
if uuo.pets == nil {
|
|
uuo.pets = make(map[int]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([]int, len(p))
|
|
for i := range p {
|
|
ids[i] = p[i].ID
|
|
}
|
|
return uuo.AddPetIDs(ids...)
|
|
}
|
|
|
|
// AddFriendIDs adds the friends edge to User by ids.
|
|
func (uuo *UserUpdateOne) AddFriendIDs(ids ...int) *UserUpdateOne {
|
|
if uuo.friends == nil {
|
|
uuo.friends = make(map[int]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([]int, len(u))
|
|
for i := range u {
|
|
ids[i] = u[i].ID
|
|
}
|
|
return uuo.AddFriendIDs(ids...)
|
|
}
|
|
|
|
// AddGroupIDs adds the groups edge to Group by ids.
|
|
func (uuo *UserUpdateOne) AddGroupIDs(ids ...int) *UserUpdateOne {
|
|
if uuo.groups == nil {
|
|
uuo.groups = make(map[int]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([]int, len(g))
|
|
for i := range g {
|
|
ids[i] = g[i].ID
|
|
}
|
|
return uuo.AddGroupIDs(ids...)
|
|
}
|
|
|
|
// AddManageIDs adds the manage edge to Group by ids.
|
|
func (uuo *UserUpdateOne) AddManageIDs(ids ...int) *UserUpdateOne {
|
|
if uuo.manage == nil {
|
|
uuo.manage = make(map[int]struct{})
|
|
}
|
|
for i := range ids {
|
|
uuo.manage[ids[i]] = struct{}{}
|
|
}
|
|
return uuo
|
|
}
|
|
|
|
// AddManage adds the manage edges to Group.
|
|
func (uuo *UserUpdateOne) AddManage(g ...*Group) *UserUpdateOne {
|
|
ids := make([]int, len(g))
|
|
for i := range g {
|
|
ids[i] = g[i].ID
|
|
}
|
|
return uuo.AddManageIDs(ids...)
|
|
}
|
|
|
|
// RemovePetIDs removes the pets edge to Pet by ids.
|
|
func (uuo *UserUpdateOne) RemovePetIDs(ids ...int) *UserUpdateOne {
|
|
if uuo.removedPets == nil {
|
|
uuo.removedPets = make(map[int]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([]int, len(p))
|
|
for i := range p {
|
|
ids[i] = p[i].ID
|
|
}
|
|
return uuo.RemovePetIDs(ids...)
|
|
}
|
|
|
|
// RemoveFriendIDs removes the friends edge to User by ids.
|
|
func (uuo *UserUpdateOne) RemoveFriendIDs(ids ...int) *UserUpdateOne {
|
|
if uuo.removedFriends == nil {
|
|
uuo.removedFriends = make(map[int]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([]int, len(u))
|
|
for i := range u {
|
|
ids[i] = u[i].ID
|
|
}
|
|
return uuo.RemoveFriendIDs(ids...)
|
|
}
|
|
|
|
// RemoveGroupIDs removes the groups edge to Group by ids.
|
|
func (uuo *UserUpdateOne) RemoveGroupIDs(ids ...int) *UserUpdateOne {
|
|
if uuo.removedGroups == nil {
|
|
uuo.removedGroups = make(map[int]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([]int, len(g))
|
|
for i := range g {
|
|
ids[i] = g[i].ID
|
|
}
|
|
return uuo.RemoveGroupIDs(ids...)
|
|
}
|
|
|
|
// RemoveManageIDs removes the manage edge to Group by ids.
|
|
func (uuo *UserUpdateOne) RemoveManageIDs(ids ...int) *UserUpdateOne {
|
|
if uuo.removedManage == nil {
|
|
uuo.removedManage = make(map[int]struct{})
|
|
}
|
|
for i := range ids {
|
|
uuo.removedManage[ids[i]] = struct{}{}
|
|
}
|
|
return uuo
|
|
}
|
|
|
|
// RemoveManage removes manage edges to Group.
|
|
func (uuo *UserUpdateOne) RemoveManage(g ...*Group) *UserUpdateOne {
|
|
ids := make([]int, len(g))
|
|
for i := range g {
|
|
ids[i] = g[i].ID
|
|
}
|
|
return uuo.RemoveManageIDs(ids...)
|
|
}
|
|
|
|
// Save executes the query and returns the updated entity.
|
|
func (uuo *UserUpdateOne) Save(ctx context.Context) (*User, error) {
|
|
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) {
|
|
spec := &sqlgraph.UpdateSpec{
|
|
Node: &sqlgraph.NodeSpec{
|
|
Table: user.Table,
|
|
Columns: user.Columns,
|
|
ID: &sqlgraph.FieldSpec{
|
|
Value: uuo.id,
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
if value := uuo.age; value != nil {
|
|
spec.Fields.Set = append(spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Value: *value,
|
|
Column: user.FieldAge,
|
|
})
|
|
}
|
|
if value := uuo.addage; value != nil {
|
|
spec.Fields.Add = append(spec.Fields.Add, &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Value: *value,
|
|
Column: user.FieldAge,
|
|
})
|
|
}
|
|
if value := uuo.name; value != nil {
|
|
spec.Fields.Set = append(spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeString,
|
|
Value: *value,
|
|
Column: user.FieldName,
|
|
})
|
|
}
|
|
if nodes := uuo.removedPets; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.PetsTable,
|
|
Columns: []string{user.PetsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: pet.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uuo.pets; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.PetsTable,
|
|
Columns: []string{user.PetsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: pet.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Add = append(spec.Edges.Add, edge)
|
|
}
|
|
if nodes := uuo.removedFriends; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.FriendsTable,
|
|
Columns: user.FriendsPrimaryKey,
|
|
Bidi: true,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uuo.friends; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.FriendsTable,
|
|
Columns: user.FriendsPrimaryKey,
|
|
Bidi: true,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Add = append(spec.Edges.Add, edge)
|
|
}
|
|
if nodes := uuo.removedGroups; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: true,
|
|
Table: user.GroupsTable,
|
|
Columns: user.GroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: group.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uuo.groups; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: true,
|
|
Table: user.GroupsTable,
|
|
Columns: user.GroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: group.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Add = append(spec.Edges.Add, edge)
|
|
}
|
|
if nodes := uuo.removedManage; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: true,
|
|
Table: user.ManageTable,
|
|
Columns: []string{user.ManageColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: group.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Clear = append(spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uuo.manage; len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: true,
|
|
Table: user.ManageTable,
|
|
Columns: []string{user.ManageColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: group.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for k, _ := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
spec.Edges.Add = append(spec.Edges.Add, edge)
|
|
}
|
|
u = &User{config: uuo.config}
|
|
spec.Assign = u.assignValues
|
|
spec.ScanValues = u.scanValues()
|
|
if err = sqlgraph.UpdateNode(ctx, uuo.driver, spec); err != nil {
|
|
if cerr, ok := isSQLConstraintError(err); ok {
|
|
err = cerr
|
|
}
|
|
return nil, err
|
|
}
|
|
return u, nil
|
|
}
|