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