From 5f815df137f6edaaaddf869097b7e2718324acc6 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 13 十月 2023 13:53:22 +0800 Subject: [PATCH] 跟进记录简单数据权限 --- service/salesDetails.go | 53 ++++++++++++++++++++++++++++++++++++++++++++--------- 1 files changed, 44 insertions(+), 9 deletions(-) diff --git a/service/salesDetails.go b/service/salesDetails.go index fb9b9dd..fa497bf 100644 --- a/service/salesDetails.go +++ b/service/salesDetails.go @@ -5,6 +5,7 @@ "aps_crm/model" "aps_crm/pkg/ecode" "github.com/shopspring/decimal" + "gorm.io/gorm" ) type SalesDetailsService struct{} @@ -52,26 +53,60 @@ func (SalesDetailsService) UpdateSalesDetails(salesDetails *model.SalesDetails) int { // check salesDetails exist - old, err := model.NewSalesDetailsSearch().SetId(salesDetails.Id).First() + old, err := model.NewSalesDetailsSearch().SetId(salesDetails.Id).SetPreload(true).First() if err != nil { return ecode.SalesDetailsNotExist } var totalAmount decimal.Decimal totalAmount = salesDetails.AmountTotal - newProducts, removedProducts := NewProductsService().PickDiffProducts(salesDetails.Products, old.Products) - for _, product := range newProducts { + for _, product := range salesDetails.Products { totalAmount = totalAmount.Add(product.Amount.Mul(product.Price)) - } - removedProductIds := make([]uint, 0, len(removedProducts)) - for _, product := range removedProducts { - totalAmount = totalAmount.Sub(product.Amount.Mul(product.Price)) - removedProductIds = append(removedProductIds, product.Id) } salesDetails.AmountTotal = totalAmount.Round(2) salesDetails.AmountReceivable = salesDetails.AmountTotal.Sub(salesDetails.AmountReceived) salesDetails.AmountUnInvoiced = salesDetails.AmountTotal.Sub(salesDetails.AmountInvoiced) - err = model.NewSalesDetailsSearch().SetId(salesDetails.Id).Update(salesDetails) + newProducts, removedProducts := NewProductsService().PickDiffProducts(salesDetails.Products, old.Products) + err = model.WithTransaction(func(db *gorm.DB) error { + err = model.NewSalesDetailsSearch().SetId(salesDetails.Id).Update(salesDetails) + if err != nil { + return err + } + if len(removedProducts) > 0 { + removedProductIds := make([]uint, 0, len(removedProducts)) + for _, product := range removedProducts { + totalAmount = totalAmount.Sub(product.Amount.Mul(product.Price)) + removedProductIds = append(removedProductIds, product.Id) + } + err = model.NewProductSearch(db).SetIds(removedProductIds).Delete() + if err != nil { + return err + } + } + if len(newProducts) > 0 { + for _, p := range newProducts { + p.Id = 0 + } + err = model.NewProductSearch(db).CreateBatch(newProducts) + if err != nil { + return err + } + var rel []*model.SalesDetailsProduct + for _, p := range newProducts { + rel = append(rel, &model.SalesDetailsProduct{ + SalesDetailsId: salesDetails.Id, + ProductId: p.Id, + }) + } + err = model.NewSalesDetailsProductSearch().CreateBatch(rel) + if err != nil { + return err + } + } + return nil + + }) + if err != nil { return ecode.SalesDetailsSetErr } -- Gitblit v1.8.0