zhangqian
2023-11-25 efec800478214f3a412f879d2b51c2968515995d
model/production_progress.go
@@ -20,10 +20,11 @@
   ProductionProgressSearch struct {
      ProductionProgress
      Order    string
      PageNum  int
      PageSize int
      Orm      *gorm.DB
      Order      string
      PageNum    int
      PageSize   int
      Orm        *gorm.DB
      UnFinished bool
   }
)
@@ -78,6 +79,11 @@
   return slf
}
func (slf *ProductionProgressSearch) SetUnFinished() *ProductionProgressSearch {
   slf.UnFinished = true
   return slf
}
func (slf *ProductionProgressSearch) build() *gorm.DB {
   var db = slf.Orm.Model(&ProductionProgress{})
@@ -109,6 +115,10 @@
      db = db.Where("channel = ?", slf.Channel)
   }
   if slf.UnFinished {
      db = db.Where("finished_quantity <  total_quantity")
   }
   return db
}
@@ -118,17 +128,6 @@
   if err := db.Create(record).Error; err != nil {
      return fmt.Errorf("create err: %v, record: %+v", err, record)
   }
   return nil
}
// CreateBatch 批量插入
func (slf *ProductionProgressSearch) CreateBatch(records []*ProductionProgress) error {
   var db = slf.build()
   if err := db.Create(&records).Error; err != nil {
      return fmt.Errorf("create batch err: %v, records: %+v", err, records)
   }
   return nil