From 9e864a183115ba2420797a8bda661187b1069e00 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期二, 23 四月 2024 11:46:41 +0800
Subject: [PATCH] 考勤统计与车间修改

---
 models/fineness.go |   29 ++++++++++++++++++++++++++++-
 1 files changed, 28 insertions(+), 1 deletions(-)

diff --git a/models/fineness.go b/models/fineness.go
index c82e5a5..4ea1e11 100644
--- a/models/fineness.go
+++ b/models/fineness.go
@@ -37,7 +37,7 @@
 )
 
 func (slf *FinenessRegister) TableName() string {
-	return "fineness_register"
+	return "silk_fineness_register"
 }
 
 func NewFinenessRegisterSearch() *FinenessRegisterSearch {
@@ -79,6 +79,21 @@
 	return slf
 }
 
+func (slf *FinenessRegisterSearch) SetWorkshop(workshop string) *FinenessRegisterSearch {
+	slf.Workshop = workshop
+	return slf
+}
+
+func (slf *FinenessRegisterSearch) SetMarket(market string) *FinenessRegisterSearch {
+	slf.Market = market
+	return slf
+}
+
+func (slf *FinenessRegisterSearch) SetWorkshopGroup(group int) *FinenessRegisterSearch {
+	slf.WorkshopGroup = group
+	return slf
+}
+
 func (slf *FinenessRegisterSearch) build() *gorm.DB {
 	var db = slf.Orm.Table(slf.TableName())
 
@@ -102,6 +117,18 @@
 		db = db.Where("number like ?", fmt.Sprintf("%%%s%%", slf.Keyword))
 	}
 
+	if slf.Workshop != "" {
+		db = db.Where("workshop = ?", slf.Workshop)
+	}
+
+	if slf.Market != "" {
+		db = db.Where("market = ?", slf.Market)
+	}
+
+	if slf.WorkshopGroup > 0 {
+		db = db.Where("workshop_group = ?", slf.WorkshopGroup)
+	}
+
 	return db
 }
 

--
Gitblit v1.8.0