From a98b2ed2ddcf15db5b20907cd502e157d04d1b88 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 20 十月 2023 10:22:06 +0800 Subject: [PATCH] 去掉无用配置 --- models/location_product_amount.go | 29 +++++++++++++++++++++++------ 1 files changed, 23 insertions(+), 6 deletions(-) diff --git a/models/location_product_amount.go b/models/location_product_amount.go index ec242ff..b834d4b 100644 --- a/models/location_product_amount.go +++ b/models/location_product_amount.go @@ -20,12 +20,13 @@ 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 { @@ -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