From c950586b718ae6fc198bedf424609a4ac94cb5d1 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 04 七月 2024 18:16:07 +0800 Subject: [PATCH] 增加根据位置或仓库查询产品及库存列表 --- models/operation.go | 70 ++++++++++++++++++++++++++--------- 1 files changed, 52 insertions(+), 18 deletions(-) diff --git a/models/operation.go b/models/operation.go index 9caeb40..12ea942 100644 --- a/models/operation.go +++ b/models/operation.go @@ -61,27 +61,29 @@ 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;"` + WarehouseId int `json:"warehouseId" gorm:"type:int;not null;default:0;comment:浠撳簱id"` + Warehouse Warehouse `json:"warehouse" gorm:"foreignKey:WarehouseId"` + IsInternalOutput bool `json:"isInternalOutput"` //鏄惁璋冩嫧浜х敓鐨勫嚭搴� + InventoryDealerType int `json:"inventoryDealerType" gorm:"type:varchar(255);comment:璋冩嫧鍑哄叆搴撳垎绫�(瀵瑰簲dict瀛楀吀琛ㄧ殑ID)"` + InventoryDealer MiniDict `json:"inventoryDealer" gorm:"foreignKey:InventoryDealerType;"` } OperationSearch struct { Operation - 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 + 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 // } ) @@ -91,6 +93,12 @@ 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 + } } func (slf *OperationSearch) SetOrm(tx *gorm.DB) *OperationSearch { @@ -189,6 +197,16 @@ 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 +} + func (slf *OperationSearch) build() *gorm.DB { var db = slf.Orm.Model(&Operation{}) @@ -220,7 +238,14 @@ } if slf.Preload { - db = db.Model(&Operation{}).Preload("Details").Preload("Details.Product").Preload("LogisticCompany").Preload("Location").Preload("ToLocation").Preload("Details.FromLocation").Preload("Details.ToLocation").Preload("InventoryDealer") + db = db.Model(&Operation{}).Preload("Details"). + Preload("Details.Product"). + Preload("LogisticCompany"). + Preload("Location"). + Preload("ToLocation"). + Preload("Details.FromLocation"). + Preload("Details.ToLocation"). + Preload("InventoryDealer").Preload("Warehouse") } if slf.Disuse { @@ -258,6 +283,15 @@ 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 } -- Gitblit v1.8.0