zhangqian
2023-10-20 bc7ee359dfe4e66a05c2cca9deb7a945534009f3
model/device.go
@@ -19,10 +19,11 @@
   DeviceSearch struct {
      Device
      Order    string
      PageNum  int
      PageSize int
      Orm      *gorm.DB
      Order     string
      PageNum   int
      PageSize  int
      Orm       *gorm.DB
      DeviceIDs []string
   }
)
@@ -64,6 +65,11 @@
   return slf
}
func (slf *DeviceSearch) SetDeviceIds(deviceIds []string) *DeviceSearch {
   slf.DeviceIDs = deviceIds
   return slf
}
func (slf *DeviceSearch) build() *gorm.DB {
   var db = slf.Orm.Table(slf.TableName())
@@ -75,6 +81,10 @@
      db = db.Where("device_id = ?", slf.DeviceID)
   }
   if len(slf.DeviceIDs) != 0 {
      db = db.Where("device_id in (?)", slf.DeviceIDs)
   }
   if slf.Order != "" {
      db = db.Order(slf.Order)
   }