yinbentan
2024-07-25 f874e81b77282079398c22c67af754a23d260c91
models/yield_register_circle.go
@@ -22,7 +22,9 @@
   }
   YieldRegisterCircleSearch struct {
      YieldRegisterCircle
      Orm *gorm.DB
      YieldRegisterIds []uint
      Order            string
      Orm              *gorm.DB
   }
)
@@ -44,11 +46,29 @@
   return slf
}
func (slf *YieldRegisterCircleSearch) SetYieldRegisterIds(ids []uint) *YieldRegisterCircleSearch {
   slf.YieldRegisterIds = ids
   return slf
}
func (slf *YieldRegisterCircleSearch) SetOrder(order string) *YieldRegisterCircleSearch {
   slf.Order = order
   return slf
}
func (slf *YieldRegisterCircleSearch) build() *gorm.DB {
   db := slf.Orm.Table(slf.TableName())
   if slf.YieldRegisterId > 0 {
      db = db.Where("yield_register_id = ?", slf.YieldRegisterId)
   }
   if len(slf.YieldRegisterIds) > 0 {
      db = db.Where("yield_register_id in ?", slf.YieldRegisterIds)
   }
   if slf.Order != "" {
      db = db.Order(slf.Order)
   }
   return db
@@ -70,7 +90,7 @@
      db      = slf.build()
   )
   if err := db.Find(&records).Error; err != nil {
   if err := db.Order("car_number").Find(&records).Error; err != nil {
      return records, fmt.Errorf("find records err: %v", err)
   }