From 7a3a9a739392d048655efac6a28592410a8809c2 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期五, 17 十一月 2023 09:57:29 +0800
Subject: [PATCH] 在库产品类型修改

---
 models/location_product_amount.go |   31 +++++++++++++++++++++++++++++++
 1 files changed, 31 insertions(+), 0 deletions(-)

diff --git a/models/location_product_amount.go b/models/location_product_amount.go
index 6f3f767..4f1af2b 100644
--- a/models/location_product_amount.go
+++ b/models/location_product_amount.go
@@ -35,6 +35,7 @@
 		//LocationProductIds []int
 		LocationIds []int
 		ProductIds  []string
+		Ids         []int
 	}
 
 	LocationProductAmountWithOperation struct {
@@ -80,6 +81,11 @@
 
 func (slf *LocationProductAmountSearch) SetID(id int) *LocationProductAmountSearch {
 	slf.Id = id
+	return slf
+}
+
+func (slf *LocationProductAmountSearch) SetIds(ids []int) *LocationProductAmountSearch {
+	slf.Ids = ids
 	return slf
 }
 
@@ -151,6 +157,10 @@
 
 	if slf.ProductId != "" {
 		db = db.Where("product_id=?", slf.ProductId)
+	}
+
+	if len(slf.Ids) > 0 {
+		db = db.Where("id in (?)", slf.Ids)
 	}
 
 	return db
@@ -247,3 +257,24 @@
 
 	return record, db.First(&record)
 }
+
+func (slf *LocationProductAmountSearch) Count() (int64, error) {
+	var (
+		total int64
+		db    = slf.build()
+	)
+	err := db.Count(&total).Error
+	return total, err
+}
+
+func (slf *LocationProductAmountSearch) UpdateByMap(upMap map[string]interface{}) error {
+	var (
+		db = slf.build()
+	)
+
+	if err := db.Updates(upMap).Error; err != nil {
+		return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap)
+	}
+
+	return nil
+}

--
Gitblit v1.8.0