package service
|
|
import (
|
"aps_crm/model"
|
"aps_crm/model/request"
|
"aps_crm/pkg/ecode"
|
)
|
|
type EnterpriseNatureService struct{}
|
|
func (EnterpriseNatureService) AddEnterpriseNature(enterpriseNature *model.EnterpriseNature) int {
|
err := model.NewEnterpriseNatureSearch().Create(enterpriseNature)
|
if err != nil {
|
return ecode.EnterpriseNatureExist
|
}
|
|
return ecode.OK
|
}
|
|
func (EnterpriseNatureService) DeleteEnterpriseNature(id int) int {
|
_, err := model.NewEnterpriseNatureSearch().SetId(id).Find()
|
if err != nil {
|
return ecode.EnterpriseNatureNotExist
|
}
|
|
err = model.NewEnterpriseNatureSearch().SetId(id).Delete()
|
if err != nil {
|
return ecode.EnterpriseNatureNotExist
|
}
|
return ecode.OK
|
}
|
|
func (EnterpriseNatureService) GetEnterpriseNatureList() ([]*model.EnterpriseNature, int) {
|
list, err := model.NewEnterpriseNatureSearch().FindAll()
|
if err != nil {
|
return nil, ecode.EnterpriseNatureListErr
|
}
|
|
return list, ecode.OK
|
}
|
|
func (EnterpriseNatureService) UpdateEnterpriseNature(enterpriseNatures []*request.UpdateEnterpriseNature) int {
|
for _, v := range enterpriseNatures {
|
// check enterpriseNature exist
|
_, err := model.NewEnterpriseNatureSearch().SetId(v.Id).Find()
|
if err != nil {
|
return ecode.EnterpriseNatureNotExist
|
}
|
|
err = model.NewEnterpriseNatureSearch().SetId(v.Id).Updates(map[string]interface{}{
|
"name": v.Name,
|
})
|
if err != nil {
|
return ecode.EnterpriseNatureSetErr
|
}
|
}
|
|
return ecode.OK
|
}
|