From 1691d8ee8fe10799d98b296625fd30183407ef25 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 28 六月 2024 12:17:22 +0800
Subject: [PATCH] 新增操作类型增加基础操作类型和仓库id来支持库存盘点

---
 controllers/operation.go |   37 ++++++++++++++++++++++++++++---------
 1 files changed, 28 insertions(+), 9 deletions(-)

diff --git a/controllers/operation.go b/controllers/operation.go
index af310c4..e174c67 100644
--- a/controllers/operation.go
+++ b/controllers/operation.go
@@ -72,17 +72,28 @@
 		util.ResponseFormat(c, code.RequestParamError, "鏄庣粏涓笉鑳藉瓨鍦ㄩ噸澶嶇殑浜у搧")
 		return
 	}
-
-	operationType, err := models.NewOperationTypeSearch().SetID(uint(params.OperationTypeId)).First()
-	if err != nil {
-		util.ResponseFormat(c, code.RequestParamError, err.Error())
-		return
+	var operationType *models.OperationType
+	var err error
+	if params.OperationTypeId == 0 {
+		operationType, err = models.NewOperationTypeSearch().SetWarehouseId(params.WarehouseId).SetBaseOperationType(params.BaseOperationType).First()
+		if err != nil || params.WarehouseId != operationType.WarehouseId {
+			util.ResponseFormat(c, code.RequestParamError, "鍙傛暟缂哄け")
+			return
+		}
+	} else {
+		operationType, err = models.NewOperationTypeSearch().SetID(uint(params.OperationTypeId)).First()
+		if err != nil {
+			util.ResponseFormat(c, code.RequestParamError, err.Error())
+			return
+		}
 	}
+
 	userInfo := middleware.GetUserInfo(c)
 	params.CreatedBy = userInfo.Username
 	params.Status = constvar.OperationStatus_Ready
 	//params.Number = strconv.FormatInt(time.Now().Unix(), 10)
 	params.BaseOperationType = operationType.BaseOperationType
+	params.WarehouseId = operationType.WarehouseId
 
 	var numberNum int64
 	if err := mysqlx.GetDB().Model(&models.Operation{}).Where("number=?", params.Number).Count(&numberNum).Error; err != nil {
@@ -191,7 +202,13 @@
 	}
 
 	if params.OperationTypeId == 0 && int(params.BaseOperationType) == 0 {
-		return errors.New("鏈瘑鍒璁板綍绫诲瀷")
+		return errors.New("鏈瘑鍒璁板綍绫诲瀷1")
+	}
+
+	if params.BaseOperationType != 0 {
+		if params.WarehouseId == 0 {
+			return errors.New("鏈瘑鍒璁板綍绫诲瀷2")
+		}
 	}
 
 	if params.OperationDate == "" {
@@ -282,7 +299,9 @@
 		costMap[material.ID] = material.Cost
 		salePriceMap[material.ID] = material.SalePrice
 	}
-	list, total, err := search.SetOperationTypeId(params.OperationTypeId).SetPreload(true).SetOrder("created_at desc").Find()
+	list, total, err := search.SetOperationTypeId(params.OperationTypeId).
+		SetBaseOperationType(params.BaseOperationType).
+		SetPreload(true).SetOrder("created_at desc").Find()
 	if err != nil {
 		util.ResponseFormat(c, code.RequestError, "鏌ユ壘澶辫触:"+err.Error())
 		return
@@ -303,8 +322,8 @@
 	if !params.PageInfo.Check() {
 		return errors.New("鏁版嵁鍒嗛〉淇℃伅閿欒")
 	}
-	if params.OperationTypeId == 0 {
-		return errors.New("operationTypeId涓�0")
+	if params.OperationTypeId == 0 && params.BaseOperationType == 0 {
+		return errors.New("operationTypeId鍜宐aseOperationType涓嶈兘鍚屾椂涓�0")
 	}
 	return nil
 }

--
Gitblit v1.8.0