From 178d6e7507ef4806ca02692c4ee24bb98dba8362 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 19 三月 2024 19:19:46 +0800
Subject: [PATCH] 生丝检验部分逻辑修正

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

diff --git a/models/workshop_manage.go b/models/workshop_manage.go
index 6d920cb..dc5166b 100644
--- a/models/workshop_manage.go
+++ b/models/workshop_manage.go
@@ -49,11 +49,29 @@
 	return slf
 }
 
+func (slf *WorkshopManageSearch) SetWorkshopNumber(workshopNumber string) *WorkshopManageSearch {
+	slf.WorkshopNumber = workshopNumber
+	return slf
+}
+
+func (slf *WorkshopManageSearch) SetGroupNumber(groupNumber int) *WorkshopManageSearch {
+	slf.GroupNumber = groupNumber
+	return slf
+}
+
 func (slf *WorkshopManageSearch) build() *gorm.DB {
 	db := slf.Orm.Table(slf.TableName())
 
 	if slf.ID > 0 {
 		db = db.Where("id = ?", slf.ID)
+	}
+
+	if slf.WorkshopNumber != "" {
+		db = db.Where("workshop_number = ?", slf.WorkshopNumber)
+	}
+
+	if slf.GroupNumber != 0 {
+		db = db.Where("group_number = ?", slf.GroupNumber)
 	}
 
 	return db
@@ -89,6 +107,19 @@
 	return records, total, nil
 }
 
+func (slf *WorkshopManageSearch) FindAll() ([]*WorkshopManage, error) {
+	var (
+		records = make([]*WorkshopManage, 0)
+		db      = slf.build()
+	)
+
+	if err := db.Find(&records).Error; err != nil {
+		return records, fmt.Errorf("find records err: %v", err)
+	}
+
+	return records, nil
+}
+
 func (slf *WorkshopManageSearch) Save(record *WorkshopManage) error {
 	var db = slf.build()
 

--
Gitblit v1.8.0