// Code generated by ent, DO NOT EDIT. package ent import ( "context" "errors" "fmt" "entgo.io/ent/dialect/sql" "entgo.io/ent/dialect/sql/sqlgraph" "entgo.io/ent/schema/field" "github.com/xmdhs/authlib-skin/db/ent/predicate" "github.com/xmdhs/authlib-skin/db/ent/skin" "github.com/xmdhs/authlib-skin/db/ent/user" "github.com/xmdhs/authlib-skin/db/ent/userprofile" ) // UserUpdate is the builder for updating User entities. type UserUpdate struct { config hooks []Hook mutation *UserMutation } // Where appends a list predicates to the UserUpdate builder. func (uu *UserUpdate) Where(ps ...predicate.User) *UserUpdate { uu.mutation.Where(ps...) return uu } // SetEmail sets the "email" field. func (uu *UserUpdate) SetEmail(s string) *UserUpdate { uu.mutation.SetEmail(s) return uu } // SetPassword sets the "password" field. func (uu *UserUpdate) SetPassword(s string) *UserUpdate { uu.mutation.SetPassword(s) return uu } // SetSalt sets the "salt" field. func (uu *UserUpdate) SetSalt(s string) *UserUpdate { uu.mutation.SetSalt(s) return uu } // SetState sets the "state" field. func (uu *UserUpdate) SetState(i int) *UserUpdate { uu.mutation.ResetState() uu.mutation.SetState(i) return uu } // AddState adds i to the "state" field. func (uu *UserUpdate) AddState(i int) *UserUpdate { uu.mutation.AddState(i) return uu } // SetRegTime sets the "reg_time" field. func (uu *UserUpdate) SetRegTime(i int64) *UserUpdate { uu.mutation.ResetRegTime() uu.mutation.SetRegTime(i) return uu } // AddRegTime adds i to the "reg_time" field. func (uu *UserUpdate) AddRegTime(i int64) *UserUpdate { uu.mutation.AddRegTime(i) return uu } // AddSkinIDs adds the "skin" edge to the Skin entity by IDs. func (uu *UserUpdate) AddSkinIDs(ids ...int) *UserUpdate { uu.mutation.AddSkinIDs(ids...) return uu } // AddSkin adds the "skin" edges to the Skin entity. func (uu *UserUpdate) AddSkin(s ...*Skin) *UserUpdate { ids := make([]int, len(s)) for i := range s { ids[i] = s[i].ID } return uu.AddSkinIDs(ids...) } // SetProfileID sets the "profile" edge to the UserProfile entity by ID. func (uu *UserUpdate) SetProfileID(id int) *UserUpdate { uu.mutation.SetProfileID(id) return uu } // SetNillableProfileID sets the "profile" edge to the UserProfile entity by ID if the given value is not nil. func (uu *UserUpdate) SetNillableProfileID(id *int) *UserUpdate { if id != nil { uu = uu.SetProfileID(*id) } return uu } // SetProfile sets the "profile" edge to the UserProfile entity. func (uu *UserUpdate) SetProfile(u *UserProfile) *UserUpdate { return uu.SetProfileID(u.ID) } // Mutation returns the UserMutation object of the builder. func (uu *UserUpdate) Mutation() *UserMutation { return uu.mutation } // ClearSkin clears all "skin" edges to the Skin entity. func (uu *UserUpdate) ClearSkin() *UserUpdate { uu.mutation.ClearSkin() return uu } // RemoveSkinIDs removes the "skin" edge to Skin entities by IDs. func (uu *UserUpdate) RemoveSkinIDs(ids ...int) *UserUpdate { uu.mutation.RemoveSkinIDs(ids...) return uu } // RemoveSkin removes "skin" edges to Skin entities. func (uu *UserUpdate) RemoveSkin(s ...*Skin) *UserUpdate { ids := make([]int, len(s)) for i := range s { ids[i] = s[i].ID } return uu.RemoveSkinIDs(ids...) } // ClearProfile clears the "profile" edge to the UserProfile entity. func (uu *UserUpdate) ClearProfile() *UserUpdate { uu.mutation.ClearProfile() 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) { return withHooks(ctx, uu.sqlSave, uu.mutation, uu.hooks) } // 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.NewUpdateSpec(user.Table, user.Columns, sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt)) 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.Email(); ok { _spec.SetField(user.FieldEmail, field.TypeString, value) } if value, ok := uu.mutation.Password(); ok { _spec.SetField(user.FieldPassword, field.TypeString, value) } if value, ok := uu.mutation.Salt(); ok { _spec.SetField(user.FieldSalt, field.TypeString, value) } if value, ok := uu.mutation.State(); ok { _spec.SetField(user.FieldState, field.TypeInt, value) } if value, ok := uu.mutation.AddedState(); ok { _spec.AddField(user.FieldState, field.TypeInt, value) } if value, ok := uu.mutation.RegTime(); ok { _spec.SetField(user.FieldRegTime, field.TypeInt64, value) } if value, ok := uu.mutation.AddedRegTime(); ok { _spec.AddField(user.FieldRegTime, field.TypeInt64, value) } if uu.mutation.SkinCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: true, Table: user.SkinTable, Columns: []string{user.SkinColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(skin.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := uu.mutation.RemovedSkinIDs(); len(nodes) > 0 && !uu.mutation.SkinCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: true, Table: user.SkinTable, Columns: []string{user.SkinColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(skin.FieldID, field.TypeInt), }, } for _, k := range nodes { edge.Target.Nodes = append(edge.Target.Nodes, k) } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := uu.mutation.SkinIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: true, Table: user.SkinTable, Columns: []string{user.SkinColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(skin.FieldID, field.TypeInt), }, } for _, k := range nodes { edge.Target.Nodes = append(edge.Target.Nodes, k) } _spec.Edges.Add = append(_spec.Edges.Add, edge) } if uu.mutation.ProfileCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2O, Inverse: false, Table: user.ProfileTable, Columns: []string{user.ProfileColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(userprofile.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := uu.mutation.ProfileIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2O, Inverse: false, Table: user.ProfileTable, Columns: []string{user.ProfileColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(userprofile.FieldID, field.TypeInt), }, } 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 sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } return 0, err } uu.mutation.done = true return n, nil } // UserUpdateOne is the builder for updating a single User entity. type UserUpdateOne struct { config fields []string hooks []Hook mutation *UserMutation } // SetEmail sets the "email" field. func (uuo *UserUpdateOne) SetEmail(s string) *UserUpdateOne { uuo.mutation.SetEmail(s) return uuo } // SetPassword sets the "password" field. func (uuo *UserUpdateOne) SetPassword(s string) *UserUpdateOne { uuo.mutation.SetPassword(s) return uuo } // SetSalt sets the "salt" field. func (uuo *UserUpdateOne) SetSalt(s string) *UserUpdateOne { uuo.mutation.SetSalt(s) return uuo } // SetState sets the "state" field. func (uuo *UserUpdateOne) SetState(i int) *UserUpdateOne { uuo.mutation.ResetState() uuo.mutation.SetState(i) return uuo } // AddState adds i to the "state" field. func (uuo *UserUpdateOne) AddState(i int) *UserUpdateOne { uuo.mutation.AddState(i) return uuo } // SetRegTime sets the "reg_time" field. func (uuo *UserUpdateOne) SetRegTime(i int64) *UserUpdateOne { uuo.mutation.ResetRegTime() uuo.mutation.SetRegTime(i) return uuo } // AddRegTime adds i to the "reg_time" field. func (uuo *UserUpdateOne) AddRegTime(i int64) *UserUpdateOne { uuo.mutation.AddRegTime(i) return uuo } // AddSkinIDs adds the "skin" edge to the Skin entity by IDs. func (uuo *UserUpdateOne) AddSkinIDs(ids ...int) *UserUpdateOne { uuo.mutation.AddSkinIDs(ids...) return uuo } // AddSkin adds the "skin" edges to the Skin entity. func (uuo *UserUpdateOne) AddSkin(s ...*Skin) *UserUpdateOne { ids := make([]int, len(s)) for i := range s { ids[i] = s[i].ID } return uuo.AddSkinIDs(ids...) } // SetProfileID sets the "profile" edge to the UserProfile entity by ID. func (uuo *UserUpdateOne) SetProfileID(id int) *UserUpdateOne { uuo.mutation.SetProfileID(id) return uuo } // SetNillableProfileID sets the "profile" edge to the UserProfile entity by ID if the given value is not nil. func (uuo *UserUpdateOne) SetNillableProfileID(id *int) *UserUpdateOne { if id != nil { uuo = uuo.SetProfileID(*id) } return uuo } // SetProfile sets the "profile" edge to the UserProfile entity. func (uuo *UserUpdateOne) SetProfile(u *UserProfile) *UserUpdateOne { return uuo.SetProfileID(u.ID) } // Mutation returns the UserMutation object of the builder. func (uuo *UserUpdateOne) Mutation() *UserMutation { return uuo.mutation } // ClearSkin clears all "skin" edges to the Skin entity. func (uuo *UserUpdateOne) ClearSkin() *UserUpdateOne { uuo.mutation.ClearSkin() return uuo } // RemoveSkinIDs removes the "skin" edge to Skin entities by IDs. func (uuo *UserUpdateOne) RemoveSkinIDs(ids ...int) *UserUpdateOne { uuo.mutation.RemoveSkinIDs(ids...) return uuo } // RemoveSkin removes "skin" edges to Skin entities. func (uuo *UserUpdateOne) RemoveSkin(s ...*Skin) *UserUpdateOne { ids := make([]int, len(s)) for i := range s { ids[i] = s[i].ID } return uuo.RemoveSkinIDs(ids...) } // ClearProfile clears the "profile" edge to the UserProfile entity. func (uuo *UserUpdateOne) ClearProfile() *UserUpdateOne { uuo.mutation.ClearProfile() return uuo } // Where appends a list predicates to the UserUpdate builder. func (uuo *UserUpdateOne) Where(ps ...predicate.User) *UserUpdateOne { uuo.mutation.Where(ps...) return uuo } // Select allows selecting one or more fields (columns) of the returned entity. // The default is selecting all fields defined in the entity schema. func (uuo *UserUpdateOne) Select(field string, fields ...string) *UserUpdateOne { uuo.fields = append([]string{field}, fields...) return uuo } // Save executes the query and returns the updated User entity. func (uuo *UserUpdateOne) Save(ctx context.Context) (*User, error) { return withHooks(ctx, uuo.sqlSave, uuo.mutation, uuo.hooks) } // 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.NewUpdateSpec(user.Table, user.Columns, sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt)) id, ok := uuo.mutation.ID() if !ok { return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "User.id" for update`)} } _spec.Node.ID.Value = id if fields := uuo.fields; len(fields) > 0 { _spec.Node.Columns = make([]string, 0, len(fields)) _spec.Node.Columns = append(_spec.Node.Columns, user.FieldID) for _, f := range fields { if !user.ValidColumn(f) { return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)} } if f != user.FieldID { _spec.Node.Columns = append(_spec.Node.Columns, f) } } } if ps := uuo.mutation.predicates; len(ps) > 0 { _spec.Predicate = func(selector *sql.Selector) { for i := range ps { ps[i](selector) } } } if value, ok := uuo.mutation.Email(); ok { _spec.SetField(user.FieldEmail, field.TypeString, value) } if value, ok := uuo.mutation.Password(); ok { _spec.SetField(user.FieldPassword, field.TypeString, value) } if value, ok := uuo.mutation.Salt(); ok { _spec.SetField(user.FieldSalt, field.TypeString, value) } if value, ok := uuo.mutation.State(); ok { _spec.SetField(user.FieldState, field.TypeInt, value) } if value, ok := uuo.mutation.AddedState(); ok { _spec.AddField(user.FieldState, field.TypeInt, value) } if value, ok := uuo.mutation.RegTime(); ok { _spec.SetField(user.FieldRegTime, field.TypeInt64, value) } if value, ok := uuo.mutation.AddedRegTime(); ok { _spec.AddField(user.FieldRegTime, field.TypeInt64, value) } if uuo.mutation.SkinCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: true, Table: user.SkinTable, Columns: []string{user.SkinColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(skin.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := uuo.mutation.RemovedSkinIDs(); len(nodes) > 0 && !uuo.mutation.SkinCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: true, Table: user.SkinTable, Columns: []string{user.SkinColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(skin.FieldID, field.TypeInt), }, } for _, k := range nodes { edge.Target.Nodes = append(edge.Target.Nodes, k) } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := uuo.mutation.SkinIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2M, Inverse: true, Table: user.SkinTable, Columns: []string{user.SkinColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(skin.FieldID, field.TypeInt), }, } for _, k := range nodes { edge.Target.Nodes = append(edge.Target.Nodes, k) } _spec.Edges.Add = append(_spec.Edges.Add, edge) } if uuo.mutation.ProfileCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2O, Inverse: false, Table: user.ProfileTable, Columns: []string{user.ProfileColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(userprofile.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := uuo.mutation.ProfileIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.O2O, Inverse: false, Table: user.ProfileTable, Columns: []string{user.ProfileColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(userprofile.FieldID, field.TypeInt), }, } 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 sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } return nil, err } uuo.mutation.done = true return _node, nil }