From 7d06c6472f1acea6b2e49e7b8837bf555e94e219 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 07 五月 2024 16:21:06 +0800
Subject: [PATCH] fox

---
 models/yield_register.go |   24 ++++++++++++++++++------
 1 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/models/yield_register.go b/models/yield_register.go
index 8508d60..b22e269 100644
--- a/models/yield_register.go
+++ b/models/yield_register.go
@@ -14,7 +14,7 @@
 		Number                  string                 `json:"number" gorm:"type:varchar(255);not null;comment:缂栧彿"`
 		CreateTime              string                 `json:"createTime" gorm:"type:varchar(255);comment:鍒涘缓鏃堕棿"`
 		MarketId                uint                   `json:"marketId" gorm:"type:int(11);comment:搴勫彛id"`
-		WorkshopId              uint                   `json:"workshopId" gorm:"type:int(11);comment:杞﹂棿id"`
+		WorkshopNumber          string                 `json:"workshopNumber" gorm:"type:varchar(255);comment:杞﹂棿缂栫爜"`
 		GroupNumber             int                    `json:"groupNumber" gorm:"type:int(11);comment:缁勫埆"`
 		Spec                    string                 `json:"spec" gorm:"type:varchar(255);comment:瑙勬牸"`
 		JieZhuang               string                 `json:"jieZhuang" gorm:"type:varchar(255);comment:鏄惁缁撳簞"`
@@ -56,7 +56,7 @@
 )
 
 func (slf YieldRegister) TableName() string {
-	return "yield_register"
+	return "silk_yield_register"
 }
 
 func NewYieldRegisterSearch() *YieldRegisterSearch {
@@ -93,8 +93,8 @@
 	return slf
 }
 
-func (slf *YieldRegisterSearch) SetWorkshopId(workshopId uint) *YieldRegisterSearch {
-	slf.WorkshopId = workshopId
+func (slf *YieldRegisterSearch) SetWorkshopNumber(workshopNumber string) *YieldRegisterSearch {
+	slf.WorkshopNumber = workshopNumber
 	return slf
 }
 
@@ -123,8 +123,8 @@
 		db = db.Where("create_time = ?", slf.CreateTime)
 	}
 
-	if slf.WorkshopId > 0 {
-		db = db.Where("workshop_id = ?", slf.WorkshopId)
+	if slf.WorkshopNumber != "" {
+		db = db.Where("workshop_number = ?", slf.WorkshopNumber)
 	}
 
 	if slf.MarketId > 0 {
@@ -216,3 +216,15 @@
 
 	return nil
 }
+
+func (slf *YieldRegisterSearch) Count() (int64, error) {
+	var (
+		total int64
+		db    = slf.build()
+	)
+
+	if err := db.Count(&total).Error; err != nil {
+		return total, fmt.Errorf("find count err: %v", err)
+	}
+	return total, nil
+}

--
Gitblit v1.8.0