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/client.go | 139 +++++++++++++++++++++++++++++++++++---------- 1 files changed, 107 insertions(+), 32 deletions(-) diff --git a/service/client.go b/service/client.go index 260ef66..20cd752 100644 --- a/service/client.go +++ b/service/client.go @@ -3,41 +3,35 @@ import ( "aps_crm/model" "aps_crm/pkg/ecode" + "aps_crm/pkg/mysqlx" + "gorm.io/gorm" ) type ClientService struct{} -func (ClientService) AddClient(client *model.Client) int { - err := model.NewClientSearch(nil).Create(client) - if err != nil { - return ecode.ClientExist +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 -} - -func (ClientService) DeleteClient(id int) int { - // check client exist - _, err := model.NewClientSearch(nil).SetId(id).First() - if err != nil { - return ecode.ClientNotExist - } - - // delete client - err = model.NewClientSearch(nil).SetId(id).Delete() - if err != nil { - return ecode.ClientDeleteErr - } - return ecode.OK -} - -func (ClientService) GetClientList() (int, []*model.Client) { - // get client list - clients, err := model.NewClientSearch(nil).Find() - if err != nil { - return ecode.ClientListErr, nil - } - - return ecode.OK, clients } // CheckClientExist check client exist @@ -50,7 +44,7 @@ return ecode.OK } -func (ClientService) UpdateClient(client *model.Client) int { +func (ClientService) UpdateClient(client *model.Client, conId int) int { // check client exist errCode := CheckClientExist(client.Id) if errCode != ecode.OK { @@ -58,9 +52,90 @@ } // update client - err := model.NewClientSearch(nil).SetId(client.Id).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 + + // check isFirst + errCode = setFirstContact(tx, &contact) + if errCode != ecode.OK { + return errCode + } + + if conId == 0 { + err = model.NewContactSearch(tx).Create(&contact) + if err != nil { + tx.Rollback() + return ecode.ClientUpdateErr + } + } else { + contact.Id = conId + 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 ecode.ClientUpdateErr + 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 +} + +func (ClientService) CheckName(name string) int { + // check client exist + _, err := model.NewClientSearch(nil).SetName(name).First() + if err != gorm.ErrRecordNotFound { + return ecode.ClientExist } return ecode.OK -- Gitblit v1.8.0