// Code generated by ent, DO NOT EDIT. package ent import ( "EnGin/internal/ent/predicate" "EnGin/internal/ent/role" "EnGin/internal/ent/user" "context" "errors" "fmt" "time" "entgo.io/ent/dialect/sql" "entgo.io/ent/dialect/sql/sqlgraph" "entgo.io/ent/schema/field" ) // 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 (_u *UserUpdate) Where(ps ...predicate.User) *UserUpdate { _u.mutation.Where(ps...) return _u } // SetUpdatedAt sets the "updated_at" field. func (_u *UserUpdate) SetUpdatedAt(v time.Time) *UserUpdate { _u.mutation.SetUpdatedAt(v) return _u } // SetUsername sets the "username" field. func (_u *UserUpdate) SetUsername(v string) *UserUpdate { _u.mutation.SetUsername(v) return _u } // SetNillableUsername sets the "username" field if the given value is not nil. func (_u *UserUpdate) SetNillableUsername(v *string) *UserUpdate { if v != nil { _u.SetUsername(*v) } return _u } // SetPassword sets the "password" field. func (_u *UserUpdate) SetPassword(v string) *UserUpdate { _u.mutation.SetPassword(v) return _u } // SetNillablePassword sets the "password" field if the given value is not nil. func (_u *UserUpdate) SetNillablePassword(v *string) *UserUpdate { if v != nil { _u.SetPassword(*v) } return _u } // SetNickname sets the "nickname" field. func (_u *UserUpdate) SetNickname(v string) *UserUpdate { _u.mutation.SetNickname(v) return _u } // SetNillableNickname sets the "nickname" field if the given value is not nil. func (_u *UserUpdate) SetNillableNickname(v *string) *UserUpdate { if v != nil { _u.SetNickname(*v) } return _u } // ClearNickname clears the value of the "nickname" field. func (_u *UserUpdate) ClearNickname() *UserUpdate { _u.mutation.ClearNickname() return _u } // SetEmail sets the "email" field. func (_u *UserUpdate) SetEmail(v string) *UserUpdate { _u.mutation.SetEmail(v) return _u } // SetNillableEmail sets the "email" field if the given value is not nil. func (_u *UserUpdate) SetNillableEmail(v *string) *UserUpdate { if v != nil { _u.SetEmail(*v) } return _u } // ClearEmail clears the value of the "email" field. func (_u *UserUpdate) ClearEmail() *UserUpdate { _u.mutation.ClearEmail() return _u } // SetIP sets the "ip" field. func (_u *UserUpdate) SetIP(v string) *UserUpdate { _u.mutation.SetIP(v) return _u } // SetNillableIP sets the "ip" field if the given value is not nil. func (_u *UserUpdate) SetNillableIP(v *string) *UserUpdate { if v != nil { _u.SetIP(*v) } return _u } // ClearIP clears the value of the "ip" field. func (_u *UserUpdate) ClearIP() *UserUpdate { _u.mutation.ClearIP() return _u } // SetAddr sets the "addr" field. func (_u *UserUpdate) SetAddr(v string) *UserUpdate { _u.mutation.SetAddr(v) return _u } // SetNillableAddr sets the "addr" field if the given value is not nil. func (_u *UserUpdate) SetNillableAddr(v *string) *UserUpdate { if v != nil { _u.SetAddr(*v) } return _u } // ClearAddr clears the value of the "addr" field. func (_u *UserUpdate) ClearAddr() *UserUpdate { _u.mutation.ClearAddr() return _u } // AddRoleIDs adds the "roles" edge to the Role entity by IDs. func (_u *UserUpdate) AddRoleIDs(ids ...int) *UserUpdate { _u.mutation.AddRoleIDs(ids...) return _u } // AddRoles adds the "roles" edges to the Role entity. func (_u *UserUpdate) AddRoles(v ...*Role) *UserUpdate { ids := make([]int, len(v)) for i := range v { ids[i] = v[i].ID } return _u.AddRoleIDs(ids...) } // Mutation returns the UserMutation object of the builder. func (_u *UserUpdate) Mutation() *UserMutation { return _u.mutation } // ClearRoles clears all "roles" edges to the Role entity. func (_u *UserUpdate) ClearRoles() *UserUpdate { _u.mutation.ClearRoles() return _u } // RemoveRoleIDs removes the "roles" edge to Role entities by IDs. func (_u *UserUpdate) RemoveRoleIDs(ids ...int) *UserUpdate { _u.mutation.RemoveRoleIDs(ids...) return _u } // RemoveRoles removes "roles" edges to Role entities. func (_u *UserUpdate) RemoveRoles(v ...*Role) *UserUpdate { ids := make([]int, len(v)) for i := range v { ids[i] = v[i].ID } return _u.RemoveRoleIDs(ids...) } // Save executes the query and returns the number of nodes affected by the update operation. func (_u *UserUpdate) Save(ctx context.Context) (int, error) { _u.defaults() return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks) } // SaveX is like Save, but panics if an error occurs. func (_u *UserUpdate) SaveX(ctx context.Context) int { affected, err := _u.Save(ctx) if err != nil { panic(err) } return affected } // Exec executes the query. func (_u *UserUpdate) Exec(ctx context.Context) error { _, err := _u.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (_u *UserUpdate) ExecX(ctx context.Context) { if err := _u.Exec(ctx); err != nil { panic(err) } } // defaults sets the default values of the builder before save. func (_u *UserUpdate) defaults() { if _, ok := _u.mutation.UpdatedAt(); !ok { v := user.UpdateDefaultUpdatedAt() _u.mutation.SetUpdatedAt(v) } } // check runs all checks and user-defined validators on the builder. func (_u *UserUpdate) check() error { if v, ok := _u.mutation.Username(); ok { if err := user.UsernameValidator(v); err != nil { return &ValidationError{Name: "username", err: fmt.Errorf(`ent: validator failed for field "User.username": %w`, err)} } } if v, ok := _u.mutation.Nickname(); ok { if err := user.NicknameValidator(v); err != nil { return &ValidationError{Name: "nickname", err: fmt.Errorf(`ent: validator failed for field "User.nickname": %w`, err)} } } if v, ok := _u.mutation.Email(); ok { if err := user.EmailValidator(v); err != nil { return &ValidationError{Name: "email", err: fmt.Errorf(`ent: validator failed for field "User.email": %w`, err)} } } if v, ok := _u.mutation.IP(); ok { if err := user.IPValidator(v); err != nil { return &ValidationError{Name: "ip", err: fmt.Errorf(`ent: validator failed for field "User.ip": %w`, err)} } } if v, ok := _u.mutation.Addr(); ok { if err := user.AddrValidator(v); err != nil { return &ValidationError{Name: "addr", err: fmt.Errorf(`ent: validator failed for field "User.addr": %w`, err)} } } return nil } func (_u *UserUpdate) sqlSave(ctx context.Context) (_node int, err error) { if err := _u.check(); err != nil { return _node, err } _spec := sqlgraph.NewUpdateSpec(user.Table, user.Columns, sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt)) if ps := _u.mutation.predicates; len(ps) > 0 { _spec.Predicate = func(selector *sql.Selector) { for i := range ps { ps[i](selector) } } } if value, ok := _u.mutation.UpdatedAt(); ok { _spec.SetField(user.FieldUpdatedAt, field.TypeTime, value) } if value, ok := _u.mutation.Username(); ok { _spec.SetField(user.FieldUsername, field.TypeString, value) } if value, ok := _u.mutation.Password(); ok { _spec.SetField(user.FieldPassword, field.TypeString, value) } if value, ok := _u.mutation.Nickname(); ok { _spec.SetField(user.FieldNickname, field.TypeString, value) } if _u.mutation.NicknameCleared() { _spec.ClearField(user.FieldNickname, field.TypeString) } if value, ok := _u.mutation.Email(); ok { _spec.SetField(user.FieldEmail, field.TypeString, value) } if _u.mutation.EmailCleared() { _spec.ClearField(user.FieldEmail, field.TypeString) } if value, ok := _u.mutation.IP(); ok { _spec.SetField(user.FieldIP, field.TypeString, value) } if _u.mutation.IPCleared() { _spec.ClearField(user.FieldIP, field.TypeString) } if value, ok := _u.mutation.Addr(); ok { _spec.SetField(user.FieldAddr, field.TypeString, value) } if _u.mutation.AddrCleared() { _spec.ClearField(user.FieldAddr, field.TypeString) } if _u.mutation.RolesCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: user.RolesTable, Columns: user.RolesPrimaryKey, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(role.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := _u.mutation.RemovedRolesIDs(); len(nodes) > 0 && !_u.mutation.RolesCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: user.RolesTable, Columns: user.RolesPrimaryKey, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(role.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 := _u.mutation.RolesIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: user.RolesTable, Columns: user.RolesPrimaryKey, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(role.FieldID, field.TypeInt), }, } for _, k := range nodes { edge.Target.Nodes = append(edge.Target.Nodes, k) } _spec.Edges.Add = append(_spec.Edges.Add, edge) } if _node, err = sqlgraph.UpdateNodes(ctx, _u.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 } _u.mutation.done = true return _node, nil } // UserUpdateOne is the builder for updating a single User entity. type UserUpdateOne struct { config fields []string hooks []Hook mutation *UserMutation } // SetUpdatedAt sets the "updated_at" field. func (_u *UserUpdateOne) SetUpdatedAt(v time.Time) *UserUpdateOne { _u.mutation.SetUpdatedAt(v) return _u } // SetUsername sets the "username" field. func (_u *UserUpdateOne) SetUsername(v string) *UserUpdateOne { _u.mutation.SetUsername(v) return _u } // SetNillableUsername sets the "username" field if the given value is not nil. func (_u *UserUpdateOne) SetNillableUsername(v *string) *UserUpdateOne { if v != nil { _u.SetUsername(*v) } return _u } // SetPassword sets the "password" field. func (_u *UserUpdateOne) SetPassword(v string) *UserUpdateOne { _u.mutation.SetPassword(v) return _u } // SetNillablePassword sets the "password" field if the given value is not nil. func (_u *UserUpdateOne) SetNillablePassword(v *string) *UserUpdateOne { if v != nil { _u.SetPassword(*v) } return _u } // SetNickname sets the "nickname" field. func (_u *UserUpdateOne) SetNickname(v string) *UserUpdateOne { _u.mutation.SetNickname(v) return _u } // SetNillableNickname sets the "nickname" field if the given value is not nil. func (_u *UserUpdateOne) SetNillableNickname(v *string) *UserUpdateOne { if v != nil { _u.SetNickname(*v) } return _u } // ClearNickname clears the value of the "nickname" field. func (_u *UserUpdateOne) ClearNickname() *UserUpdateOne { _u.mutation.ClearNickname() return _u } // SetEmail sets the "email" field. func (_u *UserUpdateOne) SetEmail(v string) *UserUpdateOne { _u.mutation.SetEmail(v) return _u } // SetNillableEmail sets the "email" field if the given value is not nil. func (_u *UserUpdateOne) SetNillableEmail(v *string) *UserUpdateOne { if v != nil { _u.SetEmail(*v) } return _u } // ClearEmail clears the value of the "email" field. func (_u *UserUpdateOne) ClearEmail() *UserUpdateOne { _u.mutation.ClearEmail() return _u } // SetIP sets the "ip" field. func (_u *UserUpdateOne) SetIP(v string) *UserUpdateOne { _u.mutation.SetIP(v) return _u } // SetNillableIP sets the "ip" field if the given value is not nil. func (_u *UserUpdateOne) SetNillableIP(v *string) *UserUpdateOne { if v != nil { _u.SetIP(*v) } return _u } // ClearIP clears the value of the "ip" field. func (_u *UserUpdateOne) ClearIP() *UserUpdateOne { _u.mutation.ClearIP() return _u } // SetAddr sets the "addr" field. func (_u *UserUpdateOne) SetAddr(v string) *UserUpdateOne { _u.mutation.SetAddr(v) return _u } // SetNillableAddr sets the "addr" field if the given value is not nil. func (_u *UserUpdateOne) SetNillableAddr(v *string) *UserUpdateOne { if v != nil { _u.SetAddr(*v) } return _u } // ClearAddr clears the value of the "addr" field. func (_u *UserUpdateOne) ClearAddr() *UserUpdateOne { _u.mutation.ClearAddr() return _u } // AddRoleIDs adds the "roles" edge to the Role entity by IDs. func (_u *UserUpdateOne) AddRoleIDs(ids ...int) *UserUpdateOne { _u.mutation.AddRoleIDs(ids...) return _u } // AddRoles adds the "roles" edges to the Role entity. func (_u *UserUpdateOne) AddRoles(v ...*Role) *UserUpdateOne { ids := make([]int, len(v)) for i := range v { ids[i] = v[i].ID } return _u.AddRoleIDs(ids...) } // Mutation returns the UserMutation object of the builder. func (_u *UserUpdateOne) Mutation() *UserMutation { return _u.mutation } // ClearRoles clears all "roles" edges to the Role entity. func (_u *UserUpdateOne) ClearRoles() *UserUpdateOne { _u.mutation.ClearRoles() return _u } // RemoveRoleIDs removes the "roles" edge to Role entities by IDs. func (_u *UserUpdateOne) RemoveRoleIDs(ids ...int) *UserUpdateOne { _u.mutation.RemoveRoleIDs(ids...) return _u } // RemoveRoles removes "roles" edges to Role entities. func (_u *UserUpdateOne) RemoveRoles(v ...*Role) *UserUpdateOne { ids := make([]int, len(v)) for i := range v { ids[i] = v[i].ID } return _u.RemoveRoleIDs(ids...) } // Where appends a list predicates to the UserUpdate builder. func (_u *UserUpdateOne) Where(ps ...predicate.User) *UserUpdateOne { _u.mutation.Where(ps...) return _u } // Select allows selecting one or more fields (columns) of the returned entity. // The default is selecting all fields defined in the entity schema. func (_u *UserUpdateOne) Select(field string, fields ...string) *UserUpdateOne { _u.fields = append([]string{field}, fields...) return _u } // Save executes the query and returns the updated User entity. func (_u *UserUpdateOne) Save(ctx context.Context) (*User, error) { _u.defaults() return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks) } // SaveX is like Save, but panics if an error occurs. func (_u *UserUpdateOne) SaveX(ctx context.Context) *User { node, err := _u.Save(ctx) if err != nil { panic(err) } return node } // Exec executes the query on the entity. func (_u *UserUpdateOne) Exec(ctx context.Context) error { _, err := _u.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (_u *UserUpdateOne) ExecX(ctx context.Context) { if err := _u.Exec(ctx); err != nil { panic(err) } } // defaults sets the default values of the builder before save. func (_u *UserUpdateOne) defaults() { if _, ok := _u.mutation.UpdatedAt(); !ok { v := user.UpdateDefaultUpdatedAt() _u.mutation.SetUpdatedAt(v) } } // check runs all checks and user-defined validators on the builder. func (_u *UserUpdateOne) check() error { if v, ok := _u.mutation.Username(); ok { if err := user.UsernameValidator(v); err != nil { return &ValidationError{Name: "username", err: fmt.Errorf(`ent: validator failed for field "User.username": %w`, err)} } } if v, ok := _u.mutation.Nickname(); ok { if err := user.NicknameValidator(v); err != nil { return &ValidationError{Name: "nickname", err: fmt.Errorf(`ent: validator failed for field "User.nickname": %w`, err)} } } if v, ok := _u.mutation.Email(); ok { if err := user.EmailValidator(v); err != nil { return &ValidationError{Name: "email", err: fmt.Errorf(`ent: validator failed for field "User.email": %w`, err)} } } if v, ok := _u.mutation.IP(); ok { if err := user.IPValidator(v); err != nil { return &ValidationError{Name: "ip", err: fmt.Errorf(`ent: validator failed for field "User.ip": %w`, err)} } } if v, ok := _u.mutation.Addr(); ok { if err := user.AddrValidator(v); err != nil { return &ValidationError{Name: "addr", err: fmt.Errorf(`ent: validator failed for field "User.addr": %w`, err)} } } return nil } func (_u *UserUpdateOne) sqlSave(ctx context.Context) (_node *User, err error) { if err := _u.check(); err != nil { return _node, err } _spec := sqlgraph.NewUpdateSpec(user.Table, user.Columns, sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt)) id, ok := _u.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 := _u.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 := _u.mutation.predicates; len(ps) > 0 { _spec.Predicate = func(selector *sql.Selector) { for i := range ps { ps[i](selector) } } } if value, ok := _u.mutation.UpdatedAt(); ok { _spec.SetField(user.FieldUpdatedAt, field.TypeTime, value) } if value, ok := _u.mutation.Username(); ok { _spec.SetField(user.FieldUsername, field.TypeString, value) } if value, ok := _u.mutation.Password(); ok { _spec.SetField(user.FieldPassword, field.TypeString, value) } if value, ok := _u.mutation.Nickname(); ok { _spec.SetField(user.FieldNickname, field.TypeString, value) } if _u.mutation.NicknameCleared() { _spec.ClearField(user.FieldNickname, field.TypeString) } if value, ok := _u.mutation.Email(); ok { _spec.SetField(user.FieldEmail, field.TypeString, value) } if _u.mutation.EmailCleared() { _spec.ClearField(user.FieldEmail, field.TypeString) } if value, ok := _u.mutation.IP(); ok { _spec.SetField(user.FieldIP, field.TypeString, value) } if _u.mutation.IPCleared() { _spec.ClearField(user.FieldIP, field.TypeString) } if value, ok := _u.mutation.Addr(); ok { _spec.SetField(user.FieldAddr, field.TypeString, value) } if _u.mutation.AddrCleared() { _spec.ClearField(user.FieldAddr, field.TypeString) } if _u.mutation.RolesCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: user.RolesTable, Columns: user.RolesPrimaryKey, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(role.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := _u.mutation.RemovedRolesIDs(); len(nodes) > 0 && !_u.mutation.RolesCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: user.RolesTable, Columns: user.RolesPrimaryKey, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(role.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 := _u.mutation.RolesIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2M, Inverse: false, Table: user.RolesTable, Columns: user.RolesPrimaryKey, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(role.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: _u.config} _spec.Assign = _node.assignValues _spec.ScanValues = _node.scanValues if err = sqlgraph.UpdateNode(ctx, _u.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 } _u.mutation.done = true return _node, nil }