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/serviceContract.go | 197 ++++++++++++++++++++++++++++++++++--------------
1 files changed, 138 insertions(+), 59 deletions(-)
diff --git a/service/serviceContract.go b/service/serviceContract.go
index 2175971..c4fed1c 100644
--- a/service/serviceContract.go
+++ b/service/serviceContract.go
@@ -1,59 +1,138 @@
-package service
-
-import (
- "aps_crm/model"
- "aps_crm/pkg/ecode"
-)
-
-type SContractService struct{}
-
-func (SContractService) AddServiceContract(serviceContract *model.ServiceContract) int {
- err := model.NewServiceContractSearch().Create(serviceContract)
- if err != nil {
- return ecode.SContractExist
- }
-
- return ecode.OK
-}
-
-func (SContractService) UpdateServiceContract(serviceContract *model.ServiceContract) int {
- // check serviceContract exist
- _, err := model.NewServiceContractSearch().SetId(serviceContract.Id).Find()
- if err != nil {
- return ecode.SContractNotExist
- }
-
- err = model.NewServiceContractSearch().SetId(serviceContract.Id).Update(serviceContract)
- if err != nil {
- return ecode.SContractSetErr
- }
-
- return ecode.OK
-}
-
-//func (SContractService) GetServiceContractByContractId(contractId int) ([]*model.ServiceContract, int) {
-// list, err := model.NewServiceContractSearch().SetId(contractId).FindAll()
-// if err != nil {
-// return nil, ecode.SContractListErr
-// }
-//
-// return list, ecode.OK
-//}
-
-func (SContractService) GetServiceContractList(page, pageSize int, keyword string) ([]*model.ServiceContract, int64, int) {
- // get contact list
- contacts, total, err := model.NewServiceContractSearch().SetKeyword(keyword).SetPage(page, pageSize).FindAll()
- if err != nil {
- return nil, 0, ecode.SContractListErr
- }
- return contacts, total, ecode.OK
-}
-
-func (SContractService) DeleteServiceContract(ids []int) int {
- // delete client
- err := model.NewServiceContractSearch().SetIds(ids).Delete()
- if err != nil {
- return ecode.SContractDeleteErr
- }
- return ecode.OK
-}
+package service
+
+import (
+ "aps_crm/model"
+ "aps_crm/model/request"
+ "aps_crm/pkg/ecode"
+ "github.com/shopspring/decimal"
+ "gorm.io/gorm"
+)
+
+type SContractService struct{}
+
+func (SContractService) AddServiceContract(serviceContract *model.ServiceContract) int {
+ serviceContract.AmountReceivable = decimal.Zero.Round(2)
+ serviceContract.AmountInvoiced = decimal.Zero.Round(2)
+ serviceContract.AmountReceived = decimal.Zero.Round(2)
+ serviceContract.AmountUnInvoiced = decimal.Zero.Round(2)
+ serviceContract.AmountTotal = decimal.Zero.Round(2)
+ for _, product := range serviceContract.Products {
+ serviceContract.AmountTotal = serviceContract.AmountTotal.Add(product.Amount.Mul(product.Price))
+ }
+ serviceContract.AmountTotal = serviceContract.AmountTotal.Round(2)
+ serviceContract.AmountReceivable = serviceContract.AmountTotal
+ serviceContract.AmountUnInvoiced = serviceContract.AmountTotal
+
+ err := model.NewServiceContractSearch().Create(serviceContract)
+ if err != nil {
+ return ecode.SContractExist
+ }
+
+ return ecode.OK
+}
+
+func (SContractService) UpdateServiceContract(serviceContract *model.ServiceContract) int {
+ // check serviceContract exist
+ old, err := model.NewServiceContractSearch().SetId(serviceContract.Id).SetPreload(true).First()
+ if err != nil {
+ return ecode.SContractNotExist
+ }
+ var totalAmount decimal.Decimal
+ totalAmount = serviceContract.AmountTotal
+ newProducts, removedProducts := NewProductsService().PickDiffProducts(serviceContract.Products, old.Products)
+ for _, product := range serviceContract.Products {
+ totalAmount = totalAmount.Add(product.Amount.Mul(product.Price))
+ }
+ serviceContract.AmountTotal = totalAmount.Round(2)
+ serviceContract.AmountReceivable = serviceContract.AmountTotal.Sub(serviceContract.AmountReceived)
+ serviceContract.AmountUnInvoiced = serviceContract.AmountTotal.Sub(serviceContract.AmountInvoiced)
+ err = model.WithTransaction(func(db *gorm.DB) error {
+ err := model.NewServiceContractSearch().SetId(serviceContract.Id).Update(serviceContract)
+ if err != nil {
+ return err
+ }
+ if len(removedProducts) > 0 {
+ removedProductIds := make([]uint, 0, len(removedProducts))
+ for _, product := range removedProducts {
+ 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.ServiceContractProduct
+ for _, p := range newProducts {
+ rel = append(rel, &model.ServiceContractProduct{
+ ServiceContractId: serviceContract.Id,
+ ProductId: p.Id,
+ })
+ }
+ err = model.NewServiceContractProductSearch().CreateBatch(rel)
+ if err != nil {
+ return err
+ }
+ }
+ return nil
+ })
+
+ if err != nil {
+ return ecode.DBErr
+ }
+
+ return ecode.OK
+}
+
+//func (SContractService) GetServiceContractByContractId(contractId int) ([]*model.ServiceContract, int) {
+// list, err := model.NewServiceContractSearch().SetId(contractId).FindAll()
+// if err != nil {
+// return nil, ecode.SContractListErr
+// }
+//
+// return list, ecode.OK
+//}
+
+func (SContractService) DeleteServiceContract(ids []int) int {
+ // delete client
+ err := model.NewServiceContractSearch().SetIds(ids).Delete()
+ if err != nil {
+ return ecode.SContractDeleteErr
+ }
+ return ecode.OK
+}
+
+func (SContractService) BatchDeleteServiceContract(ids []int) int {
+ err := model.NewServiceContractSearch().DeleteByIds(ids)
+ if err != nil {
+ return ecode.DBErr
+ }
+ return ecode.OK
+}
+
+func (SContractService) GetServiceContractList(params request.GetServiceContractList, memberIds []int) ([]*model.ServiceContract, int64, int) {
+ // get contact list
+ contacts, total, err := model.NewServiceContractSearch().
+ SetKeyword(params.Keyword).
+ SetKeywordType(params.KeywordType).
+ SetQueryClass(params.QueryClass).
+ SetPage(params.Page, params.PageSize).
+ SetSalesDetailsId(params.SalesDetailsId).
+ SetQuotationId(params.QuotationId).
+ SetSaleChanceId(params.SaleChanceId).
+ SetContactId(params.ContactId).
+ SetMemberIds(memberIds).
+ SetPreload(true).
+ Find()
+ if err != nil {
+ return nil, 0, ecode.SContractListErr
+ }
+ return contacts, total, ecode.OK
+}
--
Gitblit v1.8.0