From 530fed8ec225453572d57b15c200ab062c335457 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期三, 01 十一月 2023 19:20:21 +0800 Subject: [PATCH] 公海member_id使用0 --- service/receipt.go | 171 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 162 insertions(+), 9 deletions(-) diff --git a/service/receipt.go b/service/receipt.go index 834c6df..b4999cf 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,24 +16,128 @@ } func (ReceiptService) AddReceipt(receipt *model.Receipt) int { - err := model.NewReceiptSearch().Create(receipt) + var moneyAdder model.MoneyAdder + var sourceId int + var collection *model.ServiceCollectionPlan + var err error + switch receipt.SourceType { + case constvar.ReceiptSourceTypeSalesDetails: + moneyAdder = model.NewSalesDetailsSearch() + sourceId = receipt.SourceId + case constvar.ReceiptSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + sourceId = receipt.SourceId + case constvar.ReceiptSourceTypeCollectionPlan: + collection, err = model.NewServiceCollectionPlanSearch().SetId(receipt.SourceId).First() + if err != nil { + return ecode.DBErr + } + sourceId = collection.SourceId + switch collection.SourceType { + case constvar.CollectionSourceTypeSaleDetails: + moneyAdder = model.NewSalesDetailsSearch() + case constvar.CollectionSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + } + } + err = model.WithTransaction(func(db *gorm.DB) error { + err := model.NewReceiptSearch().SetOrm(db).Create(receipt) + if err != nil { + return err + } + if receipt.SourceType == constvar.ReceiptSourceTypeCollectionPlan { + err = model.NewServiceCollectionPlanSearch().SetOrm(db).UpdateStatusAndAmount(collection, receipt.Amount) + if err != nil { + return err + } + } + + // todo tmp + if sourceId != 0 { + err = moneyAdder.AmountReceivableAdd(db, sourceId, decimal.Zero.Sub(receipt.Amount).Round(2)) + if err != nil { + return err + } + + err = moneyAdder.AmountReceivedAdd(db, sourceId, receipt.Amount.Round(2)) + if err != nil { + return err + } + } + + return nil + }) 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 + var sourceId int + var collection *model.ServiceCollectionPlan + switch receipt.SourceType { + case constvar.ReceiptSourceTypeSalesDetails: + moneyAdder = model.NewSalesDetailsSearch() + sourceId = receipt.SourceId + case constvar.ReceiptSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + sourceId = receipt.SourceId + case constvar.ReceiptSourceTypeCollectionPlan: + collection, err = model.NewServiceCollectionPlanSearch().SetId(receipt.SourceId).First() + if err != nil { + return ecode.DBErr + } + sourceId = collection.SourceId + switch collection.SourceType { + case constvar.CollectionSourceTypeSaleDetails: + moneyAdder = model.NewSalesDetailsSearch() + case constvar.CollectionSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + } + } + err = model.WithTransaction(func(db *gorm.DB) error { + err := model.NewReceiptSearch().SetOrm(db).SetId(id).Delete() + if err != nil { + return err + } + + if receipt.SourceType == constvar.ReceiptSourceTypeCollectionPlan { + err = model.NewServiceCollectionPlanSearch().SetOrm(db).UpdateStatusAndAmount(collection, decimal.Zero.Sub(receipt.Amount)) + if err != nil { + return err + } + } + + err = moneyAdder.AmountReceivableAdd(db, sourceId, receipt.Amount.Round(2)) + if err != nil { + return err + } + err = moneyAdder.AmountReceivedAdd(db, sourceId, decimal.Zero.Sub(receipt.Amount).Round(2)) + if err != nil { + return err + } + return nil + }) if err != nil { return ecode.DBErr } return ecode.OK } -func (ReceiptService) GetReceiptList() ([]*model.Receipt, int64, int) { - list, total, err := model.NewReceiptSearch().Find() +func (ReceiptService) GetReceiptList(params request.GetReceiptList) ([]*model.Receipt, int64, int) { + list, total, err := model.NewReceiptSearch(). + SetPage(params.Page, params.PageSize). + SetSourceType(params.SourceType). + SetSourceId(params.SourceId). + SetClientId(params.ClientId). + SetPrincipalIds(params.PrincipleIds). + Find() if err != nil { return nil, 0, ecode.DBErr } @@ -46,9 +153,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 +163,57 @@ } func (ReceiptService) UpdateReceipt(receipt *model.Receipt) int { - err := model.NewReceiptSearch().SetId(receipt.Id).Save(receipt) + old, err := model.NewReceiptSearch().SetId(receipt.Id).First() + if err != nil { + return ecode.DBErr + } + var moneyAdder model.MoneyAdder + var sourceId int + var collection *model.ServiceCollectionPlan + switch receipt.SourceType { + case constvar.ReceiptSourceTypeSalesDetails: + moneyAdder = model.NewSalesDetailsSearch() + sourceId = receipt.SourceId + case constvar.ReceiptSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + sourceId = receipt.SourceId + case constvar.ReceiptSourceTypeCollectionPlan: + collection, err = model.NewServiceCollectionPlanSearch().SetId(receipt.SourceId).First() if err != nil { return ecode.DBErr } + sourceId = collection.SourceId + switch collection.SourceType { + case constvar.CollectionSourceTypeSaleDetails: + moneyAdder = model.NewSalesDetailsSearch() + case constvar.CollectionSourceTypeServiceContract: + moneyAdder = model.NewServiceContractSearch() + } + } + err = model.WithTransaction(func(db *gorm.DB) error { + err = model.NewReceiptSearch().SetId(receipt.Id).Update(receipt) + if err != nil { + return err + } + diffAmount := receipt.Amount.Sub(old.Amount) + if receipt.SourceType == constvar.ReceiptSourceTypeCollectionPlan { + err = model.NewServiceCollectionPlanSearch().SetOrm(db).UpdateStatusAndAmount(collection, diffAmount) + if err != nil { + return err + } + } + err = moneyAdder.AmountReceivedAdd(db, sourceId, diffAmount.Round(2)) + if err != nil { + return err + } + err = moneyAdder.AmountReceivableAdd(db, sourceId, decimal.Zero.Sub(diffAmount).Round(2)) + if err != nil { + return err + } + return nil + }) + if err != nil { + return ecode.DBErr + } return ecode.OK } -- Gitblit v1.8.0