生丝检验部分逻辑修正

zhangqian

2024-03-19 178d6e7507ef4806ca02692c4ee24bb98dba8362

 models/workshop_manage.go

@@ -49,11 +49,29 @@
   return slf
}

func (slf *WorkshopManageSearch) SetWorkshopNumber(workshopNumber string) *WorkshopManageSearch {
   slf.WorkshopNumber = workshopNumber
   return slf
}

func (slf *WorkshopManageSearch) SetGroupNumber(groupNumber int) *WorkshopManageSearch {
   slf.GroupNumber = groupNumber
   return slf
}

func (slf *WorkshopManageSearch) build() *gorm.DB {
   db := slf.Orm.Table(slf.TableName())

   if slf.ID > 0 {
      db = db.Where("id = ?", slf.ID)
   }

   if slf.WorkshopNumber != "" {
      db = db.Where("workshop_number = ?", slf.WorkshopNumber)
   }

   if slf.GroupNumber != 0 {
      db = db.Where("group_number = ?", slf.GroupNumber)
   }

   return db
@@ -89,6 +107,19 @@
   return records, total, nil
}

func (slf *WorkshopManageSearch) FindAll() ([]*WorkshopManage, error) {
   var (
      records = make([]*WorkshopManage, 0)
      db      = slf.build()
   )

   if err := db.Find(&records).Error; err != nil {
      return records, fmt.Errorf("find records err: %v", err)
   }

   return records, nil
}

func (slf *WorkshopManageSearch) Save(record *WorkshopManage) error {
   var db = slf.build()