// Code generated by ent, DO NOT EDIT. package ent import ( "bj_power_mes/ent/bomitem" "context" "errors" "fmt" "time" "entgo.io/ent/dialect/sql/sqlgraph" "entgo.io/ent/schema/field" ) // BomItemCreate is the builder for creating a BomItem entity. type BomItemCreate struct { config mutation *BomItemMutation hooks []Hook } // SetProductCode sets the "productCode" field. func (_c *BomItemCreate) SetProductCode(v string) *BomItemCreate { _c.mutation.SetProductCode(v) return _c } // SetNillableProductCode sets the "productCode" field if the given value is not nil. func (_c *BomItemCreate) SetNillableProductCode(v *string) *BomItemCreate { if v != nil { _c.SetProductCode(*v) } return _c } // SetMaterialCode sets the "materialCode" field. func (_c *BomItemCreate) SetMaterialCode(v string) *BomItemCreate { _c.mutation.SetMaterialCode(v) return _c } // SetMaterialName sets the "materialName" field. func (_c *BomItemCreate) SetMaterialName(v string) *BomItemCreate { _c.mutation.SetMaterialName(v) return _c } // SetNillableMaterialName sets the "materialName" field if the given value is not nil. func (_c *BomItemCreate) SetNillableMaterialName(v *string) *BomItemCreate { if v != nil { _c.SetMaterialName(*v) } return _c } // SetSpec sets the "spec" field. func (_c *BomItemCreate) SetSpec(v string) *BomItemCreate { _c.mutation.SetSpec(v) return _c } // SetNillableSpec sets the "spec" field if the given value is not nil. func (_c *BomItemCreate) SetNillableSpec(v *string) *BomItemCreate { if v != nil { _c.SetSpec(*v) } return _c } // SetUnit sets the "unit" field. func (_c *BomItemCreate) SetUnit(v string) *BomItemCreate { _c.mutation.SetUnit(v) return _c } // SetNillableUnit sets the "unit" field if the given value is not nil. func (_c *BomItemCreate) SetNillableUnit(v *string) *BomItemCreate { if v != nil { _c.SetUnit(*v) } return _c } // SetManageMode sets the "manageMode" field. func (_c *BomItemCreate) SetManageMode(v string) *BomItemCreate { _c.mutation.SetManageMode(v) return _c } // SetNillableManageMode sets the "manageMode" field if the given value is not nil. func (_c *BomItemCreate) SetNillableManageMode(v *string) *BomItemCreate { if v != nil { _c.SetManageMode(*v) } return _c } // SetUnitQty sets the "unitQty" field. func (_c *BomItemCreate) SetUnitQty(v float64) *BomItemCreate { _c.mutation.SetUnitQty(v) return _c } // SetNillableUnitQty sets the "unitQty" field if the given value is not nil. func (_c *BomItemCreate) SetNillableUnitQty(v *float64) *BomItemCreate { if v != nil { _c.SetUnitQty(*v) } return _c } // SetLossRate sets the "lossRate" field. func (_c *BomItemCreate) SetLossRate(v float64) *BomItemCreate { _c.mutation.SetLossRate(v) return _c } // SetNillableLossRate sets the "lossRate" field if the given value is not nil. func (_c *BomItemCreate) SetNillableLossRate(v *float64) *BomItemCreate { if v != nil { _c.SetLossRate(*v) } return _c } // SetRequiredQty sets the "requiredQty" field. func (_c *BomItemCreate) SetRequiredQty(v float64) *BomItemCreate { _c.mutation.SetRequiredQty(v) return _c } // SetNillableRequiredQty sets the "requiredQty" field if the given value is not nil. func (_c *BomItemCreate) SetNillableRequiredQty(v *float64) *BomItemCreate { if v != nil { _c.SetRequiredQty(*v) } return _c } // SetSerialSns sets the "serialSns" field. func (_c *BomItemCreate) SetSerialSns(v []string) *BomItemCreate { _c.mutation.SetSerialSns(v) return _c } // SetCreatedAt sets the "createdAt" field. func (_c *BomItemCreate) SetCreatedAt(v time.Time) *BomItemCreate { _c.mutation.SetCreatedAt(v) return _c } // SetNillableCreatedAt sets the "createdAt" field if the given value is not nil. func (_c *BomItemCreate) SetNillableCreatedAt(v *time.Time) *BomItemCreate { if v != nil { _c.SetCreatedAt(*v) } return _c } // SetID sets the "id" field. func (_c *BomItemCreate) SetID(v int) *BomItemCreate { _c.mutation.SetID(v) return _c } // Mutation returns the BomItemMutation object of the builder. func (_c *BomItemCreate) Mutation() *BomItemMutation { return _c.mutation } // Save creates the BomItem in the database. func (_c *BomItemCreate) Save(ctx context.Context) (*BomItem, error) { _c.defaults() return withHooks(ctx, _c.sqlSave, _c.mutation, _c.hooks) } // SaveX calls Save and panics if Save returns an error. func (_c *BomItemCreate) SaveX(ctx context.Context) *BomItem { v, err := _c.Save(ctx) if err != nil { panic(err) } return v } // Exec executes the query. func (_c *BomItemCreate) Exec(ctx context.Context) error { _, err := _c.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (_c *BomItemCreate) ExecX(ctx context.Context) { if err := _c.Exec(ctx); err != nil { panic(err) } } // defaults sets the default values of the builder before save. func (_c *BomItemCreate) defaults() { if _, ok := _c.mutation.ProductCode(); !ok { v := bomitem.DefaultProductCode _c.mutation.SetProductCode(v) } if _, ok := _c.mutation.MaterialName(); !ok { v := bomitem.DefaultMaterialName _c.mutation.SetMaterialName(v) } if _, ok := _c.mutation.Spec(); !ok { v := bomitem.DefaultSpec _c.mutation.SetSpec(v) } if _, ok := _c.mutation.Unit(); !ok { v := bomitem.DefaultUnit _c.mutation.SetUnit(v) } if _, ok := _c.mutation.ManageMode(); !ok { v := bomitem.DefaultManageMode _c.mutation.SetManageMode(v) } if _, ok := _c.mutation.UnitQty(); !ok { v := bomitem.DefaultUnitQty _c.mutation.SetUnitQty(v) } if _, ok := _c.mutation.LossRate(); !ok { v := bomitem.DefaultLossRate _c.mutation.SetLossRate(v) } if _, ok := _c.mutation.RequiredQty(); !ok { v := bomitem.DefaultRequiredQty _c.mutation.SetRequiredQty(v) } if _, ok := _c.mutation.CreatedAt(); !ok { v := bomitem.DefaultCreatedAt() _c.mutation.SetCreatedAt(v) } } // check runs all checks and user-defined validators on the builder. func (_c *BomItemCreate) check() error { if _, ok := _c.mutation.ProductCode(); !ok { return &ValidationError{Name: "productCode", err: errors.New(`ent: missing required field "BomItem.productCode"`)} } if v, ok := _c.mutation.ProductCode(); ok { if err := bomitem.ProductCodeValidator(v); err != nil { return &ValidationError{Name: "productCode", err: fmt.Errorf(`ent: validator failed for field "BomItem.productCode": %w`, err)} } } if _, ok := _c.mutation.MaterialCode(); !ok { return &ValidationError{Name: "materialCode", err: errors.New(`ent: missing required field "BomItem.materialCode"`)} } if v, ok := _c.mutation.MaterialCode(); ok { if err := bomitem.MaterialCodeValidator(v); err != nil { return &ValidationError{Name: "materialCode", err: fmt.Errorf(`ent: validator failed for field "BomItem.materialCode": %w`, err)} } } if _, ok := _c.mutation.MaterialName(); !ok { return &ValidationError{Name: "materialName", err: errors.New(`ent: missing required field "BomItem.materialName"`)} } if v, ok := _c.mutation.MaterialName(); ok { if err := bomitem.MaterialNameValidator(v); err != nil { return &ValidationError{Name: "materialName", err: fmt.Errorf(`ent: validator failed for field "BomItem.materialName": %w`, err)} } } if _, ok := _c.mutation.Spec(); !ok { return &ValidationError{Name: "spec", err: errors.New(`ent: missing required field "BomItem.spec"`)} } if v, ok := _c.mutation.Spec(); ok { if err := bomitem.SpecValidator(v); err != nil { return &ValidationError{Name: "spec", err: fmt.Errorf(`ent: validator failed for field "BomItem.spec": %w`, err)} } } if _, ok := _c.mutation.Unit(); !ok { return &ValidationError{Name: "unit", err: errors.New(`ent: missing required field "BomItem.unit"`)} } if v, ok := _c.mutation.Unit(); ok { if err := bomitem.UnitValidator(v); err != nil { return &ValidationError{Name: "unit", err: fmt.Errorf(`ent: validator failed for field "BomItem.unit": %w`, err)} } } if _, ok := _c.mutation.ManageMode(); !ok { return &ValidationError{Name: "manageMode", err: errors.New(`ent: missing required field "BomItem.manageMode"`)} } if v, ok := _c.mutation.ManageMode(); ok { if err := bomitem.ManageModeValidator(v); err != nil { return &ValidationError{Name: "manageMode", err: fmt.Errorf(`ent: validator failed for field "BomItem.manageMode": %w`, err)} } } if _, ok := _c.mutation.UnitQty(); !ok { return &ValidationError{Name: "unitQty", err: errors.New(`ent: missing required field "BomItem.unitQty"`)} } if _, ok := _c.mutation.LossRate(); !ok { return &ValidationError{Name: "lossRate", err: errors.New(`ent: missing required field "BomItem.lossRate"`)} } if _, ok := _c.mutation.RequiredQty(); !ok { return &ValidationError{Name: "requiredQty", err: errors.New(`ent: missing required field "BomItem.requiredQty"`)} } if _, ok := _c.mutation.CreatedAt(); !ok { return &ValidationError{Name: "createdAt", err: errors.New(`ent: missing required field "BomItem.createdAt"`)} } if v, ok := _c.mutation.ID(); ok { if err := bomitem.IDValidator(v); err != nil { return &ValidationError{Name: "id", err: fmt.Errorf(`ent: validator failed for field "BomItem.id": %w`, err)} } } return nil } func (_c *BomItemCreate) sqlSave(ctx context.Context) (*BomItem, error) { if err := _c.check(); err != nil { return nil, err } _node, _spec := _c.createSpec() if err := sqlgraph.CreateNode(ctx, _c.driver, _spec); err != nil { if sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } return nil, err } if _spec.ID.Value != _node.ID { id := _spec.ID.Value.(int64) _node.ID = int(id) } _c.mutation.id = &_node.ID _c.mutation.done = true return _node, nil } func (_c *BomItemCreate) createSpec() (*BomItem, *sqlgraph.CreateSpec) { var ( _node = &BomItem{config: _c.config} _spec = sqlgraph.NewCreateSpec(bomitem.Table, sqlgraph.NewFieldSpec(bomitem.FieldID, field.TypeInt)) ) if id, ok := _c.mutation.ID(); ok { _node.ID = id _spec.ID.Value = id } if value, ok := _c.mutation.ProductCode(); ok { _spec.SetField(bomitem.FieldProductCode, field.TypeString, value) _node.ProductCode = value } if value, ok := _c.mutation.MaterialCode(); ok { _spec.SetField(bomitem.FieldMaterialCode, field.TypeString, value) _node.MaterialCode = value } if value, ok := _c.mutation.MaterialName(); ok { _spec.SetField(bomitem.FieldMaterialName, field.TypeString, value) _node.MaterialName = value } if value, ok := _c.mutation.Spec(); ok { _spec.SetField(bomitem.FieldSpec, field.TypeString, value) _node.Spec = value } if value, ok := _c.mutation.Unit(); ok { _spec.SetField(bomitem.FieldUnit, field.TypeString, value) _node.Unit = value } if value, ok := _c.mutation.ManageMode(); ok { _spec.SetField(bomitem.FieldManageMode, field.TypeString, value) _node.ManageMode = value } if value, ok := _c.mutation.UnitQty(); ok { _spec.SetField(bomitem.FieldUnitQty, field.TypeFloat64, value) _node.UnitQty = value } if value, ok := _c.mutation.LossRate(); ok { _spec.SetField(bomitem.FieldLossRate, field.TypeFloat64, value) _node.LossRate = value } if value, ok := _c.mutation.RequiredQty(); ok { _spec.SetField(bomitem.FieldRequiredQty, field.TypeFloat64, value) _node.RequiredQty = value } if value, ok := _c.mutation.SerialSns(); ok { _spec.SetField(bomitem.FieldSerialSns, field.TypeJSON, value) _node.SerialSns = value } if value, ok := _c.mutation.CreatedAt(); ok { _spec.SetField(bomitem.FieldCreatedAt, field.TypeTime, value) _node.CreatedAt = value } return _node, _spec } // BomItemCreateBulk is the builder for creating many BomItem entities in bulk. type BomItemCreateBulk struct { config err error builders []*BomItemCreate } // Save creates the BomItem entities in the database. func (_c *BomItemCreateBulk) Save(ctx context.Context) ([]*BomItem, error) { if _c.err != nil { return nil, _c.err } specs := make([]*sqlgraph.CreateSpec, len(_c.builders)) nodes := make([]*BomItem, len(_c.builders)) mutators := make([]Mutator, len(_c.builders)) for i := range _c.builders { func(i int, root context.Context) { builder := _c.builders[i] builder.defaults() var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { mutation, ok := m.(*BomItemMutation) if !ok { return nil, fmt.Errorf("unexpected mutation type %T", m) } if err := builder.check(); err != nil { return nil, err } builder.mutation = mutation var err error nodes[i], specs[i] = builder.createSpec() if i < len(mutators)-1 { _, err = mutators[i+1].Mutate(root, _c.builders[i+1].mutation) } else { spec := &sqlgraph.BatchCreateSpec{Nodes: specs} // Invoke the actual operation on the latest mutation in the chain. if err = sqlgraph.BatchCreate(ctx, _c.driver, spec); err != nil { if sqlgraph.IsConstraintError(err) { err = &ConstraintError{msg: err.Error(), wrap: err} } } } if err != nil { return nil, err } mutation.id = &nodes[i].ID if specs[i].ID.Value != nil && nodes[i].ID == 0 { id := specs[i].ID.Value.(int64) nodes[i].ID = int(id) } mutation.done = true return nodes[i], nil }) for i := len(builder.hooks) - 1; i >= 0; i-- { mut = builder.hooks[i](mut) } mutators[i] = mut }(i, ctx) } if len(mutators) > 0 { if _, err := mutators[0].Mutate(ctx, _c.builders[0].mutation); err != nil { return nil, err } } return nodes, nil } // SaveX is like Save, but panics if an error occurs. func (_c *BomItemCreateBulk) SaveX(ctx context.Context) []*BomItem { v, err := _c.Save(ctx) if err != nil { panic(err) } return v } // Exec executes the query. func (_c *BomItemCreateBulk) Exec(ctx context.Context) error { _, err := _c.Save(ctx) return err } // ExecX is like Exec, but panics if an error occurs. func (_c *BomItemCreateBulk) ExecX(ctx context.Context) { if err := _c.Exec(ctx); err != nil { panic(err) } }