From ed81fc8d860adb66676d4990e4ce0ff56cff79a6 Mon Sep 17 00:00:00 2001 From: jiangshuai <291802688@qq.com> Date: 星期三, 22 十一月 2023 19:48:16 +0800 Subject: [PATCH] Merge branch 'master' of http://192.168.5.5:10010/r/aps/WMS --- controllers/operation.go | 324 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 321 insertions(+), 3 deletions(-) diff --git a/controllers/operation.go b/controllers/operation.go index 5bd16ef..4e5ab53 100644 --- a/controllers/operation.go +++ b/controllers/operation.go @@ -1,16 +1,21 @@ 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" "time" + "wms/conf" "wms/constvar" "wms/extend/code" "wms/extend/util" @@ -18,7 +23,11 @@ "wms/opa" "wms/pkg/logx" "wms/pkg/structx" + "wms/proto/product_inventory" + "wms/proto/purchase_wms" "wms/request" + "wms/utils/http" + "wms/utils/upload" ) type OperationController struct { @@ -257,7 +266,7 @@ util.ResponseFormat(c, code.Success, "淇敼鎴愬姛") } -// DeleteDevice +// Delete // // @Tags 鍏ュ簱/鍑哄簱 // @Summary 鍒犻櫎鍏ュ簱/鍑哄簱淇℃伅 @@ -299,7 +308,7 @@ util.ResponseFormat(c, code.Success, "鍒犻櫎鎴愬姛") } -// DeleteDevice +// Finish // // @Tags 鍏ュ簱/鍑哄簱 // @Summary 鏇存敼璁板綍鐘舵�� @@ -407,6 +416,7 @@ 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 } @@ -456,6 +466,7 @@ 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 } @@ -472,6 +483,7 @@ 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 } @@ -482,6 +494,9 @@ if err := tx.Create(&operationInputs).Error; err != nil { return err } + } + if operation.SourceNumber != "" { + go UpdatePurchaseStatus(operation.SourceNumber) } } @@ -525,6 +540,9 @@ if err := models.NewLocationProductAmountSearch().SetID(locAmount.Id).Update(locAmount); err != nil { return err } + } + if operation.SourceNumber != "" { + go UpdateSalesDetailStatus(operation.SourceNumber) } } @@ -599,6 +617,7 @@ 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 } @@ -647,6 +666,7 @@ 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 } @@ -679,10 +699,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 @@ -712,3 +780,253 @@ } 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, "鎿嶄綔鎴愬姛") +} + +// 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) { + 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", operation.FromLocation.Name) + f.SetCellValue("Sheet1", "H2", operation.WaybillNumber) + f.SetCellValue("Sheet1", "H3", operation.SourceNumber) + f.SetCellValue("Sheet1", "K3", operation.Number) + f.SetCellValue("Sheet1", "C4", operation.CompanyName) + + 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 + +} -- Gitblit v1.8.0