fix

wangpengfei

2023-08-24 274b44fa5b934eace2fc6dbcdccbb96a032545d8

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
package service
 
import (
    "aps_crm/model"
    "aps_crm/pkg/ecode"
    "aps_crm/pkg/mysqlx"
)
 
type SubOrderService struct{}
 
func (SubOrderService) AddSubOrder(subOrder *model.SubOrder) int {
    err := model.NewSubOrderSearch(nil).Create(subOrder)
    if err != nil {
        return ecode.SubOrderExist
    }
 
    return ecode.OK
}
 
func (SubOrderService) UpdateSubOrder(subOrder *model.SubOrder) int {
    // check subOrder exist
    tmp, err := model.NewSubOrderSearch(nil).SetId(subOrder.Id).Find()
    if err != nil {
        return ecode.SubOrderNotExist
    }
 
    if len(subOrder.Products) == 0 {
        err = model.NewSubOrderSearch(nil).SetId(subOrder.Id).Update(subOrder)
        if err != nil {
            return ecode.SubOrderSetErr
        }
    } else {
        tx := mysqlx.GetDB().Begin()
        err = tx.Model(tmp).Association("Products").Clear()
        if err != nil {
            tx.Rollback()
            return ecode.SubOrderSetErr
        }
 
        err = model.NewSubOrderSearch(tx).SetId(subOrder.Id).Update(subOrder)
        if err != nil {
            tx.Rollback()
            return ecode.SubOrderSetErr
        }
 
        for _, product := range subOrder.Products {
            if product.Id == 0 {
                err = model.NewProductSearch(tx).Create(&product)
                if err != nil {
                    tx.Rollback()
                    return ecode.SubOrderSetErr
                }
            } else {
                err = model.NewProductSearch(tx).SetId(product.Id).Update(&product)
                if err != nil {
                    tx.Rollback()
                    return ecode.SubOrderSetErr
                }
            }
            err = tx.Model(&tmp).Association("Products").Append(&product)
            if err != nil {
                tx.Rollback()
                return ecode.SubOrderSetErr
            }
        }
        tx.Commit()
    }
 
    return ecode.OK
}
 
func (SubOrderService) GetSubOrderList(page, pageSize int, data map[string]interface{}) ([]*model.SubOrder, int64, int) {
    // get contact list
    contacts, total, err := model.NewSubOrderSearch(nil).SetSearchMap(data).SetPage(page, pageSize).SetSearchMap(data).FindAll()
    if err != nil {
        return nil, 0, ecode.SubOrderListErr
    }
    return contacts, total, ecode.OK
}
 
func (SubOrderService) DeleteSubOrder(ids []int) int {
    // delete client
    err := model.NewSubOrderSearch(nil).SetIds(ids).Delete()
    if err != nil {
        return ecode.SubOrderDeleteErr
    }
    return ecode.OK
}