From 527bc709815b9ea0f2ed14450262c9a5832c141f Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 14 三月 2024 16:09:26 +0800
Subject: [PATCH] 根据仓库返回产品剩余数量

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

diff --git a/models/warehouse.go b/models/warehouse.go
index ebdf7b3..5832dd7 100644
--- a/models/warehouse.go
+++ b/models/warehouse.go
@@ -38,6 +38,7 @@
 		Keyword  string
 		Orm      *gorm.DB
 		Preload  bool
+		Codes    []string
 	}
 )
 
@@ -79,6 +80,11 @@
 	return slf
 }
 
+func (slf *WarehouseSearch) SetCodes(codes []string) *WarehouseSearch {
+	slf.Codes = codes
+	return slf
+}
+
 func (slf *WarehouseSearch) SetName(name string) *WarehouseSearch {
 	slf.Name = name
 	return slf
@@ -116,10 +122,15 @@
 	if slf.Preload {
 		db = db.Preload("Company")
 	}
+
 	if slf.Code != "" {
 		db = db.Where("code = ?", slf.Code)
 	}
 
+	if len(slf.Codes) > 0 {
+		db = db.Where("code in ?", slf.Codes)
+	}
+
 	return db
 }
 

--
Gitblit v1.8.0