From 0f8093693d0cd3b8da71abac95c617aa850f4819 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期一, 22 四月 2024 15:44:55 +0800
Subject: [PATCH] 考勤管理
---
models/worker.go | 10 ++++++++++
1 files changed, 10 insertions(+), 0 deletions(-)
diff --git a/models/worker.go b/models/worker.go
index 07640e0..678a213 100644
--- a/models/worker.go
+++ b/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
}
--
Gitblit v1.8.0