fix

wangpengfei

2023-08-24 274b44fa5b934eace2fc6dbcdccbb96a032545d8

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package service
 
import (
    "aps_crm/model"
    "aps_crm/model/request"
    "aps_crm/pkg/ecode"
    "fmt"
)
 
type CityService struct{}
 
func (CityService) AddCity(city []*model.City) int {
    err := model.NewCitySearch().Create(city)
    if err != nil {
        fmt.Println(err)
        return ecode.CityExist
    }
 
    return ecode.OK
}
 
func (CityService) UpdateCity(city []request.UpdateCity) int {
    for _, v := range city {
        // check city exist
        _, err := model.NewCitySearch().SetId(v.Id).First()
        if err != nil {
            return ecode.CityNotExist
        }
 
        err = model.NewCitySearch().SetId(v.Id).Updates(map[string]interface{}{
            "name": v.Name,
        })
        if err != nil {
            return ecode.CityNotExist
        }
    }
 
    return ecode.OK
}
 
func (CityService) SetCities(provinceId int, list []int) int {
    // check province exist
    _, err := model.NewProvinceSearch(nil).SetId(provinceId).First()
    if err != nil {
        return ecode.ProvinceNotExist
    }
 
    // update city list
    err = model.NewCitySearch().UpdateList(map[string]interface{}{
        "province_id": provinceId,
    }, list)
    if err != nil {
        return ecode.CitySetErr
    }
 
    return ecode.OK
}
 
func (CityService) ListCities(id int) ([]*model.City, int) {
    // get city list
    if id != 0 {
        list, err := model.NewCitySearch().SetProvinceId(id).FindAll()
        if err != nil {
            return nil, ecode.CityListErr
        }
        return list, ecode.OK
    } else {
        list, err := model.NewCitySearch().FindAll()
        if err != nil {
            return nil, ecode.CityListErr
        }
        return list, ecode.OK
    }
}
 
func (CityService) DeleteCity(id int) int {
    // check city exist
    _, err := model.NewCitySearch().SetId(id).First()
    if err != nil {
        return ecode.CityNotExist
    }
 
    // delete city
    err = model.NewCitySearch().SetId(id).Delete()
    if err != nil {
        return ecode.CityDeleteErr
    }
 
    return ecode.OK
 
}