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 | 217 +++++++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 196 insertions(+), 21 deletions(-) diff --git a/controllers/operation.go b/controllers/operation.go index b673452..c0b34f4 100644 --- a/controllers/operation.go +++ b/controllers/operation.go @@ -14,6 +14,7 @@ "os" "sort" "strconv" + "strings" "time" "wms/conf" "wms/constvar" @@ -23,9 +24,12 @@ "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" ) @@ -337,7 +341,10 @@ } 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 err := AddMoveHistory([]*models.Operation{operation}, tx); err != nil { return err } @@ -404,7 +411,6 @@ // return err //} - //TODO:鍑哄叆搴撶殑finish鍜屾姤搴熺殑finish閮借澧炲姞瀵筶ocation_product_amount琛ㄦ暟閲忕殑鏇存柊,鍥犱负姝よ〃鏈塒roductCategory瀛楁锛屾墍浠peration_details琛ㄤ腑瑕佸鍔燩roductCategoryId瀛楁 locAmount, err := models.NewLocationProductAmountSearch(). SetProductId(v.ID). SetLocationId(locationRule.LocationId). @@ -454,7 +460,6 @@ // return err //} - //TODO:鍑哄叆搴撶殑finish鍜屾姤搴熺殑finish閮借澧炲姞瀵筶ocation_product_amount琛ㄦ暟閲忕殑鏇存柊,鍥犱负姝よ〃鏈塒roductCategory瀛楁锛屾墍浠peration_details琛ㄤ腑瑕佸鍔燩roductCategoryId瀛楁 locAmount, err := models.NewLocationProductAmountSearch(). SetProductId(v.ID). SetLocationId(locationRule.LocationId). @@ -471,7 +476,6 @@ return res.Error } } else { - //TODO:鍑哄叆搴撶殑finish鍜屾姤搴熺殑finish閮借澧炲姞瀵筶ocation_product_amount琛ㄦ暟閲忕殑鏇存柊,鍥犱负姝よ〃鏈塒roductCategory瀛楁锛屾墍浠peration_details琛ㄤ腑瑕佸鍔燩roductCategoryId瀛楁 locAmount, err := models.NewLocationProductAmountSearch(). SetProductId(v.ID). SetLocationId(operation.ToLocationID). @@ -703,6 +707,38 @@ util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛") } +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 @@ -824,6 +860,7 @@ 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 @@ -831,15 +868,15 @@ util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛") } -// inputSelfmade +// outputOperation // // @Tags 鍏ュ簱/鍑哄簱 -// @Summary 鎵撳嵃-鑷埗-鍏ュ簱 +// @Summary 鎵撳嵃 // @Produce application/json // @Param id path int true "id" // @Success 200 {object} util.Response "鎴愬姛" -// @Router /api-wms/v1/operation/inputSelfmade/{id} [put] -func (slf OperationController) InputSelfmade(c *gin.Context) { +// @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鍊�") @@ -854,17 +891,45 @@ util.ResponseFormat(c, code.RequestParamError, "鏈壘鍒扮浉鍏冲嚭鍏ュ簱淇℃伅:"+err.Error()) return } - - readerCloser, err := http.HttpGetWithReadCloser(conf.FileTemplateConf.InputSelfmadeAddr) - if err != nil { - util.ResponseFormat(c, code.RequestParamError, "鑾峰彇妯$増澶辫触:"+err.Error()) + 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 { - util.ResponseFormat(c, code.RequestParamError, "璇诲彇excel妯$増澶辫触锛�"+err.Error()) - return + return "", errors.New("璇诲彇excel妯$増澶辫触锛�" + err.Error()) } readerCloser.Close() @@ -905,8 +970,7 @@ totalAmount = totalAmount.Add(v.Amount) } if err := f.MergeCell("Sheet1", "A"+strconv.Itoa(rowIndex), "C"+strconv.Itoa(rowIndex)); err != nil { - util.ResponseFormat(c, code.RequestParamError, err.Error()) - return + return "", err } f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), "鍚堣") f.SetCellValue("Sheet1", "E"+strconv.Itoa(rowIndex), totalAmount.String()) @@ -920,16 +984,127 @@ buf, err := f.WriteToBuffer() if err != nil { - util.ResponseFormat(c, code.RequestParamError, err.Error()) - return + return "", err } fileUrl, err := upload.UploadFileToSeaWeed(string(constvar.FileType_File), "xlsx", buf.Bytes()) if err != nil { logx.Errorf("file upload err: %v", err) - util.ResponseFormat(c, code.RequestParamError, err.Error()) - return + return "", err } - util.ResponseFormat(c, code.Success, fileUrl) + 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