From 2716487d13f3521adeb25a48aa07a89f885d32c7 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期一, 09 十月 2023 09:36:08 +0800
Subject: [PATCH] 仓库功能修改

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

diff --git a/models/operation_type.go b/models/operation_type.go
index 26842f7..4c5f86a 100644
--- a/models/operation_type.go
+++ b/models/operation_type.go
@@ -103,6 +103,11 @@
 	return slf
 }
 
+func (slf *OperationTypeSearch) SetWarehouseId(warehouseId int) *OperationTypeSearch {
+	slf.WarehouseId = warehouseId
+	return slf
+}
+
 func (slf *OperationTypeSearch) build() *gorm.DB {
 	var db = slf.Orm.Model(&OperationType{})
 
@@ -126,6 +131,10 @@
 		db = db.Preload("Company").Preload("Warehouse").Preload("DefaultLocationSrc").Preload("DefaultLocationDest")
 	}
 
+	if slf.WarehouseId > 0 {
+		db = db.Where("warehouse_id = ?", slf.WarehouseId)
+	}
+
 	return db
 }
 

--
Gitblit v1.8.0