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