From a0125ef3d6ce35be7b6dc4919c4192dba4a7329a Mon Sep 17 00:00:00 2001
From: jiangshuai <291802688@qq.com>
Date: 星期四, 16 十一月 2023 14:33:51 +0800
Subject: [PATCH] 报废数量bug

---
 controllers/operation.go |  352 ++++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 316 insertions(+), 36 deletions(-)

diff --git a/controllers/operation.go b/controllers/operation.go
index 478bff6..1e56fd1 100644
--- a/controllers/operation.go
+++ b/controllers/operation.go
@@ -1,20 +1,29 @@
 package controllers
 
 import (
+	"context"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"github.com/gin-gonic/gin"
 	"github.com/shopspring/decimal"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
 	"gorm.io/gorm"
+	"os"
 	"sort"
 	"strconv"
 	"time"
+	"wms/conf"
 	"wms/constvar"
 	"wms/extend/code"
 	"wms/extend/util"
 	"wms/models"
+	"wms/opa"
 	"wms/pkg/logx"
 	"wms/pkg/structx"
+	"wms/proto/product_inventory"
+	"wms/proto/purchase_wms"
 	"wms/request"
 )
 
@@ -71,7 +80,7 @@
 }
 
 func CheckDetailsRepeat(details []*models.OperationDetails) bool {
-	detailIDs := []string{}
+	var detailIDs []string
 	var tempID string
 	for _, v := range details {
 		detailIDs = append(detailIDs, v.ProductId)
@@ -254,7 +263,7 @@
 	util.ResponseFormat(c, code.Success, "淇敼鎴愬姛")
 }
 
-// DeleteDevice
+// Delete
 //
 //	@Tags		鍏ュ簱/鍑哄簱
 //	@Summary	鍒犻櫎鍏ュ簱/鍑哄簱淇℃伅
@@ -296,7 +305,7 @@
 	util.ResponseFormat(c, code.Success, "鍒犻櫎鎴愬姛")
 }
 
-// DeleteDevice
+// Finish
 //
 //	@Tags		鍏ュ簱/鍑哄簱
 //	@Summary	鏇存敼璁板綍鐘舵��
@@ -346,7 +355,7 @@
 		}
 
 		if operation.BaseOperationType == constvar.BaseOperationTypeIncoming {
-			operationInputs := []*models.Operation{}
+			var operationInputs []*models.Operation
 			for k, v := range listProdt {
 				value, ok := mapProdt[v.ID]
 				if !ok {
@@ -362,6 +371,7 @@
 				if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
 					return err
 				}
+
 				if err == nil {
 					operationTransfer := &models.Operation{
 						Number:            operation.Number,
@@ -379,20 +389,10 @@
 						Comment:           operation.Comment,
 						BaseOperationType: constvar.BaseOperationTypeInternal,
 						Details: []*models.OperationDetails{
-							&models.OperationDetails{
+							{
 								ProductId: v.ID,
 								Amount:    value,
 							},
-						},
-					}
-					operationTransfer.Id = 0
-					operationTransfer.FromLocationID = locationRule.AreaId
-					operationTransfer.ToLocationID = locationRule.LocationId
-					operationTransfer.Status = constvar.OperationStatus_Finish
-					operationTransfer.Details = []*models.OperationDetails{
-						&models.OperationDetails{
-							ProductId: v.ID,
-							Amount:    value,
 						},
 					}
 					operationInputs = append(operationInputs, operationTransfer)
@@ -413,24 +413,77 @@
 					locAmount.ProductId = v.ID
 					locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
 					locAmount.Amount = locAmount.Amount.Add(value)
+					locAmount.ProductCategoryID = v.CategoryId
 					if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
 						return res.Error
 					}
 				} else {
-					//TODO:鍑哄叆搴撶殑finish鍜屾姤搴熺殑finish閮借澧炲姞瀵筶ocation_product_amount琛ㄦ暟閲忕殑鏇存柊,鍥犱负姝よ〃鏈塒roductCategory瀛楁锛屾墍浠peration_details琛ㄤ腑瑕佸鍔燩roductCategoryId瀛楁
-					locAmount, err := models.NewLocationProductAmountSearch().
-						SetProductId(v.ID).
-						SetLocationId(operation.ToLocationID).
-						First()
+					locationRule, err = models.NewLocationProductSearch().SetProductCategoryId(v.CategoryId).SetAreaId(operation.ToLocationID).First()
 					if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
 						return err
 					}
-					locAmount.LocationId = operation.ToLocationID
-					locAmount.ProductId = v.ID
-					locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
-					locAmount.Amount = locAmount.Amount.Add(value)
-					if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
-						return res.Error
+					if err == nil {
+						operationTransfer := &models.Operation{
+							Number:            operation.Number,
+							SourceNumber:      operation.SourceNumber,
+							OperationTypeId:   0,
+							OperationTypeName: operation.OperationTypeName,
+							Status:            constvar.OperationStatus_Finish,
+							FromLocationID:    locationRule.AreaId,
+							ToLocationID:      locationRule.LocationId,
+							OperationDate:     operation.OperationDate,
+							ContacterID:       operation.ContacterID,
+							ContacterName:     operation.ContacterName,
+							CompanyID:         operation.CompanyID,
+							CompanyName:       operation.CompanyName,
+							Comment:           operation.Comment,
+							BaseOperationType: constvar.BaseOperationTypeInternal,
+							Details: []*models.OperationDetails{
+								{
+									ProductId: v.ID,
+									Amount:    value,
+								},
+							},
+						}
+						operationInputs = append(operationInputs, operationTransfer)
+
+						//if err := tx.Create(&operationTransfer).Error; err != nil {
+						//	return err
+						//}
+
+						//TODO:鍑哄叆搴撶殑finish鍜屾姤搴熺殑finish閮借澧炲姞瀵筶ocation_product_amount琛ㄦ暟閲忕殑鏇存柊,鍥犱负姝よ〃鏈塒roductCategory瀛楁锛屾墍浠peration_details琛ㄤ腑瑕佸鍔燩roductCategoryId瀛楁
+						locAmount, err := models.NewLocationProductAmountSearch().
+							SetProductId(v.ID).
+							SetLocationId(locationRule.LocationId).
+							First()
+						if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+							return err
+						}
+						locAmount.LocationId = locationRule.LocationId
+						locAmount.ProductId = v.ID
+						locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
+						locAmount.Amount = locAmount.Amount.Add(value)
+						locAmount.ProductCategoryID = v.CategoryId
+						if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
+							return res.Error
+						}
+					} else {
+						//TODO:鍑哄叆搴撶殑finish鍜屾姤搴熺殑finish閮借澧炲姞瀵筶ocation_product_amount琛ㄦ暟閲忕殑鏇存柊,鍥犱负姝よ〃鏈塒roductCategory瀛楁锛屾墍浠peration_details琛ㄤ腑瑕佸鍔燩roductCategoryId瀛楁
+						locAmount, err := models.NewLocationProductAmountSearch().
+							SetProductId(v.ID).
+							SetLocationId(operation.ToLocationID).
+							First()
+						if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+							return err
+						}
+						locAmount.LocationId = operation.ToLocationID
+						locAmount.ProductId = v.ID
+						locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
+						locAmount.Amount = locAmount.Amount.Add(value)
+						locAmount.ProductCategoryID = v.CategoryId
+						if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
+							return res.Error
+						}
 					}
 				}
 			}
@@ -438,6 +491,9 @@
 				if err := tx.Create(&operationInputs).Error; err != nil {
 					return err
 				}
+			}
+			if operation.SourceNumber != "" {
+				go UpdatePurchaseStatus(operation.SourceNumber)
 			}
 
 		}
@@ -448,8 +504,20 @@
 				if !ok {
 					return errors.New("浜у搧绉嶇被寮傚父")
 				}
-				if v.Amount.LessThan(value) {
-					return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Name, v.Amount.String(), value.String()))
+				//todo 婕旂ず娴嬭瘯鏁版嵁
+				data, err := os.ReadFile("conf/input.json")
+				if err != nil {
+					return errors.New("鏂囦欢璇诲彇澶辫触")
+				}
+				m := make(map[string]interface{})
+				err = json.Unmarshal(data, &m)
+				if err != nil {
+					return errors.New("鏍煎紡杞崲澶辫触")
+				}
+				if opa.OpaCheck(c, m, "operation") {
+					if v.Amount.LessThan(value) {
+						return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Name, v.Amount.String(), value.String()))
+					}
 				}
 				listProdt[k].Amount = listProdt[k].Amount.Sub(value)
 				if err := tx.Save(listProdt[k]).Error; err != nil {
@@ -470,8 +538,13 @@
 					return err
 				}
 			}
+			if operation.SourceNumber != "" {
+				go UpdateSalesDetailStatus(operation.SourceNumber)
+			}
 		}
+
 		if operation.BaseOperationType == constvar.BaseOperationTypeInternal {
+			var operationInputs []*models.Operation
 			for _, v := range listProdt {
 				value, ok := mapProdt[v.ID]
 				if !ok {
@@ -496,15 +569,124 @@
 					return err
 				}
 
-				toLocAmount, res := models.NewLocationProductAmountSearch().
-					SetProductId(v.ID).
-					SetLocationId(operation.ToLocationID).
-					FirstRes()
-				if res.Error != nil {
+				locationRule, err := models.NewLocationProductSearch().SetProductId(v.ID).SetAreaId(operation.ToLocationID).First()
+				if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
 					return err
 				}
-				toLocAmount.Amount = toLocAmount.Amount.Add(value)
-				if err := models.NewLocationProductAmountSearch().SetID(toLocAmount.Id).Update(toLocAmount); err != nil {
+
+				if err == nil {
+					operationTransfer := &models.Operation{
+						Number:            operation.Number,
+						SourceNumber:      operation.SourceNumber,
+						OperationTypeId:   0,
+						OperationTypeName: operation.OperationTypeName,
+						Status:            constvar.OperationStatus_Finish,
+						FromLocationID:    locationRule.AreaId,
+						ToLocationID:      locationRule.LocationId,
+						OperationDate:     operation.OperationDate,
+						ContacterID:       operation.ContacterID,
+						ContacterName:     operation.ContacterName,
+						CompanyID:         operation.CompanyID,
+						CompanyName:       operation.CompanyName,
+						Comment:           operation.Comment,
+						BaseOperationType: constvar.BaseOperationTypeInternal,
+						Details: []*models.OperationDetails{
+							{
+								ProductId: v.ID,
+								Amount:    value,
+							},
+						},
+					}
+					operationInputs = append(operationInputs, operationTransfer)
+
+					//if err := tx.Create(&operationTransfer).Error; err != nil {
+					//	return err
+					//}
+
+					locAmount, err := models.NewLocationProductAmountSearch().
+						SetProductId(v.ID).
+						SetLocationId(locationRule.LocationId).
+						First()
+					if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+						return err
+					}
+					locAmount.LocationId = locationRule.LocationId
+					locAmount.ProductId = v.ID
+					locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
+					locAmount.Amount = locAmount.Amount.Add(value)
+					locAmount.ProductCategoryID = v.CategoryId
+					if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
+						return res.Error
+					}
+				} else {
+					locationRule, err = models.NewLocationProductSearch().SetProductCategoryId(v.CategoryId).SetAreaId(operation.ToLocationID).First()
+					if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+						return err
+					}
+					if err == nil {
+						operationTransfer := &models.Operation{
+							Number:            operation.Number,
+							SourceNumber:      operation.SourceNumber,
+							OperationTypeId:   0,
+							OperationTypeName: operation.OperationTypeName,
+							Status:            constvar.OperationStatus_Finish,
+							FromLocationID:    locationRule.AreaId,
+							ToLocationID:      locationRule.LocationId,
+							OperationDate:     operation.OperationDate,
+							ContacterID:       operation.ContacterID,
+							ContacterName:     operation.ContacterName,
+							CompanyID:         operation.CompanyID,
+							CompanyName:       operation.CompanyName,
+							Comment:           operation.Comment,
+							BaseOperationType: constvar.BaseOperationTypeInternal,
+							Details: []*models.OperationDetails{
+								{
+									ProductId: v.ID,
+									Amount:    value,
+								},
+							},
+						}
+						operationInputs = append(operationInputs, operationTransfer)
+
+						//if err := tx.Create(&operationTransfer).Error; err != nil {
+						//	return err
+						//}
+
+						locAmount, err := models.NewLocationProductAmountSearch().
+							SetProductId(v.ID).
+							SetLocationId(locationRule.LocationId).
+							First()
+						if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+							return err
+						}
+						locAmount.LocationId = locationRule.LocationId
+						locAmount.ProductId = v.ID
+						locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
+						locAmount.Amount = locAmount.Amount.Add(value)
+						locAmount.ProductCategoryID = v.CategoryId
+						if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
+							return res.Error
+						}
+					} else {
+						locAmount, err := models.NewLocationProductAmountSearch().
+							SetProductId(v.ID).
+							SetLocationId(operation.ToLocationID).
+							First()
+						if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+							return err
+						}
+						locAmount.LocationId = operation.ToLocationID
+						locAmount.ProductId = v.ID
+						locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
+						locAmount.Amount = locAmount.Amount.Add(value)
+						if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
+							return res.Error
+						}
+					}
+				}
+			}
+			if len(operationInputs) > 0 {
+				if err := tx.Create(&operationInputs).Error; err != nil {
 					return err
 				}
 			}
@@ -514,10 +696,58 @@
 		util.ResponseFormat(c, code.RequestError, err.Error())
 		return
 	}
+
 	util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛")
 }
 
-// listTransfer
+var (
+	ProductInventoryServiceConn *grpc.ClientConn
+	PurchaseServiceConn         *grpc.ClientConn
+)
+
+func InitProductInventoryServiceConn() {
+	var err error
+	ProductInventoryServiceConn, err = grpc.Dial(conf.GrpcServerConf.CrmAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		logx.Errorf("grpc dial product service error: %v", err.Error())
+		return
+	}
+	PurchaseServiceConn, err = grpc.Dial(conf.GrpcServerConf.SrmAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		logx.Errorf("grpc dial product service error: %v", err.Error())
+		return
+	}
+}
+
+func CloseProductInventoryServiceConn() {
+	if ProductInventoryServiceConn != nil {
+		ProductInventoryServiceConn.Close()
+	}
+	if PurchaseServiceConn != nil {
+		PurchaseServiceConn.Close()
+	}
+}
+
+func UpdateSalesDetailStatus(number string) {
+	client := product_inventory.NewProductInventoryServiceClient(ProductInventoryServiceConn)
+	_, err := client.UpdateSalesDetailStatus(context.Background(), &product_inventory.UpdateSalesDetailStatusRequest{
+		Number:            number,
+		SalesDetailStatus: "宸插嚭搴�",
+	})
+	if err != nil {
+		logx.Errorf("grpc dial UpdateSalesDetailStatus service error: %v", err)
+	}
+}
+
+func UpdatePurchaseStatus(number string) {
+	client := purchase_wms.NewPurchaseServiceClient(PurchaseServiceConn)
+	_, err := client.UpdatePurchaseStatus(context.Background(), &purchase_wms.UpdatePurchaseStatusRequest{Number: number})
+	if err != nil {
+		logx.Errorf("grpc dial UpdatePurchaseStatus service error: %v", err)
+	}
+}
+
+// ListTransfer
 // @Tags      鍏ュ簱/鍑哄簱
 // @Summary   搴撳瓨璋冩嫧鍒楄〃
 // @Produce   application/json
@@ -547,3 +777,53 @@
 	}
 	util.ResponseFormatListWithPage(c, code.Success, list, int(total), params.Page, params.PageSize)
 }
+
+// GetLogisticCompanyList
+// @Tags      鍏ュ簱/鍑哄簱
+// @Summary   鑾峰彇鐗╂祦鍏徃鍒楄〃
+// @Produce   application/json
+// @Success   200 {object} util.ResponseList{data=[]models.LogisticCompany} "鎴愬姛"
+// @Router    /api-wms/v1/operation/getLogisticCompanyList [get]
+func (slf OperationController) GetLogisticCompanyList(c *gin.Context) {
+	companies, err := models.NewLogisticCompanySearch().FindNotTotal()
+	if err != nil {
+		util.ResponseFormat(c, code.RequestError, "鏌ユ壘澶辫触:"+err.Error())
+		return
+	}
+	util.ResponseFormat(c, code.Success, companies)
+}
+
+// Cancel
+//
+//	@Tags		鍏ュ簱/鍑哄簱
+//	@Summary	鍙栨秷
+//	@Produce	application/json
+//	@Param		id	path		int			true	"id"
+//	@Success	200	{object}	util.Response	"鎴愬姛"
+//	@Router		/api-wms/v1/operation/cancel/{id} [put]
+func (slf OperationController) Cancel(c *gin.Context) {
+	id, err := strconv.Atoi(c.Param("id"))
+	if err != nil {
+		util.ResponseFormat(c, code.RequestParamError, "閿欒鐨刬d鍊�")
+		return
+	}
+	if id == 0 {
+		util.ResponseFormat(c, code.RequestParamError, "id涓�0")
+		return
+	}
+	operation, err := models.NewOperationSearch().SetPreload(true).SetID(id).First()
+	if err != nil {
+		util.ResponseFormat(c, code.RequestParamError, "鏈壘鍒扮浉鍏冲嚭鍏ュ簱淇℃伅:"+err.Error())
+		return
+	}
+	if operation.Status != constvar.OperationStatus_Ready {
+		util.ResponseFormat(c, code.RequestError, "璇ュ嚭鍏ュ簱淇℃伅鏃犳硶鍙栨秷")
+		return
+	}
+	operation.Status = constvar.OperationStatus_Cancel
+	if err := models.NewOperationSearch().SetID(operation.Id).Save(operation); err != nil {
+		util.ResponseFormat(c, code.SaveFail, err.Error())
+		return
+	}
+	util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛")
+}

--
Gitblit v1.8.0