fix

wangpengfei

2023-08-14 6acff3ce321e59b03e95fe7143ba45b0c48cd926

 service/client.go

@@ -3,15 +3,33 @@
import (
   "aps_crm/model"
   "aps_crm/pkg/ecode"
   "aps_crm/pkg/mysqlx"
)

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
}

@@ -33,9 +51,36 @@
   }

   // update client
   err := model.NewClientSearch(nil).SetId(client.Id).Update(client)
   if err != nil {
      return ecode.ClientUpdateErr
   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
         if contact.Id == 0 {
            err := model.NewContactSearch(tx).Create(&contact)
            if err != nil {
               tx.Rollback()
               return ecode.ClientUpdateErr
            }
         } else {
            err = model.NewContactSearch(tx).SetId(contact.Id).Update(&contact)
            if err != nil {
               tx.Rollback()
               return ecode.ClientUpdateErr
            }
         }
      }
      tx.Commit()
   }

   return ecode.OK