From c42cff043c049c0b12aca91bca31cd0799ea6f7a Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 11 八月 2023 13:54:19 +0800 Subject: [PATCH] 增删改收款单时更改销售明细和服务合同应收金额 --- service/receipt.go | 62 +++++++++++++++++++++++++++--- 1 files changed, 55 insertions(+), 7 deletions(-) diff --git a/service/receipt.go b/service/receipt.go index 834c6df..386beb5 100644 --- a/service/receipt.go +++ b/service/receipt.go @@ -1,9 +1,12 @@ package service import ( + "aps_crm/constvar" "aps_crm/model" "aps_crm/model/request" "aps_crm/pkg/ecode" + "github.com/shopspring/decimal" + "gorm.io/gorm" ) type ReceiptService struct{} @@ -13,16 +16,45 @@ } func (ReceiptService) AddReceipt(receipt *model.Receipt) int { - err := model.NewReceiptSearch().Create(receipt) + var moneyAdder model.MoneyAdder + switch receipt.SourceType { + case constvar.ReceiptSourceTypeSalesDetails: + moneyAdder = model.NewSalesDetailsSearch() + case constvar.ReceiptSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + } + err := model.WithTransaction(func(db *gorm.DB) error { + err := model.NewReceiptSearch().SetOrm(db).Create(receipt) + if err != nil { + return err + } + return moneyAdder.AmountReceivableAdd(db, receipt.SourceId, decimal.Zero.Sub(receipt.Amount).Round(2)) + }) if err != nil { return ecode.DBErr } - return ecode.OK } func (ReceiptService) DeleteReceipt(id int) int { - err := model.NewReceiptSearch().SetId(id).Delete() + receipt, err := model.NewReceiptSearch().SetId(id).First() + if err != nil { + return ecode.DBErr + } + var moneyAdder model.MoneyAdder + switch receipt.SourceType { + case constvar.ReceiptSourceTypeSalesDetails: + moneyAdder = model.NewSalesDetailsSearch() + case constvar.ReceiptSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + } + err = model.WithTransaction(func(db *gorm.DB) error { + err := model.NewReceiptSearch().SetId(id).Delete() + if err != nil { + return err + } + return moneyAdder.AmountReceivableAdd(db, receipt.SourceId, receipt.Amount.Round(2)) + }) if err != nil { return ecode.DBErr } @@ -46,9 +78,7 @@ return ecode.DBErr } - err = model.NewReceiptSearch().SetId(v.Id).Updates(map[string]interface{}{ - - }) + err = model.NewReceiptSearch().SetId(v.Id).Updates(map[string]interface{}{}) if err != nil { return ecode.DBErr } @@ -58,9 +88,27 @@ } func (ReceiptService) UpdateReceipt(receipt *model.Receipt) int { + old, err := model.NewReceiptSearch().SetId(receipt.Id).First() + if err != nil { + return ecode.DBErr + } + var moneyAdder model.MoneyAdder + switch receipt.SourceType { + case constvar.ReceiptSourceTypeSalesDetails: + moneyAdder = model.NewSalesDetailsSearch() + case constvar.ReceiptSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + } + err = model.WithTransaction(func(db *gorm.DB) error { err := model.NewReceiptSearch().SetId(receipt.Id).Save(receipt) if err != nil { - return ecode.DBErr + return err } + diffAmount := receipt.Amount.Sub(old.Amount) + return moneyAdder.AmountReceivableAdd(db, receipt.SourceId, diffAmount.Round(2)) + }) + if err != nil { + return ecode.DBErr + } return ecode.OK } -- Gitblit v1.8.0