init

wangpengfei

2023-07-04 1142bc88cebcfedac1617749bc195a2615799518

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
package service
 
import (
    "aps_crm/model"
    "aps_crm/pkg/ecode"
    "gorm.io/gorm"
)
 
type ContactService struct{}
 
func (ContactService) AddContact(contact *model.Contact) int {
 
    code := checkMemberAndCompany(contact)
    if code != ecode.OK {
        return code
    }
 
    tx := model.NewContactSearch().Orm.Begin()
 
    // check isFirst
    errCode := setFirstContact(tx, contact)
    if errCode != ecode.OK {
        return errCode
    }
 
    err := model.NewContactSearch().Create(contact)
    if err != nil {
        tx.Rollback()
        return ecode.ContactExist
    }
 
    tx.Commit()
    return ecode.OK
}
 
func (ContactService) DeleteContact(id int) int {
    // check contact exist
    _, err := model.NewContactSearch().SetId(id).First()
    if err != nil {
        return ecode.ContactNotExist
    }
 
    // delete contact
    err = model.NewContactSearch().SetId(id).Delete()
    if err != nil {
        return ecode.ContactDeleteErr
    }
    return ecode.OK
}
 
func (ContactService) GetContactList() ([]*model.Contact, int) {
    // get contact list
    contacts, err := model.NewContactSearch().FindAll()
    if err != nil {
        return nil, ecode.ContactListErr
    }
    return contacts, ecode.OK
}
 
func (ContactService) UpdateContact(contact *model.Contact) int {
    // check contact exist
    _, err := model.NewContactSearch().SetId(contact.Id).First()
    if err != nil {
        return ecode.ContactNotExist
    }
 
    code := checkMemberAndCompany(contact)
    if code != ecode.OK {
        return code
    }
 
    tx := model.NewContactSearch().Orm.Begin()
    code = setFirstContact(tx, contact)
    if code != ecode.OK {
        return code
    }
 
    // update contact
    err = model.NewContactSearch().SetId(contact.Id).Update(contact)
    if err != nil {
        tx.Rollback()
        return ecode.ContactUpdateErr
    }
 
    tx.Commit()
 
    return ecode.OK
}
 
func checkMemberAndCompany(contact *model.Contact) int {
    // check client exist
    code := CheckClientExist(contact.ClientId)
    if code != ecode.OK {
        return code
    }
 
    // check member exist
    return ecode.OK
}
 
func setFirstContact(tx *gorm.DB, contact *model.Contact) int {
    if contact.IsFirst && contact.ClientId != 0 {
        err := model.NewContactSearch().SetClientId(contact.ClientId).UpdateByMap(map[string]interface{}{
            "is_first": false,
        })
        if err != nil {
            tx.Rollback()
            return ecode.ContactUpdateErr
        }
    }
 
    return ecode.OK
}
 
// CheckContactExist check contact exist
func CheckContactExist(id int) int {
    tmp, err := model.NewContactSearch().SetId(id).First()
    if err != nil {
        return ecode.ContactNotExist
    }
 
    if tmp.Id == 0 {
        return ecode.ContactNotExist
    }
 
    return ecode.OK
}