mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
881 lines
22 KiB
Go
881 lines
22 KiB
Go
// Copyright 2019-present Facebook Inc. 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"
|
|
"fmt"
|
|
|
|
"github.com/facebook/ent/dialect/sql"
|
|
"github.com/facebook/ent/dialect/sql/sqlgraph"
|
|
"github.com/facebook/ent/entc/integration/hooks/ent/card"
|
|
"github.com/facebook/ent/entc/integration/hooks/ent/predicate"
|
|
"github.com/facebook/ent/entc/integration/hooks/ent/user"
|
|
"github.com/facebook/ent/schema/field"
|
|
)
|
|
|
|
// UserUpdate is the builder for updating User entities.
|
|
type UserUpdate struct {
|
|
config
|
|
hooks []Hook
|
|
mutation *UserMutation
|
|
}
|
|
|
|
// Where adds a new predicate for the builder.
|
|
func (uu *UserUpdate) Where(ps ...predicate.User) *UserUpdate {
|
|
uu.mutation.predicates = append(uu.mutation.predicates, ps...)
|
|
return uu
|
|
}
|
|
|
|
// SetVersion sets the version field.
|
|
func (uu *UserUpdate) SetVersion(i int) *UserUpdate {
|
|
uu.mutation.ResetVersion()
|
|
uu.mutation.SetVersion(i)
|
|
return uu
|
|
}
|
|
|
|
// SetNillableVersion sets the version field if the given value is not nil.
|
|
func (uu *UserUpdate) SetNillableVersion(i *int) *UserUpdate {
|
|
if i != nil {
|
|
uu.SetVersion(*i)
|
|
}
|
|
return uu
|
|
}
|
|
|
|
// AddVersion adds i to version.
|
|
func (uu *UserUpdate) AddVersion(i int) *UserUpdate {
|
|
uu.mutation.AddVersion(i)
|
|
return uu
|
|
}
|
|
|
|
// SetName sets the name field.
|
|
func (uu *UserUpdate) SetName(s string) *UserUpdate {
|
|
uu.mutation.SetName(s)
|
|
return uu
|
|
}
|
|
|
|
// SetWorth sets the worth field.
|
|
func (uu *UserUpdate) SetWorth(u uint) *UserUpdate {
|
|
uu.mutation.ResetWorth()
|
|
uu.mutation.SetWorth(u)
|
|
return uu
|
|
}
|
|
|
|
// SetNillableWorth sets the worth field if the given value is not nil.
|
|
func (uu *UserUpdate) SetNillableWorth(u *uint) *UserUpdate {
|
|
if u != nil {
|
|
uu.SetWorth(*u)
|
|
}
|
|
return uu
|
|
}
|
|
|
|
// AddWorth adds u to worth.
|
|
func (uu *UserUpdate) AddWorth(u uint) *UserUpdate {
|
|
uu.mutation.AddWorth(u)
|
|
return uu
|
|
}
|
|
|
|
// ClearWorth clears the value of worth.
|
|
func (uu *UserUpdate) ClearWorth() *UserUpdate {
|
|
uu.mutation.ClearWorth()
|
|
return uu
|
|
}
|
|
|
|
// AddCardIDs adds the cards edge to Card by ids.
|
|
func (uu *UserUpdate) AddCardIDs(ids ...int) *UserUpdate {
|
|
uu.mutation.AddCardIDs(ids...)
|
|
return uu
|
|
}
|
|
|
|
// AddCards adds the cards edges to Card.
|
|
func (uu *UserUpdate) AddCards(c ...*Card) *UserUpdate {
|
|
ids := make([]int, len(c))
|
|
for i := range c {
|
|
ids[i] = c[i].ID
|
|
}
|
|
return uu.AddCardIDs(ids...)
|
|
}
|
|
|
|
// AddFriendIDs adds the friends edge to User by ids.
|
|
func (uu *UserUpdate) AddFriendIDs(ids ...int) *UserUpdate {
|
|
uu.mutation.AddFriendIDs(ids...)
|
|
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...)
|
|
}
|
|
|
|
// SetBestFriendID sets the best_friend edge to User by id.
|
|
func (uu *UserUpdate) SetBestFriendID(id int) *UserUpdate {
|
|
uu.mutation.SetBestFriendID(id)
|
|
return uu
|
|
}
|
|
|
|
// SetNillableBestFriendID sets the best_friend edge to User by id if the given value is not nil.
|
|
func (uu *UserUpdate) SetNillableBestFriendID(id *int) *UserUpdate {
|
|
if id != nil {
|
|
uu = uu.SetBestFriendID(*id)
|
|
}
|
|
return uu
|
|
}
|
|
|
|
// SetBestFriend sets the best_friend edge to User.
|
|
func (uu *UserUpdate) SetBestFriend(u *User) *UserUpdate {
|
|
return uu.SetBestFriendID(u.ID)
|
|
}
|
|
|
|
// Mutation returns the UserMutation object of the builder.
|
|
func (uu *UserUpdate) Mutation() *UserMutation {
|
|
return uu.mutation
|
|
}
|
|
|
|
// ClearCards clears all "cards" edges to type Card.
|
|
func (uu *UserUpdate) ClearCards() *UserUpdate {
|
|
uu.mutation.ClearCards()
|
|
return uu
|
|
}
|
|
|
|
// RemoveCardIDs removes the cards edge to Card by ids.
|
|
func (uu *UserUpdate) RemoveCardIDs(ids ...int) *UserUpdate {
|
|
uu.mutation.RemoveCardIDs(ids...)
|
|
return uu
|
|
}
|
|
|
|
// RemoveCards removes cards edges to Card.
|
|
func (uu *UserUpdate) RemoveCards(c ...*Card) *UserUpdate {
|
|
ids := make([]int, len(c))
|
|
for i := range c {
|
|
ids[i] = c[i].ID
|
|
}
|
|
return uu.RemoveCardIDs(ids...)
|
|
}
|
|
|
|
// ClearFriends clears all "friends" edges to type User.
|
|
func (uu *UserUpdate) ClearFriends() *UserUpdate {
|
|
uu.mutation.ClearFriends()
|
|
return uu
|
|
}
|
|
|
|
// RemoveFriendIDs removes the friends edge to User by ids.
|
|
func (uu *UserUpdate) RemoveFriendIDs(ids ...int) *UserUpdate {
|
|
uu.mutation.RemoveFriendIDs(ids...)
|
|
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...)
|
|
}
|
|
|
|
// ClearBestFriend clears the "best_friend" edge to type User.
|
|
func (uu *UserUpdate) ClearBestFriend() *UserUpdate {
|
|
uu.mutation.ClearBestFriend()
|
|
return uu
|
|
}
|
|
|
|
// Save executes the query and returns the number of nodes affected by the update operation.
|
|
func (uu *UserUpdate) Save(ctx context.Context) (int, error) {
|
|
var (
|
|
err error
|
|
affected int
|
|
)
|
|
if len(uu.hooks) == 0 {
|
|
affected, err = uu.sqlSave(ctx)
|
|
} else {
|
|
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
|
|
mutation, ok := m.(*UserMutation)
|
|
if !ok {
|
|
return nil, fmt.Errorf("unexpected mutation type %T", m)
|
|
}
|
|
uu.mutation = mutation
|
|
affected, err = uu.sqlSave(ctx)
|
|
mutation.done = true
|
|
return affected, err
|
|
})
|
|
for i := len(uu.hooks) - 1; i >= 0; i-- {
|
|
mut = uu.hooks[i](mut)
|
|
}
|
|
if _, err := mut.Mutate(ctx, uu.mutation); err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
return affected, err
|
|
}
|
|
|
|
// 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.mutation.predicates; len(ps) > 0 {
|
|
_spec.Predicate = func(selector *sql.Selector) {
|
|
for i := range ps {
|
|
ps[i](selector)
|
|
}
|
|
}
|
|
}
|
|
if value, ok := uu.mutation.Version(); ok {
|
|
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Value: value,
|
|
Column: user.FieldVersion,
|
|
})
|
|
}
|
|
if value, ok := uu.mutation.AddedVersion(); ok {
|
|
_spec.Fields.Add = append(_spec.Fields.Add, &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Value: value,
|
|
Column: user.FieldVersion,
|
|
})
|
|
}
|
|
if value, ok := uu.mutation.Name(); ok {
|
|
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeString,
|
|
Value: value,
|
|
Column: user.FieldName,
|
|
})
|
|
}
|
|
if value, ok := uu.mutation.Worth(); ok {
|
|
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeUint,
|
|
Value: value,
|
|
Column: user.FieldWorth,
|
|
})
|
|
}
|
|
if value, ok := uu.mutation.AddedWorth(); ok {
|
|
_spec.Fields.Add = append(_spec.Fields.Add, &sqlgraph.FieldSpec{
|
|
Type: field.TypeUint,
|
|
Value: value,
|
|
Column: user.FieldWorth,
|
|
})
|
|
}
|
|
if uu.mutation.WorthCleared() {
|
|
_spec.Fields.Clear = append(_spec.Fields.Clear, &sqlgraph.FieldSpec{
|
|
Type: field.TypeUint,
|
|
Column: user.FieldWorth,
|
|
})
|
|
}
|
|
if uu.mutation.CardsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.CardsTable,
|
|
Columns: []string{user.CardsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: card.FieldID,
|
|
},
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uu.mutation.RemovedCardsIDs(); len(nodes) > 0 && !uu.mutation.CardsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.CardsTable,
|
|
Columns: []string{user.CardsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: card.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uu.mutation.CardsIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.CardsTable,
|
|
Columns: []string{user.CardsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: card.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if uu.mutation.FriendsCleared() {
|
|
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,
|
|
},
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uu.mutation.RemovedFriendsIDs(); len(nodes) > 0 && !uu.mutation.FriendsCleared() {
|
|
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.mutation.FriendsIDs(); 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 uu.mutation.BestFriendCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2O,
|
|
Inverse: false,
|
|
Table: user.BestFriendTable,
|
|
Columns: []string{user.BestFriendColumn},
|
|
Bidi: true,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uu.mutation.BestFriendIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2O,
|
|
Inverse: false,
|
|
Table: user.BestFriendTable,
|
|
Columns: []string{user.BestFriendColumn},
|
|
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 n, err = sqlgraph.UpdateNodes(ctx, uu.driver, _spec); err != nil {
|
|
if _, ok := err.(*sqlgraph.NotFoundError); ok {
|
|
err = &NotFoundError{user.Label}
|
|
} else 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
|
|
hooks []Hook
|
|
mutation *UserMutation
|
|
}
|
|
|
|
// SetVersion sets the version field.
|
|
func (uuo *UserUpdateOne) SetVersion(i int) *UserUpdateOne {
|
|
uuo.mutation.ResetVersion()
|
|
uuo.mutation.SetVersion(i)
|
|
return uuo
|
|
}
|
|
|
|
// SetNillableVersion sets the version field if the given value is not nil.
|
|
func (uuo *UserUpdateOne) SetNillableVersion(i *int) *UserUpdateOne {
|
|
if i != nil {
|
|
uuo.SetVersion(*i)
|
|
}
|
|
return uuo
|
|
}
|
|
|
|
// AddVersion adds i to version.
|
|
func (uuo *UserUpdateOne) AddVersion(i int) *UserUpdateOne {
|
|
uuo.mutation.AddVersion(i)
|
|
return uuo
|
|
}
|
|
|
|
// SetName sets the name field.
|
|
func (uuo *UserUpdateOne) SetName(s string) *UserUpdateOne {
|
|
uuo.mutation.SetName(s)
|
|
return uuo
|
|
}
|
|
|
|
// SetWorth sets the worth field.
|
|
func (uuo *UserUpdateOne) SetWorth(u uint) *UserUpdateOne {
|
|
uuo.mutation.ResetWorth()
|
|
uuo.mutation.SetWorth(u)
|
|
return uuo
|
|
}
|
|
|
|
// SetNillableWorth sets the worth field if the given value is not nil.
|
|
func (uuo *UserUpdateOne) SetNillableWorth(u *uint) *UserUpdateOne {
|
|
if u != nil {
|
|
uuo.SetWorth(*u)
|
|
}
|
|
return uuo
|
|
}
|
|
|
|
// AddWorth adds u to worth.
|
|
func (uuo *UserUpdateOne) AddWorth(u uint) *UserUpdateOne {
|
|
uuo.mutation.AddWorth(u)
|
|
return uuo
|
|
}
|
|
|
|
// ClearWorth clears the value of worth.
|
|
func (uuo *UserUpdateOne) ClearWorth() *UserUpdateOne {
|
|
uuo.mutation.ClearWorth()
|
|
return uuo
|
|
}
|
|
|
|
// AddCardIDs adds the cards edge to Card by ids.
|
|
func (uuo *UserUpdateOne) AddCardIDs(ids ...int) *UserUpdateOne {
|
|
uuo.mutation.AddCardIDs(ids...)
|
|
return uuo
|
|
}
|
|
|
|
// AddCards adds the cards edges to Card.
|
|
func (uuo *UserUpdateOne) AddCards(c ...*Card) *UserUpdateOne {
|
|
ids := make([]int, len(c))
|
|
for i := range c {
|
|
ids[i] = c[i].ID
|
|
}
|
|
return uuo.AddCardIDs(ids...)
|
|
}
|
|
|
|
// AddFriendIDs adds the friends edge to User by ids.
|
|
func (uuo *UserUpdateOne) AddFriendIDs(ids ...int) *UserUpdateOne {
|
|
uuo.mutation.AddFriendIDs(ids...)
|
|
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...)
|
|
}
|
|
|
|
// SetBestFriendID sets the best_friend edge to User by id.
|
|
func (uuo *UserUpdateOne) SetBestFriendID(id int) *UserUpdateOne {
|
|
uuo.mutation.SetBestFriendID(id)
|
|
return uuo
|
|
}
|
|
|
|
// SetNillableBestFriendID sets the best_friend edge to User by id if the given value is not nil.
|
|
func (uuo *UserUpdateOne) SetNillableBestFriendID(id *int) *UserUpdateOne {
|
|
if id != nil {
|
|
uuo = uuo.SetBestFriendID(*id)
|
|
}
|
|
return uuo
|
|
}
|
|
|
|
// SetBestFriend sets the best_friend edge to User.
|
|
func (uuo *UserUpdateOne) SetBestFriend(u *User) *UserUpdateOne {
|
|
return uuo.SetBestFriendID(u.ID)
|
|
}
|
|
|
|
// Mutation returns the UserMutation object of the builder.
|
|
func (uuo *UserUpdateOne) Mutation() *UserMutation {
|
|
return uuo.mutation
|
|
}
|
|
|
|
// ClearCards clears all "cards" edges to type Card.
|
|
func (uuo *UserUpdateOne) ClearCards() *UserUpdateOne {
|
|
uuo.mutation.ClearCards()
|
|
return uuo
|
|
}
|
|
|
|
// RemoveCardIDs removes the cards edge to Card by ids.
|
|
func (uuo *UserUpdateOne) RemoveCardIDs(ids ...int) *UserUpdateOne {
|
|
uuo.mutation.RemoveCardIDs(ids...)
|
|
return uuo
|
|
}
|
|
|
|
// RemoveCards removes cards edges to Card.
|
|
func (uuo *UserUpdateOne) RemoveCards(c ...*Card) *UserUpdateOne {
|
|
ids := make([]int, len(c))
|
|
for i := range c {
|
|
ids[i] = c[i].ID
|
|
}
|
|
return uuo.RemoveCardIDs(ids...)
|
|
}
|
|
|
|
// ClearFriends clears all "friends" edges to type User.
|
|
func (uuo *UserUpdateOne) ClearFriends() *UserUpdateOne {
|
|
uuo.mutation.ClearFriends()
|
|
return uuo
|
|
}
|
|
|
|
// RemoveFriendIDs removes the friends edge to User by ids.
|
|
func (uuo *UserUpdateOne) RemoveFriendIDs(ids ...int) *UserUpdateOne {
|
|
uuo.mutation.RemoveFriendIDs(ids...)
|
|
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...)
|
|
}
|
|
|
|
// ClearBestFriend clears the "best_friend" edge to type User.
|
|
func (uuo *UserUpdateOne) ClearBestFriend() *UserUpdateOne {
|
|
uuo.mutation.ClearBestFriend()
|
|
return uuo
|
|
}
|
|
|
|
// Save executes the query and returns the updated entity.
|
|
func (uuo *UserUpdateOne) Save(ctx context.Context) (*User, error) {
|
|
var (
|
|
err error
|
|
node *User
|
|
)
|
|
if len(uuo.hooks) == 0 {
|
|
node, err = uuo.sqlSave(ctx)
|
|
} else {
|
|
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
|
|
mutation, ok := m.(*UserMutation)
|
|
if !ok {
|
|
return nil, fmt.Errorf("unexpected mutation type %T", m)
|
|
}
|
|
uuo.mutation = mutation
|
|
node, err = uuo.sqlSave(ctx)
|
|
mutation.done = true
|
|
return node, err
|
|
})
|
|
for i := len(uuo.hooks) - 1; i >= 0; i-- {
|
|
mut = uuo.hooks[i](mut)
|
|
}
|
|
if _, err := mut.Mutate(ctx, uuo.mutation); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return node, err
|
|
}
|
|
|
|
// SaveX is like Save, but panics if an error occurs.
|
|
func (uuo *UserUpdateOne) SaveX(ctx context.Context) *User {
|
|
node, err := uuo.Save(ctx)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return node
|
|
}
|
|
|
|
// 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) (_node *User, err error) {
|
|
_spec := &sqlgraph.UpdateSpec{
|
|
Node: &sqlgraph.NodeSpec{
|
|
Table: user.Table,
|
|
Columns: user.Columns,
|
|
ID: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
id, ok := uuo.mutation.ID()
|
|
if !ok {
|
|
return nil, &ValidationError{Name: "ID", err: fmt.Errorf("missing User.ID for update")}
|
|
}
|
|
_spec.Node.ID.Value = id
|
|
if value, ok := uuo.mutation.Version(); ok {
|
|
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Value: value,
|
|
Column: user.FieldVersion,
|
|
})
|
|
}
|
|
if value, ok := uuo.mutation.AddedVersion(); ok {
|
|
_spec.Fields.Add = append(_spec.Fields.Add, &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Value: value,
|
|
Column: user.FieldVersion,
|
|
})
|
|
}
|
|
if value, ok := uuo.mutation.Name(); ok {
|
|
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeString,
|
|
Value: value,
|
|
Column: user.FieldName,
|
|
})
|
|
}
|
|
if value, ok := uuo.mutation.Worth(); ok {
|
|
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
|
|
Type: field.TypeUint,
|
|
Value: value,
|
|
Column: user.FieldWorth,
|
|
})
|
|
}
|
|
if value, ok := uuo.mutation.AddedWorth(); ok {
|
|
_spec.Fields.Add = append(_spec.Fields.Add, &sqlgraph.FieldSpec{
|
|
Type: field.TypeUint,
|
|
Value: value,
|
|
Column: user.FieldWorth,
|
|
})
|
|
}
|
|
if uuo.mutation.WorthCleared() {
|
|
_spec.Fields.Clear = append(_spec.Fields.Clear, &sqlgraph.FieldSpec{
|
|
Type: field.TypeUint,
|
|
Column: user.FieldWorth,
|
|
})
|
|
}
|
|
if uuo.mutation.CardsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.CardsTable,
|
|
Columns: []string{user.CardsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: card.FieldID,
|
|
},
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uuo.mutation.RemovedCardsIDs(); len(nodes) > 0 && !uuo.mutation.CardsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.CardsTable,
|
|
Columns: []string{user.CardsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: card.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uuo.mutation.CardsIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.CardsTable,
|
|
Columns: []string{user.CardsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: card.FieldID,
|
|
},
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if uuo.mutation.FriendsCleared() {
|
|
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,
|
|
},
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uuo.mutation.RemovedFriendsIDs(); len(nodes) > 0 && !uuo.mutation.FriendsCleared() {
|
|
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.mutation.FriendsIDs(); 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 uuo.mutation.BestFriendCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2O,
|
|
Inverse: false,
|
|
Table: user.BestFriendTable,
|
|
Columns: []string{user.BestFriendColumn},
|
|
Bidi: true,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: &sqlgraph.FieldSpec{
|
|
Type: field.TypeInt,
|
|
Column: user.FieldID,
|
|
},
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := uuo.mutation.BestFriendIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2O,
|
|
Inverse: false,
|
|
Table: user.BestFriendTable,
|
|
Columns: []string{user.BestFriendColumn},
|
|
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)
|
|
}
|
|
_node = &User{config: uuo.config}
|
|
_spec.Assign = _node.assignValues
|
|
_spec.ScanValues = _node.scanValues
|
|
if err = sqlgraph.UpdateNode(ctx, uuo.driver, _spec); err != nil {
|
|
if _, ok := err.(*sqlgraph.NotFoundError); ok {
|
|
err = &NotFoundError{user.Label}
|
|
} else if cerr, ok := isSQLConstraintError(err); ok {
|
|
err = cerr
|
|
}
|
|
return nil, err
|
|
}
|
|
return _node, nil
|
|
}
|