add

wangpengfei

2023-08-02 1ba1ecb2760ff21fc66db7a91fa099db489dd385

 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

}
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
}