From adb8a4d6c2806beae1ac97af7f71c8a3ce3ed0c9 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期二, 23 四月 2024 15:25:13 +0800
Subject: [PATCH] 修改考勤信息

---
 models/attendance_manage.go |   18 ++++++++++++++++++
 1 files changed, 18 insertions(+), 0 deletions(-)

diff --git a/models/attendance_manage.go b/models/attendance_manage.go
index ed14ff0..6663197 100644
--- a/models/attendance_manage.go
+++ b/models/attendance_manage.go
@@ -71,6 +71,16 @@
 	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) build() *gorm.DB {
 	var db = slf.Orm.Table(slf.TableName())
 
@@ -86,6 +96,14 @@
 		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)
+	}
+
 	return db
 }
 

--
Gitblit v1.8.0