package service import ( "aps_crm/model" "aps_crm/model/request" "aps_crm/pkg/ecode" ) type ServiceTypeService struct{} func NewServiceTypeService() ServiceTypeService { return ServiceTypeService{} } func (ServiceTypeService) AddServiceType(ServiceType *model.ServiceType) int { err := model.NewServiceTypeSearch().Create(ServiceType) if err != nil { return ecode.DBErr } return ecode.OK } func (ServiceTypeService) DeleteServiceType(id int) int { err := model.NewServiceTypeSearch().SetId(id).Delete() if err != nil { return ecode.DBErr } return ecode.OK } func (ServiceTypeService) GetServiceTypeList() ([]*model.ServiceType, int64, int) { list, total, err := model.NewServiceTypeSearch().Find() if err != nil { return nil, 0, ecode.DBErr } return list, total, ecode.OK } func (ServiceTypeService) UpdateServiceTypes(ServiceTypes []*request.UpdateServiceType) int { for _, v := range ServiceTypes { // check ServiceType exist _, err := model.NewServiceTypeSearch().SetId(v.Id).First() if err != nil { return ecode.DBErr } err = model.NewServiceTypeSearch().SetId(v.Id).Updates(map[string]interface{}{}) if err != nil { return ecode.DBErr } } return ecode.OK } func (ServiceTypeService) UpdateServiceType(serviceType *model.ServiceType) int { err := model.NewServiceTypeSearch().SetId(serviceType.Id).Update(serviceType) if err != nil { return ecode.DBErr } return ecode.OK }