From 09af2f801c0f5e18222b696bb6df7da382e7d977 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期四, 11 四月 2024 10:35:22 +0800
Subject: [PATCH] 切换产量登记表

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

diff --git a/models/yield_register.go b/models/yield_register.go
index 1163c19..8111b1e 100644
--- a/models/yield_register.go
+++ b/models/yield_register.go
@@ -80,6 +80,31 @@
 	return slf
 }
 
+func (slf *YieldRegisterSearch) SetCreateTime(createTime string) *YieldRegisterSearch {
+	slf.CreateTime = createTime
+	return slf
+}
+
+func (slf *YieldRegisterSearch) SetMarketId(marketId uint) *YieldRegisterSearch {
+	slf.MarketId = marketId
+	return slf
+}
+
+func (slf *YieldRegisterSearch) SetWorkshopId(workshopId uint) *YieldRegisterSearch {
+	slf.WorkshopId = workshopId
+	return slf
+}
+
+func (slf *YieldRegisterSearch) SetGroupNumber(groupNumber int) *YieldRegisterSearch {
+	slf.GroupNumber = groupNumber
+	return slf
+}
+
+func (slf *YieldRegisterSearch) SetSpec(spec string) *YieldRegisterSearch {
+	slf.Spec = spec
+	return slf
+}
+
 func (slf *YieldRegisterSearch) build() *gorm.DB {
 	db := slf.Orm.Table(slf.TableName())
 
@@ -91,6 +116,25 @@
 		db = db.Where("number = ?", slf.Number)
 	}
 
+	if slf.CreateTime != "" {
+		db = db.Where("create_time = ?", slf.CreateTime)
+	}
+
+	if slf.WorkshopId > 0 {
+		db = db.Where("workshop_id = ?", slf.WorkshopId)
+	}
+
+	if slf.MarketId > 0 {
+		db = db.Where("market_id = ?", slf.MarketId)
+	}
+
+	if slf.GroupNumber > 0 {
+		db = db.Where("group_number = ?", slf.GroupNumber)
+	}
+
+	if slf.Spec != "" {
+		db = db.Where("spec = ?", slf.Spec)
+	}
 	return db
 }
 

--
Gitblit v1.8.0