点击确认状态修改

liujiandao

2024-03-26 1c17ff16fd13e4d8bbab75d8a728cf18465b20e0

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
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
}