From c113e72efaceba986a733a44f999c567e4296fdb Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期五, 20 十月 2023 10:08:56 +0800
Subject: [PATCH] 位置报表修改

---
 models/location_product_amount.go |   21 +++++++++++++++------
 1 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/models/location_product_amount.go b/models/location_product_amount.go
index 9686a30..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 {
@@ -88,6 +89,11 @@
 	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{})
 
@@ -108,6 +114,9 @@
 	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