From 576039f5ee85910edd332aa4459a132b713e80bd Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期一, 18 三月 2024 20:00:50 +0800
Subject: [PATCH] 发货grpc接口

---
 models/operation_type.go |   22 ++++++++++++++++------
 1 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/models/operation_type.go b/models/operation_type.go
index 4857259..b206050 100644
--- a/models/operation_type.go
+++ b/models/operation_type.go
@@ -41,12 +41,13 @@
 
 	OperationTypeSearch struct {
 		OperationType
-		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
+		WarehouseIds []int
 	}
 
 	OperationTypeByStatus struct {
@@ -109,6 +110,11 @@
 	return slf
 }
 
+func (slf *OperationTypeSearch) SetWarehouseIds(warehouseIds []int) *OperationTypeSearch {
+	slf.WarehouseIds = warehouseIds
+	return slf
+}
+
 func (slf *OperationTypeSearch) build() *gorm.DB {
 	var db = slf.Orm.Model(&OperationType{})
 
@@ -136,6 +142,10 @@
 		db = db.Where("warehouse_id = ?", slf.WarehouseId)
 	}
 
+	if len(slf.WarehouseIds) > 0 {
+		db = db.Where("warehouse_id in ?", slf.WarehouseIds)
+	}
+
 	if int(slf.BaseOperationType) != 0 {
 		db = db.Where("base_operation_type = ?", slf.BaseOperationType)
 	}

--
Gitblit v1.8.0