From da154e5c758c3b67202390debdc5331a8f68c685 Mon Sep 17 00:00:00 2001
From: yinbentan <yinbentan@live.com>
Date: 星期六, 29 六月 2024 18:25:22 +0800
Subject: [PATCH] 库存结算时间点添加数据初始化任务

---
 models/operation.go |   55 ++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 50 insertions(+), 5 deletions(-)

diff --git a/models/operation.go b/models/operation.go
index 15e2fd9..9caeb40 100644
--- a/models/operation.go
+++ b/models/operation.go
@@ -4,6 +4,7 @@
 	"fmt"
 	"github.com/shopspring/decimal"
 	"gorm.io/gorm"
+	"time"
 	"wms/constvar"
 	"wms/pkg/mysqlx"
 )
@@ -25,7 +26,7 @@
 		OperationDate     string                   `json:"operationDate" gorm:"type:varchar(31);comment:瀹夋帓鏃ユ湡"`
 		ContacterID       int                      `json:"contacterID" gorm:"type:int;comment:鑱旂郴浜篒D"`
 		ContacterName     string                   `json:"contacterName" gorm:"type:varchar(63);comment:鑱旂郴浜哄鍚�"`
-		CompanyID         int                      `json:"companyID" gorm:"type:int;comment:鍏徃ID-瀹㈡埛"`
+		CompanyID         string                   `json:"companyID" gorm:"type:varchar(255);comment:鍏徃ID-瀹㈡埛"`
 		CompanyName       string                   `json:"companyName" gorm:"type:varchar(127);comment:鍏徃鍚嶇О-瀹㈡埛"`
 		Comment           string                   `json:"comment" gorm:"type:text;comment:澶囨敞"`
 		LogisticCompanyId string                   `json:"logisticCompanyId"   gorm:"type:varchar(191);comment:鐗╂祦鍏徃id"`
@@ -50,9 +51,20 @@
 		ToLocation         Location `json:"toLocation"      gorm:"foreignKey:ToLocationID;references:Id"` //鐩爣浣嶇疆
 		SalesDetailsNumber string   `gorm:"type:varchar(191);comment:閿�鍞槑缁嗙紪鐮�" json:"salesDetailsNumber"`
 
-		Manager    string `json:"manager" gorm:"type:varchar(255);comment:涓荤"`
-		Accountant string `json:"accountant" gorm:"type:varchar(255);comment:浼氳"`
-		Custodian  string `json:"custodian" gorm:"type:varchar(255);comment:淇濈鍛�"`
+		ManagerId    string `json:"managerId" gorm:"type:varchar(255);comment:涓荤id"`
+		Manager      string `json:"manager" gorm:"type:varchar(255);comment:涓荤鍚嶇О"`
+		AccountantId string `json:"accountantId" gorm:"type:varchar(255);comment:浼氳id"`
+		Accountant   string `json:"accountant" gorm:"type:varchar(255);comment:浼氳鍚嶇О"`
+		CustodianId  string `json:"custodianId" gorm:"type:varchar(255);comment:淇濈鍛榠d"`
+		Custodian    string `json:"custodian" gorm:"type:varchar(255);comment:淇濈鍛樺悕绉�"`
+		CreatedBy    string `json:"createBy" gorm:"type:varchar(255);comment:鍒涘缓鑰匲serId"`
+		CheckedBy    string `json:"checkedBy" gorm:"type:varchar(255);comment:楠岃瘉鑰匲serId"`
+		Remark       string `json:"remark"`
+
+		WarehouseId         int      `json:"warehouseId" gorm:"type:int;not null;comment:浠撳簱id"`
+		IsInternalOutput    bool     `json:"isInternalOutput"` //鏄惁璋冩嫧浜х敓鐨勫嚭搴�
+		InventoryDealerType int      `json:"inventoryDealerType" gorm:"type:varchar(255);comment:浼氳鍚嶇О"`
+		InventoryDealer     MiniDict `json:"inventoryDealer" gorm:"foreignKey:InventoryDealerType;"`
 	}
 
 	OperationSearch struct {
@@ -67,6 +79,9 @@
 		Ids                 []int
 		SourceNumbers       []string
 		SalesDetailsNumbers []string
+		Fields              string
+		BeginTime           time.Time
+		EndTime             time.Time
 	}
 )
 
@@ -100,6 +115,11 @@
 
 func (slf *OperationSearch) SetKeyword(keyword string) *OperationSearch {
 	slf.Keyword = keyword
+	return slf
+}
+
+func (slf *OperationSearch) SetFields(fields string) *OperationSearch {
+	slf.Fields = fields
 	return slf
 }
 
@@ -163,6 +183,12 @@
 	return slf
 }
 
+func (slf *OperationSearch) SetTimeBetween(beginTime, endTime time.Time) *OperationSearch {
+	slf.BeginTime = beginTime
+	slf.EndTime = endTime
+	return slf
+}
+
 func (slf *OperationSearch) build() *gorm.DB {
 	var db = slf.Orm.Model(&Operation{})
 
@@ -194,7 +220,7 @@
 	}
 
 	if slf.Preload {
-		db = db.Model(&Operation{}).Preload("Details").Preload("Details.Product").Preload("LogisticCompany").Preload("Location").Preload("ToLocation").Preload("Details.FromLocation").Preload("Details.ToLocation")
+		db = db.Model(&Operation{}).Preload("Details").Preload("Details.Product").Preload("LogisticCompany").Preload("Location").Preload("ToLocation").Preload("Details.FromLocation").Preload("Details.ToLocation").Preload("InventoryDealer")
 	}
 
 	if slf.Disuse {
@@ -225,6 +251,13 @@
 		db = db.Where("operation_source = ?", slf.OperationSource)
 	}
 
+	if slf.Fields != "" {
+		db = db.Select(slf.Fields)
+	}
+
+	if !slf.BeginTime.IsZero() && !slf.EndTime.IsZero() {
+		db = db.Where("created_at between ? and ?", slf.BeginTime, slf.EndTime)
+	}
 	return db
 }
 
@@ -348,6 +381,18 @@
 	return records, nil
 }
 
+func (slf *OperationSearch) FindIds() ([]int, error) {
+	var (
+		records = make([]int, 0)
+		db      = slf.build()
+	)
+	if err := db.Find(&records).Error; err != nil {
+		return records, fmt.Errorf("find records err: %v", err)
+	}
+
+	return records, nil
+}
+
 // FindByQuery 鎸囧畾鏉′欢鏌ヨ.
 func (slf *OperationSearch) FindByQuery(query string, args []interface{}) ([]*Operation, int64, error) {
 	var (

--
Gitblit v1.8.0