From 1ba1ecb2760ff21fc66db7a91fa099db489dd385 Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期三, 02 八月 2023 17:10:28 +0800 Subject: [PATCH] add --- service/client.go | 157 ++++++++++++++++++++++++++-------------------------- 1 files changed, 78 insertions(+), 79 deletions(-) diff --git a/service/client.go b/service/client.go index ada1710..9eae140 100644 --- a/service/client.go +++ b/service/client.go @@ -1,79 +1,78 @@ -package service - -import ( - "aps_crm/model" - "aps_crm/pkg/ecode" -) - -type ClientService struct{} - -func (ClientService) AddClient(client *model.Client) int { - err := model.NewClientSearch(nil).Create(client) - if err != nil { - return ecode.ClientExist - } - 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 - err := model.NewClientSearch(nil).SetId(client.Id).Update(client) - if err != nil { - return ecode.ClientUpdateErr - } - - return ecode.OK -} - -func (ClientService) GetClientList(page, pageSize int, keyword string) ([]*model.Client, int64, int) { - // get contact list - contacts, total, err := model.NewClientSearch(nil).SetKeyword(keyword).SetPage(page, pageSize).Find() - if err != nil { - return nil, 0, ecode.ClientListErr - } - return contacts, total, ecode.OK -} - -func (ClientService) Assign(id, memberId int) int { - // check client exist - errCode := CheckClientExist(id) - if errCode != ecode.OK { - return errCode - } - - // assign client - err := model.NewClientSearch(nil).SetId(id).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 -} \ No newline at end of file +package service + +import ( + "aps_crm/model" + "aps_crm/pkg/ecode" +) + +type ClientService struct{} + +func (ClientService) AddClient(client *model.Client) int { + err := model.NewClientSearch(nil).Create(client) + if err != nil { + return ecode.ClientExist + } + 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 + err := model.NewClientSearch(nil).SetId(client.Id).Update(client) + if err != nil { + return ecode.ClientUpdateErr + } + + 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(id, memberId int) int { + // check client exist + errCode := CheckClientExist(id) + if errCode != ecode.OK { + return errCode + } + + // assign client + err := model.NewClientSearch(nil).SetId(id).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 +} -- Gitblit v1.8.0