wangpengfei
2023-08-07 9c2489c0c360c8dda36d3cbe1dba79222096b2dd
service/serviceContract.go
@@ -1,59 +1,64 @@
package service
import (
   "aps_crm/model"
   "aps_crm/pkg/ecode"
)
type SContractService struct{}
func (SContractService) AddServiceContract(serviceContract *model.ServiceContract) int {
   err := model.NewServiceContractSearch().Create(serviceContract)
   if err != nil {
      return ecode.SContractExist
   }
   return ecode.OK
}
func (SContractService) UpdateServiceContract(serviceContract *model.ServiceContract) int {
   // check serviceContract exist
   _, err := model.NewServiceContractSearch().SetId(serviceContract.Id).Find()
   if err != nil {
      return ecode.SContractNotExist
   }
   err = model.NewServiceContractSearch().SetId(serviceContract.Id).Update(serviceContract)
   if err != nil {
      return ecode.SContractSetErr
   }
   return ecode.OK
}
//func (SContractService) GetServiceContractByContractId(contractId int) ([]*model.ServiceContract, int) {
//   list, err := model.NewServiceContractSearch().SetId(contractId).FindAll()
//   if err != nil {
//      return nil, ecode.SContractListErr
//   }
//
//   return list, ecode.OK
//}
func (SContractService) GetServiceContractList(page, pageSize int, keyword string) ([]*model.ServiceContract, int64, int) {
   // get contact list
   contacts, total, err := model.NewServiceContractSearch().SetKeyword(keyword).SetPage(page, pageSize).FindAll()
   if err != nil {
      return nil, 0, ecode.SContractListErr
   }
   return contacts, total, ecode.OK
}
func (SContractService) DeleteServiceContract(ids []int) int {
   // delete client
   err := model.NewServiceContractSearch().SetIds(ids).Delete()
   if err != nil {
      return ecode.SContractDeleteErr
   }
   return ecode.OK
}
package service
import (
   "aps_crm/constvar"
   "aps_crm/model"
   "aps_crm/pkg/ecode"
)
type SContractService struct{}
func (SContractService) AddServiceContract(serviceContract *model.ServiceContract) int {
   err := model.NewServiceContractSearch().Create(serviceContract)
   if err != nil {
      return ecode.SContractExist
   }
   return ecode.OK
}
func (SContractService) UpdateServiceContract(serviceContract *model.ServiceContract) int {
   // check serviceContract exist
   _, err := model.NewServiceContractSearch().SetId(serviceContract.Id).Find()
   if err != nil {
      return ecode.SContractNotExist
   }
   err = model.NewServiceContractSearch().SetId(serviceContract.Id).Update(serviceContract)
   if err != nil {
      return ecode.SContractSetErr
   }
   return ecode.OK
}
//func (SContractService) GetServiceContractByContractId(contractId int) ([]*model.ServiceContract, int) {
//   list, err := model.NewServiceContractSearch().SetId(contractId).FindAll()
//   if err != nil {
//      return nil, ecode.SContractListErr
//   }
//
//   return list, ecode.OK
//}
func (SContractService) DeleteServiceContract(ids []int) int {
   // delete client
   err := model.NewServiceContractSearch().SetIds(ids).Delete()
   if err != nil {
      return ecode.SContractDeleteErr
   }
   return ecode.OK
}
func (SContractService) GetServiceContractList(page, pageSize int, queryClass constvar.ServiceContractQueryClass, keywordType constvar.ServiceContractKeywordType, keyword string) ([]*model.ServiceContract, int64, int) {
   // get contact list
   contacts, total, err := model.NewServiceContractSearch().
      SetKeyword(keyword).
      SetKeywordType(keywordType).
      SetQueryClass(queryClass).
      SetPage(page, pageSize).FindAll()
   if err != nil {
      return nil, 0, ecode.SContractListErr
   }
   return contacts, total, ecode.OK
}