From b9a62531fb93aba42d13e62fd1b03d8dba112fa6 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期一, 23 十月 2023 12:04:31 +0800
Subject: [PATCH] 位置修改

---
 models/operation.go |   77 ++++++++++++++++++++++++++++++++------
 1 files changed, 64 insertions(+), 13 deletions(-)

diff --git a/models/operation.go b/models/operation.go
index d1c7545..faa78a2 100644
--- a/models/operation.go
+++ b/models/operation.go
@@ -28,13 +28,8 @@
 		CompanyName       string                   `json:"companyName" gorm:"type:varchar(127);comment:鍏徃鍚嶇О-瀹㈡埛"`
 		Comment           string                   `json:"comment" gorm:"type:text;comment:澶囨敞"`
 
-		Details []*OperationDetails `json:"details" gorm:"foreignKey:OperationID;references:Id"`
-
-		//Weight          decimal.Decimal          `json:"weight" gorm:"type:decimal(20,2);comment:閲嶉噺(kg)"`
-		//TransferWeight  decimal.Decimal          `json:"transferWeight" gorm:"type:decimal(20,2);comment:鐗╂祦閲嶉噺(kg)"`
-		//CarrierID       int                      `json:"carrierID" gorm:"type:int;comment:鎵胯繍鍟咺D"`
-		//CarrierName     string                   `json:"carrierName" gorm:"type:varchar(63);comment:鎵胯繍鍟嗗悕绉�"`
-		//Tracking        string                   `json:"tracking" gorm:"type:varchar(127);comment:杩借釜鍙傝��"`
+		Details           []*OperationDetails        `json:"details" gorm:"foreignKey:OperationID;references:Id"`
+		BaseOperationType constvar.BaseOperationType `json:"baseOperationType" gorm:"type:tinyint;not null;comment:鍩虹浣滀笟绫诲瀷"` //鍩虹浣滀笟绫诲瀷
 	}
 
 	OperationSearch struct {
@@ -45,6 +40,8 @@
 		Keyword  string
 		Orm      *gorm.DB
 		Preload  bool
+		Disuse   bool
+		Ids      []int
 	}
 )
 
@@ -91,6 +88,36 @@
 	return slf
 }
 
+func (slf *OperationSearch) SetNumber(number string) *OperationSearch {
+	slf.Number = number
+	return slf
+}
+
+func (slf *OperationSearch) SetSourceNumber(sourceNumber string) *OperationSearch {
+	slf.SourceNumber = sourceNumber
+	return slf
+}
+
+func (slf *OperationSearch) SetDisuse(disuse bool) *OperationSearch {
+	slf.Disuse = disuse
+	return slf
+}
+
+func (slf *OperationSearch) SetIds(ids []int) *OperationSearch {
+	slf.Ids = ids
+	return slf
+}
+
+func (slf *OperationSearch) SetStatus(status constvar.OperationStatus) *OperationSearch {
+	slf.Status = status
+	return slf
+}
+
+func (slf *OperationSearch) SetBaseOperationType(baseOperationType constvar.BaseOperationType) *OperationSearch {
+	slf.BaseOperationType = baseOperationType
+	return slf
+}
+
 func (slf *OperationSearch) build() *gorm.DB {
 	var db = slf.Orm.Model(&Operation{})
 
@@ -101,16 +128,40 @@
 		db = db.Order(slf.Order)
 	}
 
-	//if slf.Keyword != "" {
-	//	db = db.Where("product_name like ?", fmt.Sprintf("%%%v%%", slf.Keyword))
-	//}
+	if slf.Keyword != "" {
+		db = db.Where("number like ? or source_number like ?", fmt.Sprintf("%%%v%%", slf.Keyword), fmt.Sprintf("%%%v%%", slf.Keyword))
+	}
+
+	if slf.SourceNumber != "" {
+		db = db.Where("source_number like ?", fmt.Sprintf("%%%v%%", slf.SourceNumber))
+	}
+
+	if slf.Number != "" {
+		db = db.Where("number like ?", fmt.Sprintf("%%%v%%", slf.Number))
+	}
 
 	if slf.OperationTypeId != 0 {
 		db.Where("operation_type_id = ?", slf.OperationTypeId)
 	}
 
 	if slf.Preload {
-		db = db.Model(&Operation{}).Preload("Details").Preload("FromLocation").Preload("ToLocation")
+		db = db.Model(&Operation{}).Preload("Details").Preload("Details.Product").Preload("FromLocation").Preload("ToLocation")
+	}
+
+	if slf.Disuse {
+		db = db.Where("operation_type_id = ?", 0)
+	}
+
+	if len(slf.Ids) > 0 {
+		db = db.Where("id in (?)", slf.Ids)
+	}
+
+	if slf.Status > 0 {
+		db = db.Where("status = ?", slf.Status)
+	}
+
+	if slf.BaseOperationType != 0 {
+		db = db.Where("base_operation_type = ?", slf.BaseOperationType)
 	}
 
 	return db
@@ -213,7 +264,7 @@
 	if slf.PageNum*slf.PageSize > 0 {
 		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
 	}
-	if err := db.Find(&records).Error; err != nil {
+	if err := db.Preload("FromLocation").Preload("ToLocation").Find(&records).Error; err != nil {
 		return records, total, fmt.Errorf("find records err: %v", err)
 	}
 
@@ -229,7 +280,7 @@
 	if slf.PageNum*slf.PageSize > 0 {
 		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
 	}
-	if err := db.Find(&records).Error; err != nil {
+	if err := db.Preload("FromLocation").Preload("ToLocation").Find(&records).Error; err != nil {
 		return records, fmt.Errorf("find records err: %v", err)
 	}
 

--
Gitblit v1.8.0