From ccf9ccce0fefbee8428c98a008c26d8ea26b2af1 Mon Sep 17 00:00:00 2001 From: yinbentan <yinbentan@live.com> Date: 星期四, 18 七月 2024 10:26:38 +0800 Subject: [PATCH] 修改方法,修复BUG --- models/operation.go | 249 +++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 216 insertions(+), 33 deletions(-) diff --git a/models/operation.go b/models/operation.go index b0d64d4..66a33be 100644 --- a/models/operation.go +++ b/models/operation.go @@ -2,50 +2,86 @@ import ( "fmt" + "github.com/shopspring/decimal" "gorm.io/gorm" + "time" "wms/constvar" "wms/pkg/mysqlx" ) type ( - // Operation 鎿嶄綔琛� + // Operation 搴撳瓨鎿嶄綔琛� Operation struct { WmsModel Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` - Number string `json:"number" gorm:"type:varchar(255)"` //鍗曞彿 - SourceNumber string `json:"sourceNumber" gorm:"type:varchar(255)"` //婧愬崟鍙� - OperationTypeId int `json:"operationTypeId" gorm:"type:int;not null;comment:浣滀笟绫诲瀷id"` //浣滀笟绫诲瀷id - OperationTypeName string `json:"operationTypeName" gorm:"type:varchar(127);comment:浣滀笟绫诲瀷鍚嶇О"` //浣滀笟绫诲瀷鍚嶇О - Status constvar.OperationStatus `json:"status" gorm:"type:int(11);not null;comment:鐘舵��"` //鐘舵�� - FromLocationID int `json:"fromLocationId" gorm:"type:int;not null;comment:婧愪綅缃甶d"` //婧愪綅缃甶d - FromLocation Location `json:"fromLocation" gorm:"foreignKey:FromLocationID;references:Id"` //婧愪綅缃� - ToLocationID int `json:"toLocationId" gorm:"type:int;not null;comment:鐩爣浣嶇疆id"` //鐩爣浣嶇疆id - ToLocation Location `json:"toLocation" gorm:"foreignKey:ToLocationID;references:Id"` //鐩爣浣嶇疆 - 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-瀹㈡埛"` - CompanyName string `json:"companyName" gorm:"type:varchar(127);comment:鍏徃鍚嶇О-瀹㈡埛"` - Comment string `json:"comment" gorm:"type:text;comment:澶囨敞"` + Number string `json:"number" gorm:"type:varchar(255);comment:鍗曞彿"` //鍗曞彿 + SourceNumber string `json:"sourceNumber" gorm:"index;type:varchar(255);comment:婧愬崟鍙�"` //婧愬崟鍙� + OperationTypeId int `json:"operationTypeId" gorm:"type:int;not null;comment:浣滀笟绫诲瀷id"` //浣滀笟绫诲瀷id + OperationTypeName string `json:"operationTypeName" gorm:"type:varchar(127);comment:浣滀笟绫诲瀷鍚嶇О"` //浣滀笟绫诲瀷鍚嶇О + Status constvar.OperationStatus `json:"status" gorm:"type:int(11);not null;comment:鐘舵��"` //鐘舵�� + OperationDate string `json:"operationDate" gorm:"type:varchar(31);comment:瀹夋帓鏃ユ湡"` // 瀹夋帓鏃ユ湡 + ContacterID int `json:"contacterID" gorm:"type:int;comment:鑱旂郴浜篒D"` // 鑱旂郴浜篒D + ContacterName string `json:"contacterName" gorm:"type:varchar(63);comment:鑱旂郴浜哄鍚�"` // 鑱旂郴浜哄鍚� + CompanyID string `json:"companyID" gorm:"type:varchar(255);comment:鍏徃ID-瀹㈡埛"` // 鍏徃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"` // 鐗╂祦鍏徃id + LogisticCompany LogisticCompany `json:"logisticCompany" gorm:"foreignKey:LogisticCompanyId"` // 鐗╂祦鍏徃淇℃伅 + WaybillNumber string `json:"waybillNumber" gorm:"type:varchar(255);comment:杩愬崟鍙�"` //杩愬崟鍙� + Weight decimal.Decimal `json:"weight" gorm:"type:decimal(20,2);comment:閲嶉噺" ` //閲嶉噺 + LogisticWeight decimal.Decimal `json:"logisticWeight" gorm:"type:decimal(20,2);comment:鐗╂祦閲嶉噺" ` //鐗╂祦閲嶉噺 + Source string `json:"source" gorm:"type:varchar(255);comment:鏉ユ簮绯荤粺,鐢ㄤ簬杩斿洖淇敼鐘舵��"` // 鏉ユ簮绯荤粺 + OperationSource constvar.OperationSource `json:"operationSource" gorm:"type:tinyint(3);not null;default:0;comment:鎿嶄綔鏉ユ簮"` //鎿嶄綔鏉ユ簮 - Details []*OperationDetails `json:"details" gorm:"foreignKey:OperationID;references:Id"` + Details []*OperationDetails `json:"details" gorm:"foreignKey:OperationID;references:Id"` // 鎿嶄綔鏄庣粏 + BaseOperationType constvar.BaseOperationType `json:"baseOperationType" gorm:"type:tinyint;not null;comment:鍩虹浣滀笟绫诲瀷"` //鍩虹浣滀笟绫诲瀷 - //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:杩借釜鍙傝��"` + AuditDate string `json:"auditDate" gorm:"type:varchar(31);comment:瀹℃壒鏃堕棿"` // 瀹℃壒鏃堕棿 + ReceiverName string `json:"receiverName" gorm:"type:varchar(31);comment:鏀惰揣浜哄鍚�"` // 鏀惰揣浜哄鍚� + ReceiverPhone string `json:"receiverPhone" gorm:"type:varchar(31);comment:鑱旂郴鐢佃瘽"` // 鑱旂郴鐢佃瘽 + ReceiverAddr string `json:"receiverAddr" gorm:"type:varchar(255);comment:鏀惰揣鍦板潃"` // 鏀惰揣鍦板潃 + + LocationID int `json:"locationID" gorm:"type:int;not null;comment:婧愪綅缃甶d"` //婧愪綅缃甶d + Location Location `json:"location" gorm:"foreignkey:LocationID;references:Id"` //婧愪綅缃� + ToLocationID int `json:"toLocationId" gorm:"type:int;not null;comment:浠撳簱浣嶇疆id"` //鐩爣浣嶇疆id + ToLocation Location `json:"toLocation" gorm:"foreignKey:ToLocationID;references:Id"` //鐩爣浣嶇疆 + SalesDetailsNumber string `json:"salesDetailsNumber" gorm:"type:varchar(191);comment:閿�鍞槑缁嗙紪鐮�"` // 閿�鍞槑缁嗙紪鐮� + + ManagerId string `json:"managerId" gorm:"type:varchar(255);comment:涓荤id"` // 涓荤id + Manager string `json:"manager" gorm:"type:varchar(255);comment:涓荤鍚嶇О"` // 涓荤鍚嶇О + AccountantId string `json:"accountantId" gorm:"type:varchar(255);comment:浼氳id"` // 浼氳id + Accountant string `json:"accountant" gorm:"type:varchar(255);comment:浼氳鍚嶇О"` // 浼氳鍚嶇О + CustodianId string `json:"custodianId" gorm:"type:varchar(255);comment:淇濈鍛榠d"` // 淇濈鍛榠d + Custodian string `json:"custodian" gorm:"type:varchar(255);comment:淇濈鍛樺悕绉�"` // 淇濈鍛樺悕绉� + CreatedBy string `json:"createBy" gorm:"type:varchar(255);comment:鍒涘缓鑰匲serId"` // 鍒涘缓鑰匲serId + CheckedBy string `json:"checkedBy" gorm:"type:varchar(255);comment:楠岃瘉鑰匲serId"` // 楠岃瘉鑰匲serId + Remark string `json:"remark"` // 澶囨敞 + + WarehouseId int `json:"warehouseId" gorm:"type:int;not null;default:0;comment:浠撳簱id"` // 浠撳簱id + Warehouse Warehouse `json:"warehouse" gorm:"foreignKey:WarehouseId"` // 浠撳簱淇℃伅 + IsInternalOutput bool `json:"isInternalOutput" gorm:"type:tinyint(1);comment:鏄惁璋冩嫧浜х敓鐨勫嚭搴�"` //鏄惁璋冩嫧浜х敓鐨勫嚭搴� + DealerType string `json:"dealerType" gorm:"type:varchar(255);comment:璋冩嫧鍑哄叆搴撶被鍨�"` // 璋冩嫧鍑哄叆搴撶被鍨� + + // 鍢夎仈浠撳偍娣诲姞 SilkMarket + SilkMarket string `json:"silkMarket" gorm:"type:varchar(255);comment:搴勫彛"` // 搴勫彛 } OperationSearch struct { Operation - Order string - PageNum int - PageSize int - Keyword string - Orm *gorm.DB - Preload bool - Disuse bool + Order string + PageNum int + PageSize int + Keyword string + Orm *gorm.DB + Preload bool + Disuse bool + Ids []int + SourceNumbers []string + SalesDetailsNumbers []string + Fields string + BeginTime time.Time + EndTime time.Time + InventoryDealerTypeIds []int // } ) @@ -55,6 +91,13 @@ func NewOperationSearch() *OperationSearch { return &OperationSearch{Orm: mysqlx.GetDB()} +} + +func (slf *OperationSearch) BeforeCreate(tx *gorm.DB) { + for k := range slf.Details { + slf.Details[k].BaseOperationType = slf.BaseOperationType + slf.Details[k].DealerType = slf.DealerType + } } func (slf *OperationSearch) SetOrm(tx *gorm.DB) *OperationSearch { @@ -82,6 +125,11 @@ return slf } +func (slf *OperationSearch) SetFields(fields string) *OperationSearch { + slf.Fields = fields + return slf +} + func (slf *OperationSearch) SetOperationTypeId(operationTypeId int) *OperationSearch { slf.OperationTypeId = operationTypeId return slf @@ -102,8 +150,59 @@ return slf } +func (slf *OperationSearch) SetSourceNumbers(numbers []string) *OperationSearch { + slf.SourceNumbers = numbers + 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) SetSalesDetailsNumber(salesDetailsNumber string) *OperationSearch { + slf.SalesDetailsNumber = salesDetailsNumber + return slf +} + +func (slf *OperationSearch) SetSalesDetailsNumbers(salesDetailsNumbers []string) *OperationSearch { + slf.SalesDetailsNumbers = salesDetailsNumbers + return slf +} + +func (slf *OperationSearch) SetOperationSource(operationSource constvar.OperationSource) *OperationSearch { + slf.OperationSource = operationSource + return slf +} + +func (slf *OperationSearch) SetTimeBetween(beginTime, endTime time.Time) *OperationSearch { + slf.BeginTime = beginTime + slf.EndTime = endTime + return slf +} + +func (slf *OperationSearch) SetInventoryDealerTypeIds(inventoryDealerTypeArr []int) *OperationSearch { + slf.InventoryDealerTypeIds = inventoryDealerTypeArr + return slf +} + +func (slf *OperationSearch) SetWarehouseId(warehouseId int) *OperationSearch { + slf.WarehouseId = warehouseId return slf } @@ -118,11 +217,15 @@ } if slf.Keyword != "" { - db = db.Where("product_name like ?", fmt.Sprintf("%%%v%%", 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)) + db = db.Where("source_number = ?", slf.SourceNumber) + } + + if len(slf.SourceNumbers) > 0 { + db = db.Where("source_number in (?)", slf.SourceNumbers) } if slf.Number != "" { @@ -134,11 +237,57 @@ } if slf.Preload { - db = db.Model(&Operation{}).Preload("Details").Preload("FromLocation").Preload("ToLocation") + db = db.Model(&Operation{}).Preload("Details"). + Preload("Details.Product"). + Preload("LogisticCompany"). + Preload("Location"). + Preload("ToLocation"). + Preload("Details.FromLocation"). + Preload("Details.ToLocation").Preload("Warehouse") } 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) + } + + if slf.SalesDetailsNumber != "" { + db = db.Where("sales_details_number = ?", slf.SalesDetailsNumber) + } + + if len(slf.SalesDetailsNumbers) > 0 { + db = db.Where("sales_details_number in (?)", slf.SalesDetailsNumbers) + } + + if slf.OperationSource != 0 { + 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) + } + + if len(slf.InventoryDealerTypeIds) > 0 { + db = db.Where("inventory_dealer_type in (?)", slf.InventoryDealerTypeIds) + } + + if slf.WarehouseId != 0 { + db = db.Where("warehouse_id = ?", slf.WarehouseId) } return db @@ -241,7 +390,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.Order("created_at desc").Find(&records).Error; err != nil { return records, total, fmt.Errorf("find records err: %v", err) } @@ -257,6 +406,18 @@ 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 { + return records, fmt.Errorf("find records err: %v", err) + } + + 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) } @@ -301,3 +462,25 @@ return records, nil } + +func (slf *OperationSearch) MaxAutoIncr() (int, error) { + var ( + db = slf.build() + total int64 + ) + + if err := db.Count(&total).Error; err != nil { + return int(total), fmt.Errorf("max err: %v", err) + } + return int(total), nil +} + +func (slf *OperationSearch) Count() (int64, error) { + var ( + total int64 + db = slf.build() + ) + + err := db.Count(&total).Error + return total, err +} -- Gitblit v1.8.0