package service
|
|
import (
|
"aps_crm/model"
|
"aps_crm/model/request"
|
"aps_crm/pkg/ecode"
|
)
|
|
type EnterpriseScaleService struct{}
|
|
func (EnterpriseScaleService) AddEnterpriseScale(enterpriseScale *model.EnterpriseScale) int {
|
err := model.NewEnterpriseScaleSearch().Create(enterpriseScale)
|
if err != nil {
|
return ecode.EnterpriseScaleExist
|
}
|
|
return ecode.OK
|
}
|
|
func (EnterpriseScaleService) DeleteEnterpriseScale(id int) int {
|
_, err := model.NewEnterpriseScaleSearch().SetId(id).Find()
|
if err != nil {
|
return ecode.EnterpriseScaleNotExist
|
}
|
|
err = model.NewEnterpriseScaleSearch().SetId(id).Delete()
|
if err != nil {
|
return ecode.EnterpriseScaleNotExist
|
}
|
return ecode.OK
|
}
|
|
func (EnterpriseScaleService) GetEnterpriseScaleList() ([]*model.EnterpriseScale, int) {
|
list, err := model.NewEnterpriseScaleSearch().FindAll()
|
if err != nil {
|
return nil, ecode.EnterpriseScaleListErr
|
}
|
|
return list, ecode.OK
|
}
|
|
func (EnterpriseScaleService) UpdateEnterpriseScale(enterpriseScales []*request.UpdateEnterpriseScale) int {
|
for _, v := range enterpriseScales {
|
// check enterpriseScale exist
|
_, err := model.NewEnterpriseScaleSearch().SetId(v.Id).Find()
|
if err != nil {
|
return ecode.EnterpriseScaleNotExist
|
}
|
|
err = model.NewEnterpriseScaleSearch().SetId(v.Id).Updates(map[string]interface{}{
|
"name": v.Name,
|
})
|
if err != nil {
|
return ecode.EnterpriseScaleSetErr
|
}
|
}
|
|
return ecode.OK
|
}
|