liujiandao
2024-04-22 0f8093693d0cd3b8da71abac95c617aa850f4819
models/worker.go
@@ -32,6 +32,7 @@
      PageSize int
      Orm      *gorm.DB
      Fields   string
      Ids      []string
   }
)
@@ -93,6 +94,11 @@
   return slf
}
func (slf *WorkerSearch) SetIds(ids []string) *WorkerSearch {
   slf.Ids = ids
   return slf
}
func (slf *WorkerSearch) build() *gorm.DB {
   var db = slf.Orm.Table(slf.TableName())
@@ -129,6 +135,10 @@
      db = db.Select(slf.Fields)
   }
   if len(slf.Ids) > 0 {
      db = db.Where("id in (?)", slf.Ids)
   }
   return db
}