From 7bed630958e25d632903e19a57c1735b9dc9d8d8 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 17 四月 2024 13:38:34 +0800
Subject: [PATCH] 增加钉钉机器人报警功能

---
 proto/product_inventory/server.go |   41 +++++++++++++++++++++++++++++++++++++++++
 1 files changed, 41 insertions(+), 0 deletions(-)

diff --git a/proto/product_inventory/server.go b/proto/product_inventory/server.go
index 26bd44b..4bf5368 100644
--- a/proto/product_inventory/server.go
+++ b/proto/product_inventory/server.go
@@ -5,6 +5,7 @@
 	"errors"
 	"fmt"
 	"github.com/shopspring/decimal"
+	"gorm.io/gorm"
 	"strconv"
 	"strings"
 	"time"
@@ -350,6 +351,9 @@
 	inputLocationAmountMap := make(map[int64]map[string]decimal.Decimal)
 	outputLocationAmountMap := make(map[int64]map[string]decimal.Decimal)
 	for _, v := range outputList {
+		if req.WarehouseId != 0 && v.WareHouseID != req.WarehouseId {
+			continue
+		}
 		if outputProductMap[v.Number] == nil {
 			simpleInfo := &OutputSimpleInfo{
 				Number: v.Number,
@@ -363,6 +367,9 @@
 		}
 	}
 	for _, v := range inputList {
+		if req.WarehouseId != 0 && v.WareHouseID != req.WarehouseId {
+			continue
+		}
 		if inputProductMap[v.Number] == nil {
 			storeInfo := &StoreInfo{
 				Number: v.Number,
@@ -401,6 +408,9 @@
 
 	LocationIDWarehouseIDMap := make(map[int64]int64)
 	for _, output := range outputList {
+		if req.WarehouseId != 0 && output.WareHouseID != req.WarehouseId {
+			continue
+		}
 		if outputLocationAmountMap[output.LocationID] == nil {
 			outputLocationAmountMap[output.LocationID] = make(map[string]decimal.Decimal)
 		}
@@ -408,6 +418,9 @@
 		outputLocationAmountMap[output.LocationID][output.Number] = outputAmount
 	}
 	for _, input := range inputList {
+		if req.WarehouseId != 0 && input.WareHouseID != req.WarehouseId {
+			continue
+		}
 		LocationIDWarehouseIDMap[input.LocationID] = input.WareHouseID
 
 		if inputLocationAmountMap[input.LocationID] == nil {
@@ -466,3 +479,31 @@
 	resp.Msg = "success"
 	return resp, nil
 }
+
+func (s *Server) GetOutputOperationInfo(ctx context.Context, req *GetOutputOperationInfoRequest) (*GetOutputOperationInfoResponse, error) {
+	if req.Number == "" {
+		return nil, errors.New("鍙傛暟涓嶈兘涓虹┖")
+	}
+	result := new(GetOutputOperationInfoResponse)
+	first, err := models.NewOperationSearch().SetSourceNumber(req.Number).SetBaseOperationType(constvar.BaseOperationTypeOutgoing).SetStatus(constvar.OperationStatus_Finish).First()
+	if err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return result, nil
+		}
+		return nil, err
+	}
+	details, err := models.NewOperationDetailsSearch().SetOperationId(first.Id).FindNotTotal()
+	if err != nil {
+		return nil, err
+	}
+	list := make([]*OutputProduct, 0)
+	for _, detail := range details {
+		var op OutputProduct
+		op.Number = detail.ProductId
+		op.Amount = detail.Amount.String()
+		list = append(list, &op)
+	}
+	result.Products = list
+
+	return result, nil
+}

--
Gitblit v1.8.0