From e5df488268e29b272932e6cc1d2b1e7034590ba0 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期三, 12 六月 2024 11:40:45 +0800 Subject: [PATCH] weedfs包放到项目里 --- controllers/operation.go | 200 +++++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 172 insertions(+), 28 deletions(-) diff --git a/controllers/operation.go b/controllers/operation.go index 216c091..e334da1 100644 --- a/controllers/operation.go +++ b/controllers/operation.go @@ -2,7 +2,6 @@ import ( "context" - "encoding/json" "errors" "fmt" "github.com/gin-gonic/gin" @@ -10,16 +9,15 @@ "github.com/shopspring/decimal" "github.com/xuri/excelize/v2" "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/mysqlx" "wms/pkg/structx" @@ -541,23 +539,23 @@ } - if operation.BaseOperationType == constvar.BaseOperationTypeOutgoing { + if operation.BaseOperationType == constvar.BaseOperationTypeOutgoing || operation.BaseOperationType == constvar.BaseOperationTypeDisuse { for k, v := range listDetails { //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.Product.Amount.LessThan(v.Amount) { - return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Product.Name, v.Product.Amount.String(), v.Amount.String())) - } - } + //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.Product.Amount.LessThan(v.Amount) { + // return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Product.Name, v.Product.Amount.String(), v.Amount.String())) + // } + //} //todo ================end=============================== if v.Product.Amount.LessThan(v.Amount) { return errors.New(fmt.Sprintf("浜у搧锛�%v,搴撳瓨锛�%v,鍑哄簱锛�%v,鏁伴噺涓嶅锛屾棤娉曞畬鎴愬嚭搴撴搷浣�", v.Product.Name, v.Product.Amount.String(), v.Amount.String())) @@ -599,6 +597,27 @@ toLocAmount.Amount = toLocAmount.Amount.Add(v.Amount) if err := models.NewLocationProductAmountSearch().SetID(toLocAmount.Id).Update(toLocAmount); err != nil { return err + } + } else { + if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{ + LocationId: v.ToLocationID, + ProductCategoryID: v.Product.CategoryId, + ProductId: v.ProductId, + Amount: v.Amount, + CreateDate: time.Now().Format("2006-01-02 15:04:05"), + }); err != nil { + return err + } + } + } + } + + if operation.BaseOperationType == constvar.BaseOperationTypeAdjust { + for _, v := range listDetails { + if locAmount, aok := mapLocAmount[strconv.Itoa(v.ToLocationID)+v.ProductId]; aok { + locAmount.Amount = v.Amount + if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil { + return res.Error } } else { if err := models.NewLocationProductAmountSearch().Create(&models.LocationProductAmount{ @@ -870,24 +889,149 @@ } var fileUrl string + companyName := conf.WebConf.CompanyName + if companyName == "" { + util.ResponseFormat(c, code.RequestParamError, "缂哄皯宸ュ巶鍚嶇О閰嶇疆") + return + } if operation.BaseOperationType == constvar.BaseOperationTypeIncoming { - fileUrl, err = ExportInputSelfmade(constvar.FileTemplateCategory_Selfmade, operation) - if err != nil { - util.ResponseFormat(c, code.RequestParamError, err.Error()) - return + if companyName == "jialian" { + fileUrl, err = JialianOperation(constvar.FileTemplateCategory_JialianInput, operation) + if err != nil { + util.ResponseFormat(c, code.RequestParamError, err.Error()) + return + } + } else if companyName == "geruimi" { + 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 + if companyName == "jialian" { + fileUrl, err = JialianOperation(constvar.FileTemplateCategory_JialianOutput, operation) + if err != nil { + util.ResponseFormat(c, code.RequestParamError, err.Error()) + return + } + } else if companyName == "geruimi" { + fileUrl, err = ExportInputSelfmade(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) + m := make(map[string]string) + m["url"] = fileUrl + m["name"] = companyName + util.ResponseFormat(c, code.Success, m) +} + +func JialianOperation(category constvar.FileTemplateCategory, operation *models.Operation) (string, error) { + template, err := models.NewFileTemplateAttachmentSearch().SetCategory(category).First() + if err != nil { + return "", errors.New("鑾峰彇妯$増璁板綍澶辫触:" + err.Error()) + } + readerCloser, err := http.HttpGetWithReadCloser(template.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() + defer f.Close() + + f.SetCellValue("Sheet1", "H2", operation.OperationDate) + f.SetCellValue("Sheet1", "O2", operation.Number) + + rowIndex := 5 + totalAmount := decimal.NewFromInt(0) + totalPrice := decimal.NewFromInt(0) + for i, v := range operation.Details { + //璁剧疆琛ㄥ崟鏈�澶�9鏉℃暟鎹� + if i > 8 { + break + } + f.SetCellValue("Sheet1", "A"+strconv.Itoa(rowIndex), v.Product.Name) + f.SetCellValue("Sheet1", "B"+strconv.Itoa(rowIndex), v.Product.Type) + f.SetCellValue("Sheet1", "C"+strconv.Itoa(rowIndex), v.Product.Unit) + f.SetCellValue("Sheet1", "D"+strconv.Itoa(rowIndex), v.Amount.String()) + f.SetCellValue("Sheet1", "E"+strconv.Itoa(rowIndex), v.Product.SalePrice.String()) + if !v.Product.SalePrice.IsZero() { + ap := v.Amount.Mul(v.Product.SalePrice) + totalPrice = totalPrice.Add(ap) + price := ap.String() + split := strings.Split(price, ".") + if len(split) == 2 { + for n, a := range split[1] { + f.SetCellValue("Sheet1", string('M'+n)+strconv.Itoa(rowIndex), string(a)) + } + } + s := split[0] + n := 0 + for j := len(s) - 1; j >= 0; j-- { + if n == 6 { + f.SetCellValue("Sheet1", string('L'-n)+strconv.Itoa(rowIndex), s[:j]) + break + } + f.SetCellValue("Sheet1", string('L'-n)+strconv.Itoa(rowIndex), string(s[j])) + n++ + } + } + + rowIndex++ + totalAmount = totalAmount.Add(v.Amount) + } + //鍚堣 + if !totalPrice.IsZero() { + price := totalPrice.String() + split := strings.Split(price, ".") + if len(split) == 2 { + for n, a := range split[1] { + f.SetCellValue("Sheet1", string('M'+n)+"14", string(a)) + } + } else { + f.SetCellValue("Sheet1", "M14", "0") + f.SetCellValue("Sheet1", "N14", "0") + } + s := split[0] + n := 0 + for j := len(s) - 1; j >= 0; j-- { + if n == 6 { + f.SetCellValue("Sheet1", string('L'-n)+"14", s[:j]) + break + } + f.SetCellValue("Sheet1", string('L'-n)+"14", string(s[j])) + n++ + } + } + f.SetCellValue("Sheet1", "D14", totalAmount) + + f.SetCellValue("Sheet1", "B15", operation.Manager) + f.SetCellValue("Sheet1", "D15", operation.Accountant) + f.SetCellValue("Sheet1", "F15", operation.Custodian) + + buf, err := f.WriteToBuffer() + if err != nil { + return "", err + } + + fileUrl, err := upload.UploadFileToSeaWeed(string(constvar.FileType_File), uuid.NewV4().String()+".xlsx", buf.Bytes()) + if err != nil { + logx.Errorf("file upload err: %v", err) + return "", err + } + + return fileUrl, nil } func ExportInputSelfmade(category constvar.FileTemplateCategory, operation *models.Operation) (string, error) { @@ -1106,7 +1250,7 @@ Select("wms_operation.id as operation_id,wms_operation.number,wms_operation.base_operation_type,material.id AS product_id," + "material.`name` AS product_name,wms_operation_details.amount,material.unit,wms_operation_details.from_location_id," + "from_location.`name` AS from_location,wms_operation_details.to_location_id,to_location.`name` AS to_location," + - "wms_operation.operation_date as date,wms_operation.`status`,material.weight"). + "wms_operation.operation_date as date,wms_operation.`status`,material.weight,wms_operation.operation_type_name"). InnerJoins("inner join wms_operation_details ON wms_operation_details.operation_id = wms_operation.id"). InnerJoins("INNER JOIN material ON material.id = wms_operation_details.product_id"). InnerJoins("INNER JOIN wms_location AS from_location ON from_location.id = wms_operation_details.from_location_id"). -- Gitblit v1.8.0