zhangqian
2024-04-29 440392a932fb2dc8118caf1f4996e7d8b72574a1
models/attendance_manage.go
@@ -34,6 +34,8 @@
      Preload  bool
      Ids      []uint
      Month    string
      Keyword  string
      Order    string
      Orm      *gorm.DB
   }
)
@@ -71,6 +73,26 @@
   return slf
}
func (slf *AttendanceManageSearch) SetDate(date string) *AttendanceManageSearch {
   slf.Date = date
   return slf
}
func (slf *AttendanceManageSearch) SetWorkerId(workerId string) *AttendanceManageSearch {
   slf.WorkerId = workerId
   return slf
}
func (slf *AttendanceManageSearch) SetKeyword(keyword string) *AttendanceManageSearch {
   slf.Keyword = keyword
   return slf
}
func (slf *AttendanceManageSearch) SetOrder(order string) *AttendanceManageSearch {
   slf.Order = order
   return slf
}
func (slf *AttendanceManageSearch) build() *gorm.DB {
   var db = slf.Orm.Table(slf.TableName())
@@ -86,6 +108,21 @@
      db = db.Where("date like ?", slf.Month+"%")
   }
   if slf.Date != "" {
      db = db.Where("date = ?", slf.Date)
   }
   if slf.WorkerId != "" {
      db = db.Where("worker_id = ?", slf.WorkerId)
   }
   if slf.Keyword != "" {
      db = db.Where("worker_name like ? or worker_id like ?", "%"+slf.Keyword+"%", "%"+slf.Keyword+"%")
   }
   if slf.Order != "" {
      db = db.Order(slf.Order)
   }
   return db
}