// Code generated by ent, DO NOT EDIT. package ent import ( "bj_power_wms/ent/predicate" "bj_power_wms/ent/stocktakeorder" "context" "errors" "fmt" "entgo.io/ent/dialect/sql" "entgo.io/ent/dialect/sql/sqlgraph" "entgo.io/ent/schema/field" ) // StocktakeOrderUpdate is the builder for updating StocktakeOrder entities. type StocktakeOrderUpdate struct { config hooks []Hook mutation *StocktakeOrderMutation } // Where appends a list predicates to the StocktakeOrderUpdate builder. func (_u *StocktakeOrderUpdate) Where(ps ...predicate.StocktakeOrder) *StocktakeOrderUpdate { _u.mutation.Where(ps...) return _u } // SetStocktakeNo sets the "stocktake_no" field. func (_u *StocktakeOrderUpdate) SetStocktakeNo(v string) *StocktakeOrderUpdate { _u.mutation.SetStocktakeNo(v) return _u } // SetNillableStocktakeNo sets the "stocktake_no" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableStocktakeNo(v *string) *StocktakeOrderUpdate { if v != nil { _u.SetStocktakeNo(*v) } return _u } // SetStatus sets the "status" field. func (_u *StocktakeOrderUpdate) SetStatus(v string) *StocktakeOrderUpdate { _u.mutation.SetStatus(v) return _u } // SetNillableStatus sets the "status" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableStatus(v *string) *StocktakeOrderUpdate { if v != nil { _u.SetStatus(*v) } return _u } // SetOperator sets the "operator" field. func (_u *StocktakeOrderUpdate) SetOperator(v string) *StocktakeOrderUpdate { _u.mutation.SetOperator(v) return _u } // SetNillableOperator sets the "operator" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableOperator(v *string) *StocktakeOrderUpdate { if v != nil { _u.SetOperator(*v) } return _u } // ClearOperator clears the value of the "operator" field. func (_u *StocktakeOrderUpdate) ClearOperator() *StocktakeOrderUpdate { _u.mutation.ClearOperator() return _u } // SetTotalTargets sets the "total_targets" field. func (_u *StocktakeOrderUpdate) SetTotalTargets(v int) *StocktakeOrderUpdate { _u.mutation.ResetTotalTargets() _u.mutation.SetTotalTargets(v) return _u } // SetNillableTotalTargets sets the "total_targets" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableTotalTargets(v *int) *StocktakeOrderUpdate { if v != nil { _u.SetTotalTargets(*v) } return _u } // AddTotalTargets adds value to the "total_targets" field. func (_u *StocktakeOrderUpdate) AddTotalTargets(v int) *StocktakeOrderUpdate { _u.mutation.AddTotalTargets(v) return _u } // SetDiffCount sets the "diff_count" field. func (_u *StocktakeOrderUpdate) SetDiffCount(v int) *StocktakeOrderUpdate { _u.mutation.ResetDiffCount() _u.mutation.SetDiffCount(v) return _u } // SetNillableDiffCount sets the "diff_count" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableDiffCount(v *int) *StocktakeOrderUpdate { if v != nil { _u.SetDiffCount(*v) } return _u } // AddDiffCount adds value to the "diff_count" field. func (_u *StocktakeOrderUpdate) AddDiffCount(v int) *StocktakeOrderUpdate { _u.mutation.AddDiffCount(v) return _u } // SetAdjusted sets the "adjusted" field. func (_u *StocktakeOrderUpdate) SetAdjusted(v bool) *StocktakeOrderUpdate { _u.mutation.SetAdjusted(v) return _u } // SetNillableAdjusted sets the "adjusted" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableAdjusted(v *bool) *StocktakeOrderUpdate { if v != nil { _u.SetAdjusted(*v) } return _u } // SetAdjustNo sets the "adjust_no" field. func (_u *StocktakeOrderUpdate) SetAdjustNo(v string) *StocktakeOrderUpdate { _u.mutation.SetAdjustNo(v) return _u } // SetNillableAdjustNo sets the "adjust_no" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableAdjustNo(v *string) *StocktakeOrderUpdate { if v != nil { _u.SetAdjustNo(*v) } return _u } // ClearAdjustNo clears the value of the "adjust_no" field. func (_u *StocktakeOrderUpdate) ClearAdjustNo() *StocktakeOrderUpdate { _u.mutation.ClearAdjustNo() return _u } // SetWrittenAt sets the "written_at" field. func (_u *StocktakeOrderUpdate) SetWrittenAt(v int64) *StocktakeOrderUpdate { _u.mutation.ResetWrittenAt() _u.mutation.SetWrittenAt(v) return _u } // SetNillableWrittenAt sets the "written_at" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableWrittenAt(v *int64) *StocktakeOrderUpdate { if v != nil { _u.SetWrittenAt(*v) } return _u } // AddWrittenAt adds value to the "written_at" field. func (_u *StocktakeOrderUpdate) AddWrittenAt(v int64) *StocktakeOrderUpdate { _u.mutation.AddWrittenAt(v) return _u } // ClearWrittenAt clears the value of the "written_at" field. func (_u *StocktakeOrderUpdate) ClearWrittenAt() *StocktakeOrderUpdate { _u.mutation.ClearWrittenAt() return _u } // SetWrittenBy sets the "written_by" field. func (_u *StocktakeOrderUpdate) SetWrittenBy(v string) *StocktakeOrderUpdate { _u.mutation.SetWrittenBy(v) return _u } // SetNillableWrittenBy sets the "written_by" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableWrittenBy(v *string) *StocktakeOrderUpdate { if v != nil { _u.SetWrittenBy(*v) } return _u } // ClearWrittenBy clears the value of the "written_by" field. func (_u *StocktakeOrderUpdate) ClearWrittenBy() *StocktakeOrderUpdate { _u.mutation.ClearWrittenBy() return _u } // SetUncountedMode sets the "uncounted_mode" field. func (_u *StocktakeOrderUpdate) SetUncountedMode(v string) *StocktakeOrderUpdate { _u.mutation.SetUncountedMode(v) return _u } // SetNillableUncountedMode sets the "uncounted_mode" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableUncountedMode(v *string) *StocktakeOrderUpdate { if v != nil { _u.SetUncountedMode(*v) } return _u } // ClearUncountedMode clears the value of the "uncounted_mode" field. func (_u *StocktakeOrderUpdate) ClearUncountedMode() *StocktakeOrderUpdate { _u.mutation.ClearUncountedMode() return _u } // SetCancelReason sets the "cancel_reason" field. func (_u *StocktakeOrderUpdate) SetCancelReason(v string) *StocktakeOrderUpdate { _u.mutation.SetCancelReason(v) return _u } // SetNillableCancelReason sets the "cancel_reason" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableCancelReason(v *string) *StocktakeOrderUpdate { if v != nil { _u.SetCancelReason(*v) } return _u } // ClearCancelReason clears the value of the "cancel_reason" field. func (_u *StocktakeOrderUpdate) ClearCancelReason() *StocktakeOrderUpdate { _u.mutation.ClearCancelReason() return _u } // SetCreatedAt sets the "created_at" field. func (_u *StocktakeOrderUpdate) SetCreatedAt(v int64) *StocktakeOrderUpdate { _u.mutation.ResetCreatedAt() _u.mutation.SetCreatedAt(v) return _u } // SetNillableCreatedAt sets the "created_at" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableCreatedAt(v *int64) *StocktakeOrderUpdate { if v != nil { _u.SetCreatedAt(*v) } return _u } // AddCreatedAt adds value to the "created_at" field. func (_u *StocktakeOrderUpdate) AddCreatedAt(v int64) *StocktakeOrderUpdate { _u.mutation.AddCreatedAt(v) return _u } // SetFinishedAt sets the "finished_at" field. func (_u *StocktakeOrderUpdate) SetFinishedAt(v int64) *StocktakeOrderUpdate { _u.mutation.ResetFinishedAt() _u.mutation.SetFinishedAt(v) return _u } // SetNillableFinishedAt sets the "finished_at" field if the given value is not nil. func (_u *StocktakeOrderUpdate) SetNillableFinishedAt(v *int64) *StocktakeOrderUpdate { if v != nil { _u.SetFinishedAt(*v) } return _u } // AddFinishedAt adds value to the "finished_at" field. func (_u *StocktakeOrderUpdate) AddFinishedAt(v int64) *StocktakeOrderUpdate { _u.mutation.AddFinishedAt(v) return _u } // ClearFinishedAt clears the value of the "finished_at" field. func (_u *StocktakeOrderUpdate) ClearFinishedAt() *StocktakeOrderUpdate { _u.mutation.ClearFinishedAt() return _u } // Mutation returns the StocktakeOrderMutation object of the builder. func (_u *StocktakeOrderUpdate) Mutation() *StocktakeOrderMutation { return _u.mutation } // Save executes the query and returns the number of nodes affected by the update operation. func (_u *StocktakeOrderUpdate) 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 *StocktakeOrderUpdate) SaveX(ctx context.Context) int { affected, err := _u.Save(ctx) if err != nil { panic(err) } return affected } // Exec executes the query. func (_u *StocktakeOrderUpdate) Exec(ctx context.Context) error { _, err := _u.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (_u *StocktakeOrderUpdate) ExecX(ctx context.Context) { if err := _u.Exec(ctx); err != nil { panic(err) } } func (_u *StocktakeOrderUpdate) sqlSave(ctx context.Context) (_node int, err error) { _spec := sqlgraph.NewUpdateSpec(stocktakeorder.Table, stocktakeorder.Columns, sqlgraph.NewFieldSpec(stocktakeorder.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.StocktakeNo(); ok { _spec.SetField(stocktakeorder.FieldStocktakeNo, field.TypeString, value) } if value, ok := _u.mutation.Status(); ok { _spec.SetField(stocktakeorder.FieldStatus, field.TypeString, value) } if value, ok := _u.mutation.Operator(); ok { _spec.SetField(stocktakeorder.FieldOperator, field.TypeString, value) } if _u.mutation.OperatorCleared() { _spec.ClearField(stocktakeorder.FieldOperator, field.TypeString) } if value, ok := _u.mutation.TotalTargets(); ok { _spec.SetField(stocktakeorder.FieldTotalTargets, field.TypeInt, value) } if value, ok := _u.mutation.AddedTotalTargets(); ok { _spec.AddField(stocktakeorder.FieldTotalTargets, field.TypeInt, value) } if value, ok := _u.mutation.DiffCount(); ok { _spec.SetField(stocktakeorder.FieldDiffCount, field.TypeInt, value) } if value, ok := _u.mutation.AddedDiffCount(); ok { _spec.AddField(stocktakeorder.FieldDiffCount, field.TypeInt, value) } if value, ok := _u.mutation.Adjusted(); ok { _spec.SetField(stocktakeorder.FieldAdjusted, field.TypeBool, value) } if value, ok := _u.mutation.AdjustNo(); ok { _spec.SetField(stocktakeorder.FieldAdjustNo, field.TypeString, value) } if _u.mutation.AdjustNoCleared() { _spec.ClearField(stocktakeorder.FieldAdjustNo, field.TypeString) } if value, ok := _u.mutation.WrittenAt(); ok { _spec.SetField(stocktakeorder.FieldWrittenAt, field.TypeInt64, value) } if value, ok := _u.mutation.AddedWrittenAt(); ok { _spec.AddField(stocktakeorder.FieldWrittenAt, field.TypeInt64, value) } if _u.mutation.WrittenAtCleared() { _spec.ClearField(stocktakeorder.FieldWrittenAt, field.TypeInt64) } if value, ok := _u.mutation.WrittenBy(); ok { _spec.SetField(stocktakeorder.FieldWrittenBy, field.TypeString, value) } if _u.mutation.WrittenByCleared() { _spec.ClearField(stocktakeorder.FieldWrittenBy, field.TypeString) } if value, ok := _u.mutation.UncountedMode(); ok { _spec.SetField(stocktakeorder.FieldUncountedMode, field.TypeString, value) } if _u.mutation.UncountedModeCleared() { _spec.ClearField(stocktakeorder.FieldUncountedMode, field.TypeString) } if value, ok := _u.mutation.CancelReason(); ok { _spec.SetField(stocktakeorder.FieldCancelReason, field.TypeString, value) } if _u.mutation.CancelReasonCleared() { _spec.ClearField(stocktakeorder.FieldCancelReason, field.TypeString) } if value, ok := _u.mutation.CreatedAt(); ok { _spec.SetField(stocktakeorder.FieldCreatedAt, field.TypeInt64, value) } if value, ok := _u.mutation.AddedCreatedAt(); ok { _spec.AddField(stocktakeorder.FieldCreatedAt, field.TypeInt64, value) } if value, ok := _u.mutation.FinishedAt(); ok { _spec.SetField(stocktakeorder.FieldFinishedAt, field.TypeInt64, value) } if value, ok := _u.mutation.AddedFinishedAt(); ok { _spec.AddField(stocktakeorder.FieldFinishedAt, field.TypeInt64, value) } if _u.mutation.FinishedAtCleared() { _spec.ClearField(stocktakeorder.FieldFinishedAt, field.TypeInt64) } if _node, err = sqlgraph.UpdateNodes(ctx, _u.driver, _spec); err != nil { if _, ok := err.(*sqlgraph.NotFoundError); ok { err = &NotFoundError{stocktakeorder.Label} } else if sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } return 0, err } _u.mutation.done = true return _node, nil } // StocktakeOrderUpdateOne is the builder for updating a single StocktakeOrder entity. type StocktakeOrderUpdateOne struct { config fields []string hooks []Hook mutation *StocktakeOrderMutation } // SetStocktakeNo sets the "stocktake_no" field. func (_u *StocktakeOrderUpdateOne) SetStocktakeNo(v string) *StocktakeOrderUpdateOne { _u.mutation.SetStocktakeNo(v) return _u } // SetNillableStocktakeNo sets the "stocktake_no" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableStocktakeNo(v *string) *StocktakeOrderUpdateOne { if v != nil { _u.SetStocktakeNo(*v) } return _u } // SetStatus sets the "status" field. func (_u *StocktakeOrderUpdateOne) SetStatus(v string) *StocktakeOrderUpdateOne { _u.mutation.SetStatus(v) return _u } // SetNillableStatus sets the "status" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableStatus(v *string) *StocktakeOrderUpdateOne { if v != nil { _u.SetStatus(*v) } return _u } // SetOperator sets the "operator" field. func (_u *StocktakeOrderUpdateOne) SetOperator(v string) *StocktakeOrderUpdateOne { _u.mutation.SetOperator(v) return _u } // SetNillableOperator sets the "operator" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableOperator(v *string) *StocktakeOrderUpdateOne { if v != nil { _u.SetOperator(*v) } return _u } // ClearOperator clears the value of the "operator" field. func (_u *StocktakeOrderUpdateOne) ClearOperator() *StocktakeOrderUpdateOne { _u.mutation.ClearOperator() return _u } // SetTotalTargets sets the "total_targets" field. func (_u *StocktakeOrderUpdateOne) SetTotalTargets(v int) *StocktakeOrderUpdateOne { _u.mutation.ResetTotalTargets() _u.mutation.SetTotalTargets(v) return _u } // SetNillableTotalTargets sets the "total_targets" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableTotalTargets(v *int) *StocktakeOrderUpdateOne { if v != nil { _u.SetTotalTargets(*v) } return _u } // AddTotalTargets adds value to the "total_targets" field. func (_u *StocktakeOrderUpdateOne) AddTotalTargets(v int) *StocktakeOrderUpdateOne { _u.mutation.AddTotalTargets(v) return _u } // SetDiffCount sets the "diff_count" field. func (_u *StocktakeOrderUpdateOne) SetDiffCount(v int) *StocktakeOrderUpdateOne { _u.mutation.ResetDiffCount() _u.mutation.SetDiffCount(v) return _u } // SetNillableDiffCount sets the "diff_count" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableDiffCount(v *int) *StocktakeOrderUpdateOne { if v != nil { _u.SetDiffCount(*v) } return _u } // AddDiffCount adds value to the "diff_count" field. func (_u *StocktakeOrderUpdateOne) AddDiffCount(v int) *StocktakeOrderUpdateOne { _u.mutation.AddDiffCount(v) return _u } // SetAdjusted sets the "adjusted" field. func (_u *StocktakeOrderUpdateOne) SetAdjusted(v bool) *StocktakeOrderUpdateOne { _u.mutation.SetAdjusted(v) return _u } // SetNillableAdjusted sets the "adjusted" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableAdjusted(v *bool) *StocktakeOrderUpdateOne { if v != nil { _u.SetAdjusted(*v) } return _u } // SetAdjustNo sets the "adjust_no" field. func (_u *StocktakeOrderUpdateOne) SetAdjustNo(v string) *StocktakeOrderUpdateOne { _u.mutation.SetAdjustNo(v) return _u } // SetNillableAdjustNo sets the "adjust_no" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableAdjustNo(v *string) *StocktakeOrderUpdateOne { if v != nil { _u.SetAdjustNo(*v) } return _u } // ClearAdjustNo clears the value of the "adjust_no" field. func (_u *StocktakeOrderUpdateOne) ClearAdjustNo() *StocktakeOrderUpdateOne { _u.mutation.ClearAdjustNo() return _u } // SetWrittenAt sets the "written_at" field. func (_u *StocktakeOrderUpdateOne) SetWrittenAt(v int64) *StocktakeOrderUpdateOne { _u.mutation.ResetWrittenAt() _u.mutation.SetWrittenAt(v) return _u } // SetNillableWrittenAt sets the "written_at" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableWrittenAt(v *int64) *StocktakeOrderUpdateOne { if v != nil { _u.SetWrittenAt(*v) } return _u } // AddWrittenAt adds value to the "written_at" field. func (_u *StocktakeOrderUpdateOne) AddWrittenAt(v int64) *StocktakeOrderUpdateOne { _u.mutation.AddWrittenAt(v) return _u } // ClearWrittenAt clears the value of the "written_at" field. func (_u *StocktakeOrderUpdateOne) ClearWrittenAt() *StocktakeOrderUpdateOne { _u.mutation.ClearWrittenAt() return _u } // SetWrittenBy sets the "written_by" field. func (_u *StocktakeOrderUpdateOne) SetWrittenBy(v string) *StocktakeOrderUpdateOne { _u.mutation.SetWrittenBy(v) return _u } // SetNillableWrittenBy sets the "written_by" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableWrittenBy(v *string) *StocktakeOrderUpdateOne { if v != nil { _u.SetWrittenBy(*v) } return _u } // ClearWrittenBy clears the value of the "written_by" field. func (_u *StocktakeOrderUpdateOne) ClearWrittenBy() *StocktakeOrderUpdateOne { _u.mutation.ClearWrittenBy() return _u } // SetUncountedMode sets the "uncounted_mode" field. func (_u *StocktakeOrderUpdateOne) SetUncountedMode(v string) *StocktakeOrderUpdateOne { _u.mutation.SetUncountedMode(v) return _u } // SetNillableUncountedMode sets the "uncounted_mode" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableUncountedMode(v *string) *StocktakeOrderUpdateOne { if v != nil { _u.SetUncountedMode(*v) } return _u } // ClearUncountedMode clears the value of the "uncounted_mode" field. func (_u *StocktakeOrderUpdateOne) ClearUncountedMode() *StocktakeOrderUpdateOne { _u.mutation.ClearUncountedMode() return _u } // SetCancelReason sets the "cancel_reason" field. func (_u *StocktakeOrderUpdateOne) SetCancelReason(v string) *StocktakeOrderUpdateOne { _u.mutation.SetCancelReason(v) return _u } // SetNillableCancelReason sets the "cancel_reason" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableCancelReason(v *string) *StocktakeOrderUpdateOne { if v != nil { _u.SetCancelReason(*v) } return _u } // ClearCancelReason clears the value of the "cancel_reason" field. func (_u *StocktakeOrderUpdateOne) ClearCancelReason() *StocktakeOrderUpdateOne { _u.mutation.ClearCancelReason() return _u } // SetCreatedAt sets the "created_at" field. func (_u *StocktakeOrderUpdateOne) SetCreatedAt(v int64) *StocktakeOrderUpdateOne { _u.mutation.ResetCreatedAt() _u.mutation.SetCreatedAt(v) return _u } // SetNillableCreatedAt sets the "created_at" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableCreatedAt(v *int64) *StocktakeOrderUpdateOne { if v != nil { _u.SetCreatedAt(*v) } return _u } // AddCreatedAt adds value to the "created_at" field. func (_u *StocktakeOrderUpdateOne) AddCreatedAt(v int64) *StocktakeOrderUpdateOne { _u.mutation.AddCreatedAt(v) return _u } // SetFinishedAt sets the "finished_at" field. func (_u *StocktakeOrderUpdateOne) SetFinishedAt(v int64) *StocktakeOrderUpdateOne { _u.mutation.ResetFinishedAt() _u.mutation.SetFinishedAt(v) return _u } // SetNillableFinishedAt sets the "finished_at" field if the given value is not nil. func (_u *StocktakeOrderUpdateOne) SetNillableFinishedAt(v *int64) *StocktakeOrderUpdateOne { if v != nil { _u.SetFinishedAt(*v) } return _u } // AddFinishedAt adds value to the "finished_at" field. func (_u *StocktakeOrderUpdateOne) AddFinishedAt(v int64) *StocktakeOrderUpdateOne { _u.mutation.AddFinishedAt(v) return _u } // ClearFinishedAt clears the value of the "finished_at" field. func (_u *StocktakeOrderUpdateOne) ClearFinishedAt() *StocktakeOrderUpdateOne { _u.mutation.ClearFinishedAt() return _u } // Mutation returns the StocktakeOrderMutation object of the builder. func (_u *StocktakeOrderUpdateOne) Mutation() *StocktakeOrderMutation { return _u.mutation } // Where appends a list predicates to the StocktakeOrderUpdate builder. func (_u *StocktakeOrderUpdateOne) Where(ps ...predicate.StocktakeOrder) *StocktakeOrderUpdateOne { _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 *StocktakeOrderUpdateOne) Select(field string, fields ...string) *StocktakeOrderUpdateOne { _u.fields = append([]string{field}, fields...) return _u } // Save executes the query and returns the updated StocktakeOrder entity. func (_u *StocktakeOrderUpdateOne) Save(ctx context.Context) (*StocktakeOrder, error) { return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks) } // SaveX is like Save, but panics if an error occurs. func (_u *StocktakeOrderUpdateOne) SaveX(ctx context.Context) *StocktakeOrder { node, err := _u.Save(ctx) if err != nil { panic(err) } return node } // Exec executes the query on the entity. func (_u *StocktakeOrderUpdateOne) Exec(ctx context.Context) error { _, err := _u.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (_u *StocktakeOrderUpdateOne) ExecX(ctx context.Context) { if err := _u.Exec(ctx); err != nil { panic(err) } } func (_u *StocktakeOrderUpdateOne) sqlSave(ctx context.Context) (_node *StocktakeOrder, err error) { _spec := sqlgraph.NewUpdateSpec(stocktakeorder.Table, stocktakeorder.Columns, sqlgraph.NewFieldSpec(stocktakeorder.FieldID, field.TypeInt)) id, ok := _u.mutation.ID() if !ok { return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "StocktakeOrder.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, stocktakeorder.FieldID) for _, f := range fields { if !stocktakeorder.ValidColumn(f) { return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)} } if f != stocktakeorder.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.StocktakeNo(); ok { _spec.SetField(stocktakeorder.FieldStocktakeNo, field.TypeString, value) } if value, ok := _u.mutation.Status(); ok { _spec.SetField(stocktakeorder.FieldStatus, field.TypeString, value) } if value, ok := _u.mutation.Operator(); ok { _spec.SetField(stocktakeorder.FieldOperator, field.TypeString, value) } if _u.mutation.OperatorCleared() { _spec.ClearField(stocktakeorder.FieldOperator, field.TypeString) } if value, ok := _u.mutation.TotalTargets(); ok { _spec.SetField(stocktakeorder.FieldTotalTargets, field.TypeInt, value) } if value, ok := _u.mutation.AddedTotalTargets(); ok { _spec.AddField(stocktakeorder.FieldTotalTargets, field.TypeInt, value) } if value, ok := _u.mutation.DiffCount(); ok { _spec.SetField(stocktakeorder.FieldDiffCount, field.TypeInt, value) } if value, ok := _u.mutation.AddedDiffCount(); ok { _spec.AddField(stocktakeorder.FieldDiffCount, field.TypeInt, value) } if value, ok := _u.mutation.Adjusted(); ok { _spec.SetField(stocktakeorder.FieldAdjusted, field.TypeBool, value) } if value, ok := _u.mutation.AdjustNo(); ok { _spec.SetField(stocktakeorder.FieldAdjustNo, field.TypeString, value) } if _u.mutation.AdjustNoCleared() { _spec.ClearField(stocktakeorder.FieldAdjustNo, field.TypeString) } if value, ok := _u.mutation.WrittenAt(); ok { _spec.SetField(stocktakeorder.FieldWrittenAt, field.TypeInt64, value) } if value, ok := _u.mutation.AddedWrittenAt(); ok { _spec.AddField(stocktakeorder.FieldWrittenAt, field.TypeInt64, value) } if _u.mutation.WrittenAtCleared() { _spec.ClearField(stocktakeorder.FieldWrittenAt, field.TypeInt64) } if value, ok := _u.mutation.WrittenBy(); ok { _spec.SetField(stocktakeorder.FieldWrittenBy, field.TypeString, value) } if _u.mutation.WrittenByCleared() { _spec.ClearField(stocktakeorder.FieldWrittenBy, field.TypeString) } if value, ok := _u.mutation.UncountedMode(); ok { _spec.SetField(stocktakeorder.FieldUncountedMode, field.TypeString, value) } if _u.mutation.UncountedModeCleared() { _spec.ClearField(stocktakeorder.FieldUncountedMode, field.TypeString) } if value, ok := _u.mutation.CancelReason(); ok { _spec.SetField(stocktakeorder.FieldCancelReason, field.TypeString, value) } if _u.mutation.CancelReasonCleared() { _spec.ClearField(stocktakeorder.FieldCancelReason, field.TypeString) } if value, ok := _u.mutation.CreatedAt(); ok { _spec.SetField(stocktakeorder.FieldCreatedAt, field.TypeInt64, value) } if value, ok := _u.mutation.AddedCreatedAt(); ok { _spec.AddField(stocktakeorder.FieldCreatedAt, field.TypeInt64, value) } if value, ok := _u.mutation.FinishedAt(); ok { _spec.SetField(stocktakeorder.FieldFinishedAt, field.TypeInt64, value) } if value, ok := _u.mutation.AddedFinishedAt(); ok { _spec.AddField(stocktakeorder.FieldFinishedAt, field.TypeInt64, value) } if _u.mutation.FinishedAtCleared() { _spec.ClearField(stocktakeorder.FieldFinishedAt, field.TypeInt64) } _node = &StocktakeOrder{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{stocktakeorder.Label} } else if sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } return nil, err } _u.mutation.done = true return _node, nil }