liujiandao
2024-04-30 76a84d89d63041232c646ca28c59239dd00f7fc5
models/yield_register_item.go
@@ -20,7 +20,9 @@
   }
   YieldRegisterItemSearch struct {
      YieldRegisterItem
      Orm *gorm.DB
      YieldRegisterIds []uint
      CarNumbers       []int
      Orm              *gorm.DB
   }
)
@@ -42,6 +44,15 @@
   return slf
}
func (slf *YieldRegisterItemSearch) SetYieldRegisterIds(ids []uint) *YieldRegisterItemSearch {
   slf.YieldRegisterIds = ids
   return slf
}
func (slf *YieldRegisterItemSearch) SetCarNumbers(carNumbers []int) *YieldRegisterItemSearch {
   slf.CarNumbers = carNumbers
   return slf
}
func (slf *YieldRegisterItemSearch) build() *gorm.DB {
   db := slf.Orm.Table(slf.TableName())
@@ -49,6 +60,14 @@
      db = db.Where("yield_register_id = ?", slf.YieldRegisterId)
   }
   if len(slf.YieldRegisterIds) > 0 {
      db = db.Where("yield_register_id in (?)", slf.YieldRegisterIds)
   }
   if len(slf.CarNumbers) > 0 {
      db = db.Where("car_number in (?)", slf.CarNumbers)
   }
   return db
}