From a904aa32e66688ba76c9e4b71b6738b0ba5f5d4f Mon Sep 17 00:00:00 2001
From: yinbentan <yinbentan@live.com>
Date: 星期四, 25 七月 2024 11:44:09 +0800
Subject: [PATCH] 添加字段,工种添加工种编码,用于计算工资(id无法固定工种信息)

---
 models/fineness_check.go |   22 ++++++++++++++++------
 1 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/models/fineness_check.go b/models/fineness_check.go
index 278189c..180aa1a 100644
--- a/models/fineness_check.go
+++ b/models/fineness_check.go
@@ -26,12 +26,13 @@
 	}
 	FinenessCheckSearch struct {
 		FinenessCheck
-		Order    string
-		PageNum  int
-		PageSize int
-		Orm      *gorm.DB
-		Preload  bool
-		Keyword  string
+		Order               string
+		PageNum             int
+		PageSize            int
+		Orm                 *gorm.DB
+		Preload             bool
+		Keyword             string
+		FinenessRegisterIDs []uint
 	}
 )
 
@@ -73,6 +74,11 @@
 	return slf
 }
 
+func (slf *FinenessCheckSearch) SetFinenessRegisterIDs(ids []uint) *FinenessCheckSearch {
+	slf.FinenessRegisterIDs = ids
+	return slf
+}
+
 func (slf *FinenessCheckSearch) SetKeyword(keyword string) *FinenessCheckSearch {
 	slf.Keyword = keyword
 	return slf
@@ -97,6 +103,10 @@
 		db = db.Where("fineness_register_id = ?", slf.FinenessRegisterID)
 	}
 
+	if len(slf.FinenessRegisterIDs) != 0 {
+		db = db.Where("fineness_register_id in ?", slf.FinenessRegisterIDs)
+	}
+
 	if slf.Preload {
 		db = db.Model(&FinenessCheck{}).Preload("FinenessRegister")
 	}

--
Gitblit v1.8.0