// Code generated by ent, DO NOT EDIT. package ent import ( "context" "errors" "fmt" "time" "entgo.io/ent/dialect/sql" "entgo.io/ent/dialect/sql/sqlgraph" "entgo.io/ent/schema/field" "github.com/mikestefanello/pagoda/ent/passwordtoken" "github.com/mikestefanello/pagoda/ent/predicate" "github.com/mikestefanello/pagoda/ent/user" ) // PasswordTokenUpdate is the builder for updating PasswordToken entities. type PasswordTokenUpdate struct { config hooks []Hook mutation *PasswordTokenMutation } // Where appends a list predicates to the PasswordTokenUpdate builder. func (_u *PasswordTokenUpdate) Where(ps ...predicate.PasswordToken) *PasswordTokenUpdate { _u.mutation.Where(ps...) return _u } // SetToken sets the "token" field. func (_u *PasswordTokenUpdate) SetToken(v string) *PasswordTokenUpdate { _u.mutation.SetToken(v) return _u } // SetNillableToken sets the "token" field if the given value is not nil. func (_u *PasswordTokenUpdate) SetNillableToken(v *string) *PasswordTokenUpdate { if v != nil { _u.SetToken(*v) } return _u } // SetUserID sets the "user_id" field. func (_u *PasswordTokenUpdate) SetUserID(v int) *PasswordTokenUpdate { _u.mutation.SetUserID(v) return _u } // SetNillableUserID sets the "user_id" field if the given value is not nil. func (_u *PasswordTokenUpdate) SetNillableUserID(v *int) *PasswordTokenUpdate { if v != nil { _u.SetUserID(*v) } return _u } // SetCreatedAt sets the "created_at" field. func (_u *PasswordTokenUpdate) SetCreatedAt(v time.Time) *PasswordTokenUpdate { _u.mutation.SetCreatedAt(v) return _u } // SetNillableCreatedAt sets the "created_at" field if the given value is not nil. func (_u *PasswordTokenUpdate) SetNillableCreatedAt(v *time.Time) *PasswordTokenUpdate { if v != nil { _u.SetCreatedAt(*v) } return _u } // SetUser sets the "user" edge to the User entity. func (_u *PasswordTokenUpdate) SetUser(v *User) *PasswordTokenUpdate { return _u.SetUserID(v.ID) } // Mutation returns the PasswordTokenMutation object of the builder. func (_u *PasswordTokenUpdate) Mutation() *PasswordTokenMutation { return _u.mutation } // ClearUser clears the "user" edge to the User entity. func (_u *PasswordTokenUpdate) ClearUser() *PasswordTokenUpdate { _u.mutation.ClearUser() return _u } // Save executes the query and returns the number of nodes affected by the update operation. func (_u *PasswordTokenUpdate) Save(ctx context.Context) (int, error) { return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks) } // SaveX is like Save, but panics if an error occurs. func (_u *PasswordTokenUpdate) SaveX(ctx context.Context) int { affected, err := _u.Save(ctx) if err != nil { panic(err) } return affected } // Exec executes the query. func (_u *PasswordTokenUpdate) Exec(ctx context.Context) error { _, err := _u.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (_u *PasswordTokenUpdate) ExecX(ctx context.Context) { if err := _u.Exec(ctx); err != nil { panic(err) } } // check runs all checks and user-defined validators on the builder. func (_u *PasswordTokenUpdate) check() error { if v, ok := _u.mutation.Token(); ok { if err := passwordtoken.TokenValidator(v); err != nil { return &ValidationError{Name: "token", err: fmt.Errorf(`ent: validator failed for field "PasswordToken.token": %w`, err)} } } if _u.mutation.UserCleared() && len(_u.mutation.UserIDs()) > 0 { return errors.New(`ent: clearing a required unique edge "PasswordToken.user"`) } return nil } func (_u *PasswordTokenUpdate) sqlSave(ctx context.Context) (_node int, err error) { if err := _u.check(); err != nil { return _node, err } _spec := sqlgraph.NewUpdateSpec(passwordtoken.Table, passwordtoken.Columns, sqlgraph.NewFieldSpec(passwordtoken.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.Token(); ok { _spec.SetField(passwordtoken.FieldToken, field.TypeString, value) } if value, ok := _u.mutation.CreatedAt(); ok { _spec.SetField(passwordtoken.FieldCreatedAt, field.TypeTime, value) } if _u.mutation.UserCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2O, Inverse: false, Table: passwordtoken.UserTable, Columns: []string{passwordtoken.UserColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := _u.mutation.UserIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2O, Inverse: false, Table: passwordtoken.UserTable, Columns: []string{passwordtoken.UserColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.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{passwordtoken.Label} } else if sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } return 0, err } _u.mutation.done = true return _node, nil } // PasswordTokenUpdateOne is the builder for updating a single PasswordToken entity. type PasswordTokenUpdateOne struct { config fields []string hooks []Hook mutation *PasswordTokenMutation } // SetToken sets the "token" field. func (_u *PasswordTokenUpdateOne) SetToken(v string) *PasswordTokenUpdateOne { _u.mutation.SetToken(v) return _u } // SetNillableToken sets the "token" field if the given value is not nil. func (_u *PasswordTokenUpdateOne) SetNillableToken(v *string) *PasswordTokenUpdateOne { if v != nil { _u.SetToken(*v) } return _u } // SetUserID sets the "user_id" field. func (_u *PasswordTokenUpdateOne) SetUserID(v int) *PasswordTokenUpdateOne { _u.mutation.SetUserID(v) return _u } // SetNillableUserID sets the "user_id" field if the given value is not nil. func (_u *PasswordTokenUpdateOne) SetNillableUserID(v *int) *PasswordTokenUpdateOne { if v != nil { _u.SetUserID(*v) } return _u } // SetCreatedAt sets the "created_at" field. func (_u *PasswordTokenUpdateOne) SetCreatedAt(v time.Time) *PasswordTokenUpdateOne { _u.mutation.SetCreatedAt(v) return _u } // SetNillableCreatedAt sets the "created_at" field if the given value is not nil. func (_u *PasswordTokenUpdateOne) SetNillableCreatedAt(v *time.Time) *PasswordTokenUpdateOne { if v != nil { _u.SetCreatedAt(*v) } return _u } // SetUser sets the "user" edge to the User entity. func (_u *PasswordTokenUpdateOne) SetUser(v *User) *PasswordTokenUpdateOne { return _u.SetUserID(v.ID) } // Mutation returns the PasswordTokenMutation object of the builder. func (_u *PasswordTokenUpdateOne) Mutation() *PasswordTokenMutation { return _u.mutation } // ClearUser clears the "user" edge to the User entity. func (_u *PasswordTokenUpdateOne) ClearUser() *PasswordTokenUpdateOne { _u.mutation.ClearUser() return _u } // Where appends a list predicates to the PasswordTokenUpdate builder. func (_u *PasswordTokenUpdateOne) Where(ps ...predicate.PasswordToken) *PasswordTokenUpdateOne { _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 *PasswordTokenUpdateOne) Select(field string, fields ...string) *PasswordTokenUpdateOne { _u.fields = append([]string{field}, fields...) return _u } // Save executes the query and returns the updated PasswordToken entity. func (_u *PasswordTokenUpdateOne) Save(ctx context.Context) (*PasswordToken, error) { return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks) } // SaveX is like Save, but panics if an error occurs. func (_u *PasswordTokenUpdateOne) SaveX(ctx context.Context) *PasswordToken { node, err := _u.Save(ctx) if err != nil { panic(err) } return node } // Exec executes the query on the entity. func (_u *PasswordTokenUpdateOne) Exec(ctx context.Context) error { _, err := _u.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (_u *PasswordTokenUpdateOne) ExecX(ctx context.Context) { if err := _u.Exec(ctx); err != nil { panic(err) } } // check runs all checks and user-defined validators on the builder. func (_u *PasswordTokenUpdateOne) check() error { if v, ok := _u.mutation.Token(); ok { if err := passwordtoken.TokenValidator(v); err != nil { return &ValidationError{Name: "token", err: fmt.Errorf(`ent: validator failed for field "PasswordToken.token": %w`, err)} } } if _u.mutation.UserCleared() && len(_u.mutation.UserIDs()) > 0 { return errors.New(`ent: clearing a required unique edge "PasswordToken.user"`) } return nil } func (_u *PasswordTokenUpdateOne) sqlSave(ctx context.Context) (_node *PasswordToken, err error) { if err := _u.check(); err != nil { return _node, err } _spec := sqlgraph.NewUpdateSpec(passwordtoken.Table, passwordtoken.Columns, sqlgraph.NewFieldSpec(passwordtoken.FieldID, field.TypeInt)) id, ok := _u.mutation.ID() if !ok { return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "PasswordToken.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, passwordtoken.FieldID) for _, f := range fields { if !passwordtoken.ValidColumn(f) { return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)} } if f != passwordtoken.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.Token(); ok { _spec.SetField(passwordtoken.FieldToken, field.TypeString, value) } if value, ok := _u.mutation.CreatedAt(); ok { _spec.SetField(passwordtoken.FieldCreatedAt, field.TypeTime, value) } if _u.mutation.UserCleared() { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2O, Inverse: false, Table: passwordtoken.UserTable, Columns: []string{passwordtoken.UserColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } _spec.Edges.Clear = append(_spec.Edges.Clear, edge) } if nodes := _u.mutation.UserIDs(); len(nodes) > 0 { edge := &sqlgraph.EdgeSpec{ Rel: sqlgraph.M2O, Inverse: false, Table: passwordtoken.UserTable, Columns: []string{passwordtoken.UserColumn}, Bidi: false, Target: &sqlgraph.EdgeTarget{ IDSpec: sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt), }, } for _, k := range nodes { edge.Target.Nodes = append(edge.Target.Nodes, k) } _spec.Edges.Add = append(_spec.Edges.Add, edge) } _node = &PasswordToken{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{passwordtoken.Label} } else if sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } return nil, err } _u.mutation.done = true return _node, nil }