From f45e4cb8045fe60f96f467dc4ea1e2ae57445712 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 03 七月 2024 23:13:40 +0800
Subject: [PATCH] 默认位置parent_id为视图位置id

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

diff --git a/models/operation.go b/models/operation.go
index 740659d..9e2846c 100644
--- a/models/operation.go
+++ b/models/operation.go
@@ -95,6 +95,12 @@
 	return &OperationSearch{Orm: mysqlx.GetDB()}
 }
 
+func (slf *OperationSearch) BeforeCreate(tx *gorm.DB) {
+	for k := range slf.Details {
+		slf.Details[k].BaseOperationType = slf.BaseOperationType
+	}
+}
+
 func (slf *OperationSearch) SetOrm(tx *gorm.DB) *OperationSearch {
 	slf.Orm = tx
 	return slf
@@ -196,6 +202,11 @@
 	return slf
 }
 
+func (slf *OperationSearch) SetWarehouseId(warehouseId int) *OperationSearch {
+	slf.WarehouseId = warehouseId
+	return slf
+}
+
 func (slf *OperationSearch) build() *gorm.DB {
 	var db = slf.Orm.Model(&Operation{})
 
@@ -277,6 +288,10 @@
 		db = db.Where("inventory_dealer_type in (?)", slf.InventoryDealerTypeIds)
 	}
 
+	if slf.WarehouseId != 0 {
+		db = db.Where("warehouse_id = ?", slf.WarehouseId)
+	}
+
 	return db
 }
 

--
Gitblit v1.8.0