From cd9a3c446cff66e589cdb26163c3e2baf4bc6a1e Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 30 八月 2023 13:37:19 +0800
Subject: [PATCH] fix
---
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