From 7edecfcf0f48ba6d6177486a57399a558c4c79e7 Mon Sep 17 00:00:00 2001 From: jiangshuai <291802688@qq.com> Date: 星期三, 27 九月 2023 11:14:17 +0800 Subject: [PATCH] 1.二期一部分代码 --- models/location_product.go | 18 ++++++++++++++++++ 1 files changed, 18 insertions(+), 0 deletions(-) diff --git a/models/location_product.go b/models/location_product.go index b3edb61..bc98181 100644 --- a/models/location_product.go +++ b/models/location_product.go @@ -69,6 +69,16 @@ return slf } +func (slf *LocationProductSearch) SetProductId(productId string) *LocationProductSearch { + slf.ProductId = productId + return slf +} + +func (slf *LocationProductSearch) SetLocationId(locationId int) *LocationProductSearch { + slf.LocationId = locationId + return slf +} + func (slf *LocationProductSearch) build() *gorm.DB { var db = slf.Orm.Model(&LocationProduct{}) @@ -87,6 +97,14 @@ db = db.Model(&LocationProduct{}).Preload("Location").Preload("Area").Preload("ProductCategory").Preload("Product") } + if slf.LocationId != 0 { + db = db.Where("location_id = ?", slf.LocationId) + } + + if slf.ProductId != "" { + db = db.Where("product_id=?", slf.ProductId) + } + return db } -- Gitblit v1.8.0