From b9a62531fb93aba42d13e62fd1b03d8dba112fa6 Mon Sep 17 00:00:00 2001 From: liujiandao <274878379@qq.com> Date: 星期一, 23 十月 2023 12:04:31 +0800 Subject: [PATCH] 位置修改 --- models/location_product_amount.go | 31 ++++++++++++++++++++++++------- 1 files changed, 24 insertions(+), 7 deletions(-) diff --git a/models/location_product_amount.go b/models/location_product_amount.go index ec242ff..b51e00c 100644 --- a/models/location_product_amount.go +++ b/models/location_product_amount.go @@ -20,17 +20,18 @@ LocationProductAmountSearch struct { LocationProductAmount - Order string - PageNum int - PageSize int - Keyword string - Orm *gorm.DB - Preload bool + Order string + PageNum int + PageSize int + Keyword string + Orm *gorm.DB + Preload bool + LocationProductIds []int } LocationProductAmountWithOperation struct { //LocationProductAmount LocationProductAmount `json:"locationProductAmount"` - LocationProductAmountId int `json:"locationProductAmount" gorm:"location_product_amount_id"` + LocationProductAmountId int `json:"locationProductAmountId" gorm:"location_product_amount_id"` LocationId int `json:"locationId" gorm:"column:location_id"` LocationName string `json:"locationName" gorm:"column:location_name"` ProductId string `json:"productId" gorm:"column:product_id"` @@ -83,6 +84,16 @@ return slf } +func (slf *LocationProductAmountSearch) SetLocationProductId(id int) *LocationProductAmountSearch { + slf.LocationProductId = id + return slf +} + +func (slf *LocationProductAmountSearch) SetLocationProductIds(ids []int) *LocationProductAmountSearch { + slf.LocationProductIds = ids + return slf +} + func (slf *LocationProductAmountSearch) build() *gorm.DB { var db = slf.Orm.Model(&LocationProductAmount{}) @@ -100,6 +111,12 @@ if slf.Preload { db = db.Model(&LocationProductAmount{}).Preload("LocationProduct").Preload("LocationProduct.Location").Preload("LocationProduct.Product") } + if slf.LocationProductId != 0 { + db = db.Where("location_product_id=?", slf.LocationProductId) + } + if len(slf.LocationProductIds) > 0 { + db = db.Where("location_product_id in (?)", slf.LocationProductIds) + } return db } -- Gitblit v1.8.0