From 313617ceb05dd5194d2091c717a25cbf33ef39b9 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 24 十一月 2023 14:39:25 +0800
Subject: [PATCH] 新增history加入索引

---
 controllers/operation.go |  840 +++++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 782 insertions(+), 58 deletions(-)

diff --git a/controllers/operation.go b/controllers/operation.go
index 872fb4c..c0b34f4 100644
--- a/controllers/operation.go
+++ b/controllers/operation.go
@@ -1,20 +1,37 @@
 package controllers
 
 import (
+	"context"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"github.com/gin-gonic/gin"
 	"github.com/shopspring/decimal"
+	"github.com/xuri/excelize/v2"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
 	"gorm.io/gorm"
+	"os"
+	"sort"
 	"strconv"
+	"strings"
 	"time"
+	"wms/conf"
 	"wms/constvar"
 	"wms/extend/code"
 	"wms/extend/util"
 	"wms/models"
+	"wms/opa"
 	"wms/pkg/logx"
 	"wms/pkg/structx"
+	"wms/proto/client"
+	"wms/proto/product_inventory"
+	"wms/proto/purchase_wms"
+	"wms/proto/supplier"
 	"wms/request"
+	"wms/service"
+	"wms/utils/http"
+	"wms/utils/upload"
 )
 
 type OperationController struct {
@@ -47,14 +64,43 @@
 		return
 	}
 
+	if CheckDetailsRepeat(params.Details) {
+		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
+	}
+
 	params.Status = constvar.OperationStatus_Ready
 	params.Number = strconv.FormatInt(time.Now().Unix(), 10)
+	params.BaseOperationType = operationType.BaseOperationType
 	if err := models.NewOperationSearch().Create(&params); err != nil {
 		logx.Errorf("Operation create err: %v", err)
 		util.ResponseFormat(c, code.SaveFail, "娣诲姞澶辫触锛�"+err.Error())
 		return
 	}
 	util.ResponseFormat(c, code.Success, "娣诲姞鎴愬姛")
+}
+
+func CheckDetailsRepeat(details []*models.OperationDetails) bool {
+	var detailIDs []string
+	var tempID string
+	for _, v := range details {
+		detailIDs = append(detailIDs, v.ProductId)
+	}
+	sort.Strings(detailIDs)
+	for _, v := range detailIDs {
+		if v != tempID {
+			tempID = v
+		} else {
+			return true
+		}
+	}
+	return false
 }
 
 func (slf OperationController) FormatLocation(params *models.Operation) error {
@@ -98,8 +144,8 @@
 		return errors.New("璇峰~鍏ユ簮鍗曞彿")
 	}
 
-	if params.OperationTypeId == 0 {
-		return errors.New("operationTypeId涓�0")
+	if params.OperationTypeId == 0 && int(params.BaseOperationType) == 0 {
+		return errors.New("鏈瘑鍒璁板綍绫诲瀷")
 	}
 
 	if params.OperationDate == "" {
@@ -146,10 +192,10 @@
 	search := models.NewOperationSearch()
 	search.SetPage(params.Page, params.PageSize)
 	if params.Number != "" {
-		search.SetNumber(params.Number)
+		search.SetKeyword(params.Number)
 	}
-	if params.SourceNumber != "" {
-		search.SetSourceNumber(params.SourceNumber)
+	if int(params.Status) != 0 {
+		search.SetStatus(params.Status)
 	}
 	list, total, err := search.SetOperationTypeId(params.OperationTypeId).SetPreload(true).SetOrder("created_at desc").Find()
 	if err != nil {
@@ -196,6 +242,10 @@
 		util.ResponseFormat(c, code.RequestParamError, err.Error())
 		return
 	}
+	if CheckDetailsRepeat(params.Details) {
+		util.ResponseFormat(c, code.RequestParamError, "鏄庣粏涓笉鑳藉瓨鍦ㄩ噸澶嶇殑浜у搧")
+		return
+	}
 	if err := slf.FormatLocation(&params); err != nil {
 		util.ResponseFormat(c, code.RequestParamError, err.Error())
 		return
@@ -220,7 +270,7 @@
 	util.ResponseFormat(c, code.Success, "淇敼鎴愬姛")
 }
 
-// DeleteDevice
+// Delete
 //
 //	@Tags		鍏ュ簱/鍑哄簱
 //	@Summary	鍒犻櫎鍏ュ簱/鍑哄簱淇℃伅
@@ -262,7 +312,7 @@
 	util.ResponseFormat(c, code.Success, "鍒犻櫎鎴愬姛")
 }
 
-// DeleteDevice
+// Finish
 //
 //	@Tags		鍏ュ簱/鍑哄簱
 //	@Summary	鏇存敼璁板綍鐘舵��
@@ -280,7 +330,7 @@
 		util.ResponseFormat(c, code.RequestParamError, "id涓�0")
 		return
 	}
-	operation, err := models.NewOperationSearch().SetID(id).First()
+	operation, err := models.NewOperationSearch().SetPreload(true).SetID(id).First()
 	if err != nil {
 		util.ResponseFormat(c, code.RequestParamError, "鏈壘鍒扮浉鍏冲嚭鍏ュ簱淇℃伅:"+err.Error())
 		return
@@ -289,76 +339,461 @@
 		util.ResponseFormat(c, code.RequestError, "璇ュ嚭鍏ュ簱淇℃伅鏃犳硶瀹屾垚")
 		return
 	}
-	operationType, err := models.NewOperationTypeSearch().SetID(uint(operation.OperationTypeId)).First()
-	if err != nil {
-		util.ResponseFormat(c, code.RequestError, err.Error())
-		return
-	}
 	if err := models.WithTransaction(func(tx *gorm.DB) error {
 
-		if err := models.NewOperationSearch().SetOrm(tx).SetID(id).Update(&models.Operation{Status: constvar.OperationStatus_Finish}); err != nil {
+		if err := models.NewOperationSearch().SetOrm(tx).SetID(id).Update(&models.Operation{Status: constvar.OperationStatus_Finish, AuditDate: time.Now().Format("2006-01-02 15:04:05")}); err != nil {
 			return err
 		}
-		if operationType.BaseOperationType != constvar.BaseOperationTypeInternal {
-			var listProdtId []string
-			var listProdt []*models.Material
-			mapProdt := make(map[string]decimal.Decimal)
-			listDetails, err := models.NewOperationDetailsSearch().SetOperationId(operation.Id).FindAll()
-			if err != nil {
-				return err
-			}
-			for _, v := range listDetails {
-				listProdtId = append(listProdtId, v.ProductId)
-				mapProdt[v.ProductId] = v.Amount
-			}
-			if err := models.NewMaterialSearch().Orm.Where("id IN ?", listProdtId).Find(&listProdt).Error; err != nil {
-				return err
-			}
-			if operationType.BaseOperationType == constvar.BaseOperationTypeIncoming {
-				for k, v := range listProdt {
-					if value, ok := mapProdt[v.ID]; !ok {
-						return errors.New("浜у搧绉嶇被寮傚父")
-					} else {
-						listProdt[k].Amount = listProdt[k].Amount.Add(value)
-						if err := tx.Save(listProdt[k]).Error; err != nil {
+		if err := AddMoveHistory([]*models.Operation{operation}, tx); err != nil {
+			return err
+		}
+
+		var listProdtId []string
+		var listProdt []*models.Material
+		mapProdt := make(map[string]decimal.Decimal)
+		listDetails, err := models.NewOperationDetailsSearch().SetOperationId(operation.Id).FindAll()
+		if err != nil {
+			return err
+		}
+		for _, v := range listDetails {
+			listProdtId = append(listProdtId, v.ProductId)
+			mapProdt[v.ProductId] = v.Amount
+		}
+
+		if err := models.NewMaterialSearch().Orm.Where("id IN ?", listProdtId).Find(&listProdt).Error; err != nil {
+			return err
+		}
+
+		if operation.BaseOperationType == constvar.BaseOperationTypeIncoming {
+			var operationInputs []*models.Operation
+			for k, v := range listProdt {
+				value, ok := mapProdt[v.ID]
+				if !ok {
+					return errors.New("浜у搧绉嶇被寮傚父")
+				}
+
+				listProdt[k].Amount = listProdt[k].Amount.Add(value)
+				if err := tx.Save(listProdt[k]).Error; err != nil {
+					return err
+				}
+
+				locationRule, err := models.NewLocationProductSearch().SetProductId(v.ID).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 {
+					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)
+						locAmount.ProductCategoryID = v.CategoryId
+						if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
+							return res.Error
 						}
 					}
 				}
 			}
-			if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing {
-				for k, v := range listProdt {
-					if value, ok := mapProdt[v.ID]; !ok {
-						return errors.New("浜у搧绉嶇被寮傚父")
-					} else {
-						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 {
-							return err
-						}
-					}
+			if len(operationInputs) > 0 {
+				if err := tx.Create(&operationInputs).Error; err != nil {
+					return err
 				}
+			}
+			if operation.SourceNumber != "" {
+				go UpdatePurchaseStatus(operation.SourceNumber)
 			}
 
+		}
+
+		if operation.BaseOperationType == constvar.BaseOperationTypeOutgoing {
+			for k, v := range listProdt {
+				value, ok := mapProdt[v.ID]
+				if !ok {
+					return errors.New("浜у搧绉嶇被寮傚父")
+				}
+				//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 {
+					return err
+				}
+				locAmount, res := models.NewLocationProductAmountSearch().
+					SetProductId(v.ID).
+					SetLocationId(operation.ToLocationID).
+					FirstRes()
+				if res.Error != nil {
+					return err
+				}
+				if locAmount.Amount.LessThan(value) {
+					return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Name, v.Amount.String(), value.String()))
+				}
+				locAmount.Amount = locAmount.Amount.Sub(value)
+				if err := models.NewLocationProductAmountSearch().SetID(locAmount.Id).Update(locAmount); err != nil {
+					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 {
+					return errors.New("浜у搧绉嶇被寮傚父")
+				}
+				//listProdt[k].Amount = listProdt[k].Amount.Add(value)
+				//if err := tx.Save(listProdt[k]).Error; err != nil {
+				//	return err
+				//}
+				fromLocAmount, res := models.NewLocationProductAmountSearch().
+					SetProductId(v.ID).
+					SetLocationId(operation.FromLocationID).
+					FirstRes()
+				if res.Error != nil {
+					return err
+				}
+				if fromLocAmount.Amount.LessThan(value) {
+					return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,璋冩嫧锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愯皟鎷ㄦ搷浣�", v.Name, v.Amount.String(), value.String()))
+				}
+				fromLocAmount.Amount = fromLocAmount.Amount.Sub(value)
+				if err := models.NewLocationProductAmountSearch().SetID(fromLocAmount.Id).Update(fromLocAmount); err != nil {
+					return err
+				}
+
+				locationRule, err := models.NewLocationProductSearch().SetProductId(v.ID).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 {
+					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
+				}
+			}
 		}
 		return nil
 	}); err != nil {
 		util.ResponseFormat(c, code.RequestError, err.Error())
 		return
 	}
+
 	util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛")
 }
 
-// ListAll
+func AddMoveHistory(operationList []*models.Operation, db *gorm.DB) error {
+	var histories []*models.MoveHistory
+	for _, operation := range operationList {
+		for _, v := range operation.Details {
+			history := &models.MoveHistory{
+				Number:            operation.Number,
+				BaseOperationType: operation.BaseOperationType,
+				OperationTypeId:   operation.OperationTypeId,
+				OperationTypeName: operation.OperationTypeName,
+				OperationId:       operation.Id,
+				ProductId:         v.ProductId,
+				ProductName:       v.Product.Name,
+				Amount:            v.Amount,
+				Unit:              v.Product.Unit,
+				Weight:            operation.Weight,
+				FromLocationId:    operation.FromLocationID,
+				FromLocation:      operation.FromLocation.Name,
+				ToLocationId:      operation.ToLocationID,
+				ToLocation:        operation.ToLocation.Name,
+			}
+			histories = append(histories, history)
+		}
+	}
+	if err := db.Model(&models.MoveHistory{}).Create(&histories).Error; err != nil {
+		return err
+	}
+	for _, history := range histories {
+		service.AddNewHistoryReportRecord(history.Id)
+	}
+	return nil
+}
+
+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   璋冩嫧
+// @Summary   搴撳瓨璋冩嫧鍒楄〃
 // @Produce   application/json
 // @Param     object  body  request.OperationAllList true  "鍙傛暟"
 // @Success   200 {object} util.Response "鎴愬姛"
-// @Router    /api-wms/v1/operation/listAll [post]
-func (slf OperationController) ListAll(c *gin.Context) {
+// @Router    /api-wms/v1/operation/listTransfer [post]
+func (slf OperationController) ListTransfer(c *gin.Context) {
 	var params request.OperationAllList
 	if err := c.BindJSON(&params); err != nil {
 		util.ResponseFormat(c, code.RequestParamError, "鍙傛暟瑙f瀽澶辫触锛屾暟鎹被鍨嬮敊璇�:"+err.Error())
@@ -372,15 +807,304 @@
 	search.SetPage(params.Page, params.PageSize)
 	search.SetPage(params.Page, params.PageSize)
 	if params.Number != "" {
-		search.SetNumber(params.Number)
+		search.SetKeyword(params.Number)
 	}
-	if params.SourceNumber != "" {
-		search.SetSourceNumber(params.SourceNumber)
-	}
-	list, total, err := search.SetPreload(true).SetOrder("created_at desc").Find()
+	list, total, err := search.SetPreload(true).SetBaseOperationType(constvar.BaseOperationTypeInternal).SetOrder("created_at desc").Find()
 	if err != nil {
 		util.ResponseFormat(c, code.RequestError, "鏌ユ壘澶辫触:"+err.Error())
 		return
 	}
 	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
+	operation.AuditDate = time.Now().Format("2006-01-02 15:04:05")
+	if err := models.NewOperationSearch().SetID(operation.Id).Save(operation); err != nil {
+		util.ResponseFormat(c, code.SaveFail, err.Error())
+		return
+	}
+	util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛")
+}
+
+// outputOperation
+//
+//	@Tags		鍏ュ簱/鍑哄簱
+//	@Summary	鎵撳嵃
+//	@Produce	application/json
+//	@Param		id	path		int			true	"id"
+//	@Success	200	{object}	util.Response	"鎴愬姛"
+//	@Router		/api-wms/v1/operation/outputOperation/{id} [put]
+func (slf OperationController) OutputOperation(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 len(operation.Details) == 0 {
+		util.ResponseFormat(c, code.RequestParamError, "娌℃湁鐩稿叧浜у搧鏄庣粏鐨勪俊鎭�")
+		return
+	}
+
+	var fileUrl string
+	if operation.BaseOperationType == constvar.BaseOperationTypeIncoming {
+		fileUrl, err = ExportInputSelfmade(constvar.FileTemplateCategory_Selfmade, operation)
+		if err != nil {
+			util.ResponseFormat(c, code.RequestParamError, err.Error())
+			return
+		}
+	} else if operation.BaseOperationType == constvar.BaseOperationTypeOutgoing {
+		fileUrl, err = ExportOutputOperation(constvar.FileTemplateCategory_Output, operation)
+		if err != nil {
+			util.ResponseFormat(c, code.RequestParamError, err.Error())
+			return
+		}
+	} else {
+		util.ResponseFormat(c, code.RequestParamError, "璇ヨ褰曚笉鏀寔鎵撳嵃")
+		return
+	}
+
+	util.ResponseFormat(c, code.Success, fileUrl)
+}
+
+func ExportInputSelfmade(category constvar.FileTemplateCategory, operation *models.Operation) (string, error) {
+	template, err := models.NewFileTemplateAttachmentSearch().SetPreload(true).SetCategory(category).First()
+	if err != nil {
+		return "", errors.New("鑾峰彇妯$増璁板綍澶辫触:" + err.Error())
+	}
+	readerCloser, err := http.HttpGetWithReadCloser(template.Attachment.FileUrl)
+	if err != nil {
+		return "", errors.New("鑾峰彇妯$増澶辫触:" + err.Error())
+	}
+
+	f, err := excelize.OpenReader(readerCloser)
+	if err != nil {
+		return "", errors.New("璇诲彇excel妯$増澶辫触锛�" + err.Error())
+	}
+	readerCloser.Close()
+
+	style, _ := f.NewStyle(&excelize.Style{
+		Border: []excelize.Border{
+			{Type: "left", Color: "#000000", Style: 2},
+			{Type: "top", Color: "#000000", Style: 2},
+			{Type: "bottom", Color: "#000000", Style: 2},
+			{Type: "right", Color: "#000000", Style: 2},
+		},
+		Alignment: &excelize.Alignment{
+			Horizontal: "center",
+			Vertical:   "center",
+		},
+	})
+	styleLeft, _ := f.NewStyle(&excelize.Style{
+		Alignment: &excelize.Alignment{
+			Horizontal: "left",
+			Vertical:   "center",
+		},
+	})
+
+	f.SetCellValue("Sheet1", "B3", operation.CompanyName)
+	f.SetCellValue("Sheet1", "E3", operation.OperationDate)
+	f.SetCellValue("Sheet1", "H3", operation.Number)
+
+	rowIndex := 5
+	totalAmount := decimal.NewFromInt(0)
+	for _, v := range operation.Details {
+		f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), v.ProductId)
+		f.SetCellValue("Sheet1", "B"+strconv.Itoa(rowIndex), v.Product.Name)
+		f.SetCellValue("Sheet1", "C"+strconv.Itoa(rowIndex), v.Product.Specs)
+		f.SetCellValue("Sheet1", "D"+strconv.Itoa(rowIndex), v.Product.Unit)
+		f.SetCellValue("Sheet1", "E"+strconv.Itoa(rowIndex), v.Amount.String())
+		f.SetCellValue("Sheet1", "H"+strconv.Itoa(rowIndex), operation.ToLocation.Name)
+		f.SetCellValue("Sheet1", "I"+strconv.Itoa(rowIndex), operation.Comment)
+		rowIndex++
+		totalAmount = totalAmount.Add(v.Amount)
+	}
+	if err := f.MergeCell("Sheet1", "A"+strconv.Itoa(rowIndex), "C"+strconv.Itoa(rowIndex)); err != nil {
+		return "", err
+	}
+	f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), "鍚堣")
+	f.SetCellValue("Sheet1", "E"+strconv.Itoa(rowIndex), totalAmount.String())
+	f.SetCellStyle("Sheet1", "A4", "I"+strconv.Itoa(rowIndex), style)
+	rowIndex++
+
+	f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), "瀹℃牳锛�")
+	f.SetCellValue("Sheet1", "D"+strconv.Itoa(rowIndex), "妫�楠岋細")
+	f.SetCellValue("Sheet1", "G"+strconv.Itoa(rowIndex), "淇濈锛�")
+	f.SetCellStyle("Sheet1", "A"+strconv.Itoa(rowIndex), "G"+strconv.Itoa(rowIndex), styleLeft)
+
+	buf, err := f.WriteToBuffer()
+	if err != nil {
+		return "", err
+	}
+
+	fileUrl, err := upload.UploadFileToSeaWeed(string(constvar.FileType_File), "xlsx", buf.Bytes())
+	if err != nil {
+		logx.Errorf("file upload err: %v", err)
+		return "", err
+	}
+
+	return fileUrl, nil
+}
+
+func ExportOutputOperation(category constvar.FileTemplateCategory, operation *models.Operation) (string, error) {
+	repositoryLevels := strings.Split(operation.FromLocation.JointName, "/")
+	template, err := models.NewFileTemplateAttachmentSearch().SetPreload(true).SetCategory(category).First()
+	if err != nil {
+		return "", errors.New("鑾峰彇妯$増璁板綍澶辫触:" + err.Error())
+	}
+	readerCloser, err := http.HttpGetWithReadCloser(template.Attachment.FileUrl)
+	if err != nil {
+		return "", errors.New("鑾峰彇妯$増澶辫触:" + err.Error())
+	}
+
+	f, err := excelize.OpenReader(readerCloser)
+	if err != nil {
+		return "", errors.New("璇诲彇excel妯$増澶辫触锛�" + err.Error())
+	}
+	readerCloser.Close()
+
+	style, _ := f.NewStyle(&excelize.Style{
+		Border: []excelize.Border{
+			{Type: "left", Color: "#000000", Style: 1},
+			{Type: "top", Color: "#000000", Style: 1},
+			{Type: "bottom", Color: "#000000", Style: 1},
+			{Type: "right", Color: "#000000", Style: 1},
+		},
+		Alignment: &excelize.Alignment{
+			Horizontal: "center",
+			Vertical:   "center",
+		},
+	})
+
+	f.SetCellValue("Sheet1", "C2", repositoryLevels[0])
+	f.SetCellValue("Sheet1", "H2", operation.WaybillNumber)
+	f.SetCellValue("Sheet1", "K2", operation.AuditDate)
+	f.SetCellValue("Sheet1", "H3", operation.SourceNumber)
+	f.SetCellValue("Sheet1", "K3", operation.Number)
+	f.SetCellValue("Sheet1", "C4", operation.CompanyName)
+	f.SetCellValue("Sheet1", "I4", operation.ReceiverAddr)
+	f.SetCellValue("Sheet1", "C5", operation.ReceiverName)
+	f.SetCellValue("Sheet1", "H5", operation.ReceiverPhone)
+
+	rowIndex := 8
+	totalAmount := decimal.NewFromInt(0)
+	totalWeight := decimal.NewFromInt(0)
+
+	addRows := len(operation.Details) - 1
+	if addRows > 0 {
+		f.InsertRows("Sheet1", 9, addRows)
+		f.SetCellStyle("Sheet1", "A9", "K"+strconv.Itoa(rowIndex+addRows), style)
+	}
+
+	for k, v := range operation.Details {
+		productTotalWeight := v.Amount.Mul(v.Product.Weight).Div(decimal.NewFromInt(1000))
+		f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), k+1)
+		f.SetCellValue("Sheet1", "B"+strconv.Itoa(rowIndex), v.Product.Name)
+		f.SetCellValue("Sheet1", "H"+strconv.Itoa(rowIndex), v.Amount)
+		f.SetCellValue("Sheet1", "I"+strconv.Itoa(rowIndex), v.Product.Weight)
+		f.SetCellValue("Sheet1", "J"+strconv.Itoa(rowIndex), productTotalWeight)
+		rowIndex++
+		totalAmount = totalAmount.Add(v.Amount)
+		totalWeight = totalWeight.Add(productTotalWeight)
+	}
+
+	buf, err := f.WriteToBuffer()
+	if err != nil {
+		return "", err
+	}
+
+	fileUrl, err := upload.UploadFileToSeaWeed(string(constvar.FileType_File), "xlsx", buf.Bytes())
+	if err != nil {
+		logx.Errorf("file upload err: %v", err)
+		return "", err
+	}
+
+	return fileUrl, nil
+
+}
+
+// GetSupplierList
+// @Tags      鍏ュ簱/鍑哄簱
+// @Summary   鑾峰彇鐗╂祦鍏徃鍒楄〃
+// @Produce   application/json
+//
+//	@Success	200	{object}	util.Response	"鎴愬姛"
+//
+// @Router    /api-wms/v1/operation/getSupplierList [get]
+func (slf OperationController) GetSupplierList(c *gin.Context) {
+	cli := supplier.NewSupplierServiceClient(supplier.SupplierConn)
+	resp, err := cli.GetSupplierList(c, &supplier.SupplierListRequest{Status: 1})
+	if err != nil {
+		util.ResponseFormat(c, code.RequestParamError, "grpc璋冪敤澶辫触:"+err.Error())
+		return
+	}
+	util.ResponseFormat(c, code.Success, resp.List)
+}
+
+// GetClientList
+// @Tags      鍏ュ簱/鍑哄簱
+// @Summary   鑾峰彇鐗╂祦鍏徃鍒楄〃
+// @Produce   application/json
+//
+//	@Success	200	{object}	util.Response	"鎴愬姛"
+//
+// @Router    /api-wms/v1/operation/getClientList [get]
+func (slf OperationController) GetClientList(c *gin.Context) {
+	cli := client.NewClientServiceClient(client.ClientConn)
+	resp, err := cli.GetClientList(c, &client.ClientListRequest{})
+	if err != nil {
+		util.ResponseFormat(c, code.RequestParamError, "grpc璋冪敤澶辫触:"+err.Error())
+		return
+	}
+	util.ResponseFormat(c, code.Success, resp.List)
+}

--
Gitblit v1.8.0