From 3545b1af80395204e96f748cb6c52cf1dc572606 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期三, 30 八月 2023 14:34:35 +0800 Subject: [PATCH] merge --- service/purchase/purchase.go | 105 +++++++++++++++++++++++++++++++++++++--------------- 1 files changed, 75 insertions(+), 30 deletions(-) diff --git a/service/purchase/purchase.go b/service/purchase/purchase.go index bbef0f1..2251d8e 100644 --- a/service/purchase/purchase.go +++ b/service/purchase/purchase.go @@ -2,6 +2,7 @@ import ( "context" + "errors" "github.com/shopspring/decimal" "github.com/spf13/cast" "gorm.io/gorm" @@ -23,24 +24,44 @@ //@param: params *purchaserequest.AddPurchase //@return: err error -func (slf *PurchaseService) CreatePurchase(params *purchase.Purchase, productList []*purchase.PurchaseProducts) (err error) { +func (slf *PurchaseService) CreatePurchase(record *purchase.Purchase, productList []*purchase.PurchaseProducts) (err error) { + err = DealPrice(record, productList) + if err != nil { + return err + } err = global.GVA_DB.Transaction(func(tx *gorm.DB) error { - var quantity decimal.Decimal - for _, product := range productList { - quantity = quantity.Add(product.Amount) - } - params.Quantity = quantity - err = tx.Create(¶ms).Error + err = tx.Create(&record).Error if err != nil { return err } for _, product := range productList { - product.PurchaseId = cast.ToInt(params.ID) + product.PurchaseId = cast.ToInt(record.ID) } return tx.Create(productList).Error }) return err +} + +func DealPrice(record *purchase.Purchase, productList []*purchase.PurchaseProducts) error { + var quantity decimal.Decimal + var totalPrice decimal.Decimal + var realTotalPrice decimal.Decimal + for _, product := range productList { + quantity = quantity.Add(product.Amount) + totalPrice = totalPrice.Add(product.Price.Mul(product.Amount)) + } + record.Quantity = quantity + if !totalPrice.Equal(record.TotalPrice) { + return errors.New("浠风◣鎬昏璁$畻閿欒") + } + realTotalPrice = record.CalcRealTotalPrice() + if !realTotalPrice.Equal(record.RealTotalPrice) { + return errors.New("鏈�缁堜环鏍艰绠楅敊璇�") + } + record.UnInvoiceAmount = record.RealTotalPrice + record.ShouldPayAmount = record.RealTotalPrice + return nil } //@function: DeletePurchase @@ -65,12 +86,11 @@ //@return: err error func (slf *PurchaseService) UpdatePurchase(params *purchase.Purchase, productList []*purchase.PurchaseProducts) (err error) { + err = DealPrice(params, productList) + if err != nil { + return err + } err = global.GVA_DB.Transaction(func(tx *gorm.DB) error { - var quantity decimal.Decimal - for _, product := range productList { - quantity = quantity.Add(product.Amount) - } - params.Quantity = quantity err = tx.Where("id = ?", params.ID).Updates(params).Error if err != nil { return err @@ -115,7 +135,7 @@ Or("Product.name like ?", "%"+info.Keyword+"%"). Or("supplier.name like ?", "%"+info.Keyword+"%") } - var purchaseList []purchase.Purchase + var purchaseList []*purchase.Purchase err = db.Count(&total).Error if err != nil { return purchaseList, total, err @@ -132,7 +152,7 @@ func (slf *PurchaseService) GetPurchaseProductList(purchaseId uint) (list []*purchase.PurchaseProducts, err error) { db := global.GVA_DB.Model(&purchase.PurchaseProducts{}) list = make([]*purchase.PurchaseProducts, 0) - err = db.Where("purchase_id = ?", purchaseId).Find(&list).Error + err = db.Where("purchase_id = ?", purchaseId).Preload("Product").Find(&list).Error return list, err } @@ -194,19 +214,23 @@ continue } inspectOrder := &qualityinspect.QualityInspect{ - InspectType: qualityinspect.InspectType_InspectTypePurchase, - MaterialType: qualityinspect.MaterialType_MaterialTypeRaw, - MaterialName: product.Name, - MaterialId: product.Number, - MaterialTp: product.ModelNumber, - MaterialUnit: product.Unit, - Supplier: record.Supplier.Name, - WarehouseName: "閲囪喘鎬讳粨", - ReportAmount: productItem.Amount.InexactFloat64(), - InspectMethod: qualityinspect.InspectMethod_InspectMethodAll, - InspectAmount: productItem.Amount.InexactFloat64(), + InspectType: qualityinspect.InspectType_InspectTypePurchase, + MaterialType: qualityinspect.MaterialType_MaterialTypeRaw, + MaterialName: product.Name, + MaterialId: product.Number, + MaterialTp: product.ModelNumber, + MaterialUnit: product.Unit, + Supplier: record.Supplier.Name, + WarehouseName: "閲囪喘鎬讳粨", + ReportAmount: productItem.Amount.InexactFloat64(), + InspectMethod: qualityinspect.InspectMethod_InspectMethodAll, + InspectAmount: productItem.Amount.InexactFloat64(), + PurchaseOrderId: record.Number, } inspectOrders = append(inspectOrders, inspectOrder) + } + if len(inspectOrders) == 0 { + return nil } inspectRequest := qualityinspect.SendPurchaseInspectRequest{List: inspectOrders} _, err = qualityinspect.NewQualityInspectServiceClient(qualityinspect.Conn).SendPurchaseInspect(context.Background(), &inspectRequest) @@ -214,15 +238,36 @@ } func (slf *PurchaseService) SavePurchaseType(list []*purchase.PurchaseType) (err error) { + ids := make([]uint, 0) + newRecords := make([]*purchase.PurchaseType, 0) + for _, item := range list { + if item.ID != 0 { + ids = append(ids, item.ID) + } else { + newRecords = append(newRecords, item) + } + } err = global.GVA_DB.Transaction(func(tx *gorm.DB) error { - err = tx.Where("1 = ?", 1).Delete(&purchase.PurchaseType{}).Error + err = tx.Where("id not in ?", ids).Delete(&purchase.PurchaseType{}).Error if err != nil { return err } - for _, item := range list { - item.ID = 0 + if len(newRecords) > 0 { + err = tx.Create(newRecords).Error + if err != nil { + return err + } } - return tx.Create(list).Error + + for _, item := range list { + if item.ID != 0 { + err = tx.Save(item).Error + if err != nil { + return err + } + } + } + return nil }) return err } -- Gitblit v1.8.0