fix
wangpengfei
2023-08-09 d204c9732f7b406a1487d11c79f6c121eea75850
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
package service
 
import (
    "aps_crm/model"
    "aps_crm/pkg/ecode"
    "aps_crm/pkg/mysqlx"
)
 
type ClientService struct{}
 
func (ClientService) AddClient(client *model.Client, sId int) int {
    if sId == 0 {
        err := model.NewClientSearch(nil).Create(client)
        if err != nil {
            return ecode.ClientExist
        }
    } else {
        tx := mysqlx.GetDB().Begin()
        err := model.NewClientSearch(tx).Create(client)
        if err != nil {
            tx.Rollback()
            return ecode.ClientExist
        }
 
        err = model.NewSalesLeadsSearch(tx).SetId(sId).Delete()
        if err != nil {
            tx.Rollback()
            return ecode.ClientExist
        }
        tx.Commit()
    }
 
    return ecode.OK
}
 
// CheckClientExist check client exist
func CheckClientExist(id int) int {
    _, err := model.NewClientSearch(nil).SetId(id).First()
    if err != nil {
        return ecode.ClientNotExist
    }
 
    return ecode.OK
}
 
func (ClientService) UpdateClient(client *model.Client) int {
    // check client exist
    errCode := CheckClientExist(client.Id)
    if errCode != ecode.OK {
        return errCode
    }
 
    // update client
    if len(client.Contacts) == 0 {
        err := model.NewClientSearch(nil).SetId(client.Id).Update(client)
        if err != nil {
            return ecode.ClientUpdateErr
        }
    } else {
        tx := mysqlx.GetDB().Begin()
        err := model.NewClientSearch(tx).SetId(client.Id).Update(client)
        if err != nil {
            tx.Rollback()
            return ecode.ClientUpdateErr
        }
 
        for _, contact := range client.Contacts {
            contact.ClientId = client.Id
            if contact.Id == 0 {
                err := model.NewContactSearch(tx).Create(&contact)
                if err != nil {
                    tx.Rollback()
                    return ecode.ClientUpdateErr
                }
            } else {
                err = model.NewContactSearch(tx).SetId(contact.Id).Update(&contact)
                if err != nil {
                    tx.Rollback()
                    return ecode.ClientUpdateErr
                }
            }
        }
        tx.Commit()
    }
 
    return ecode.OK
}
 
func (ClientService) GetClientList(page, pageSize int, data map[string]interface{}) ([]*model.Client, int64, int) {
    // get contact list
    contacts, total, err := model.NewClientSearch(nil).SetPage(page, pageSize).SetSearchMap(data).Find()
    if err != nil {
        return nil, 0, ecode.ClientListErr
    }
    return contacts, total, ecode.OK
}
 
func (ClientService) Assign(ids []int, memberId int) int {
    // check client exist
    //errCode := CheckClientExist(id)
    //if errCode != ecode.OK {
    //    return errCode
    //}
 
    // assign client
    err := model.NewClientSearch(nil).SetIds(ids).UpdateByMap(map[string]interface{}{
        "member_id": memberId,
    })
    if err != nil {
        return ecode.AssignErr
    }
 
    return ecode.OK
}
 
func (ClientService) DeleteClient(ids []int) int {
    // delete client
    err := model.NewClientSearch(nil).SetIds(ids).Delete()
    if err != nil {
        return ecode.ClientDeleteErr
    }
    return ecode.OK
}