公海member_id使用0

zhangqian

2023-11-01 530fed8ec225453572d57b15c200ab062c335457

 service/quotation.go

@@ -3,12 +3,22 @@
import (
   "aps_crm/model"
   "aps_crm/pkg/ecode"
   "aps_crm/pkg/mysqlx"
)

type QuotationService struct{}

func (QuotationService) AddQuotation(quotation *model.Quotation) int {
   err := model.NewQuotationSearch().Create(quotation)
   //根据销售机会id去查询客户id
   if quotation.ClientId == 0 {
      record, err := model.NewSaleChanceSearch().SetId(quotation.SaleChanceId).Find()
      if err != nil {
         return ecode.SaleChanceNotExist
      }
      quotation.ClientId = record.ClientId
      quotation.ContactId = record.ContactId
   }
   err := model.NewQuotationSearch(nil).Create(quotation)
   if err != nil {
      return ecode.QuotationExist
   }
@@ -16,39 +26,72 @@
   return ecode.OK
}

func (QuotationService) DeleteQuotation(id int) int {
   _, err := model.NewQuotationSearch().SetId(id).Find()
   if err != nil {
      return ecode.QuotationNotExist
   }

   err = model.NewQuotationSearch().SetId(id).Delete()
   if err != nil {
      return ecode.QuotationNotExist
   }
   return ecode.OK
}

func (QuotationService) UpdateQuotation(quotation *model.Quotation) int {
   // check quotation exist
   _, err := model.NewQuotationSearch().SetId(quotation.Id).Find()
   tmp, err := model.NewQuotationSearch(nil).SetId(quotation.Id).Find()
   if err != nil {
      return ecode.QuotationNotExist
   }

   err = model.NewQuotationSearch().SetId(quotation.Id).Update(quotation)
   if err != nil {
      return ecode.QuotationSetErr
   if len(quotation.Products) == 0 {
      err = model.NewQuotationSearch(nil).SetId(quotation.Id).Update(quotation)
      if err != nil {
         return ecode.QuotationSetErr
      }
   } else {
      tx := mysqlx.GetDB().Begin()
      err = tx.Model(tmp).Association("Products").Clear()
      if err != nil {
         tx.Rollback()
         return ecode.QuotationSetErr
      }

      err = model.NewQuotationSearch(tx).SetId(quotation.Id).Update(quotation)
      if err != nil {
         tx.Rollback()
         return ecode.QuotationSetErr
      }

      for _, product := range quotation.Products {
         if product.Id == 0 {
            err = model.NewProductSearch(tx).Create(&product)
            if err != nil {
               tx.Rollback()
               return ecode.QuotationSetErr
            }
         } else {
            err = model.NewProductSearch(tx).SetId(product.Id).Update(&product)
            if err != nil {
               tx.Rollback()
               return ecode.QuotationSetErr
            }
         }
         err = tx.Model(&tmp).Association("Products").Append(&product)
         if err != nil {
            tx.Rollback()
            return ecode.QuotationSetErr
         }
      }
      tx.Commit()
   }

   return ecode.OK
}

func (QuotationService) GetQuotationList(page, pageSize int, keyword string) ([]*model.Quotation, int64, int) {
func (QuotationService) GetQuotationList(page, pageSize int, data map[string]interface{}) ([]*model.Quotation, int64, int) {
   // get contact list
   contacts, total, err := model.NewQuotationSearch().SetKeyword(keyword).SetPage(page, pageSize).FindAll()
   contacts, total, err := model.NewQuotationSearch(nil).SetPage(page, pageSize).SetSearchMap(data).FindAll()
   if err != nil {
      return nil, 0, ecode.QuotationListErr
   }
   return contacts, total, ecode.OK
}

func (QuotationService) DeleteQuotation(ids []int) int {
   // delete client
   err := model.NewQuotationSearch(nil).SetIds(ids).Delete()
   if err != nil {
      return ecode.QuotationDeleteErr
   }
   return ecode.OK
}