Merge branch 'master' of http://192.168.5.5:10010/r/aps/crm

jiangshuai

2024-02-06 116948bcec330a6dde841708515c222fcebf51a7

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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package model
 
import (
    "aps_crm/pkg/mysqlx"
    "gorm.io/gorm"
    "sync"
)
 
type (
    // Province 省份
    Province struct {
        Id        int    `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
        Name      string `json:"name" gorm:"column:name;type:varchar(255);comment:省份名称"`
        CountryId int    `json:"-" gorm:"column:country_id;type:int(11);comment:国家ID"`
        Cities    []City `json:"cities" gorm:"foreignKey:ProvinceId"`
    }
 
    // ProvinceSearch 省份搜索条件
    ProvinceSearch struct {
        Province
        Orm *gorm.DB
    }
)
 
func (Province) TableName() string {
    return "province"
}
 
func NewProvinceSearch(db *gorm.DB) *ProvinceSearch {
    if db == nil {
        db = mysqlx.GetDB()
    }
 
    return &ProvinceSearch{
        Orm: db,
    }
}
 
func (slf *ProvinceSearch) build() *gorm.DB {
    var db = slf.Orm.Model(&Province{})
    if slf.Id != 0 {
        db = db.Where("id = ?", slf.Id)
    }
    if slf.Name != "" {
        db = db.Where("name = ?", slf.Name)
    }
 
    if slf.CountryId != 0 {
        db = db.Where("country_id = ?", slf.CountryId)
    }
 
    return db
}
 
// Create 创建
func (slf *ProvinceSearch) Create(record *Province) error {
    var db = slf.build()
    return db.Create(record).Error
}
 
func (slf *ProvinceSearch) Delete() error {
    var db = slf.build()
    return db.Delete(&Province{}).Error
}
 
func (slf *ProvinceSearch) Update(record *Province) error {
    var db = slf.build()
    return db.Updates(record).Error
}
 
func (slf *ProvinceSearch) GetListByIds(ids []int) (result []Province, err error) {
    var db = slf.build()
    err = db.Where("id in (?)", ids).Find(&result).Error
    return
}
 
func (slf *ProvinceSearch) Find() (result []Province, err error) {
    var db = slf.build()
    err = db.Preload("country").Find(&result).Error
    return
}
 
func (slf *ProvinceSearch) SetId(id int) *ProvinceSearch {
    slf.Id = id
    return slf
}
 
func (slf *ProvinceSearch) SetName(name string) *ProvinceSearch {
    slf.Name = name
    return slf
}
 
func (slf *ProvinceSearch) SetCountryId(countryId int) *ProvinceSearch {
    slf.CountryId = countryId
    return slf
}
 
func (slf *ProvinceSearch) UpdateList(m map[string]interface{}, ids []int) error {
    var db = slf.build()
    return db.Where("id in (?)", ids).Updates(m).Error
}
 
func (slf *ProvinceSearch) First() (result Province, err error) {
    var db = slf.build()
    err = db.First(&result).Error
    return
}
 
func (slf *ProvinceSearch) FindAll() (result []*Province, err error) {
    var db = slf.build()
    err = db.Find(&result).Error
    return
}
 
func (slf *ProvinceSearch) Updates(m map[string]interface{}) error {
    var db = slf.build()
    return db.Updates(m).Error
}
 
// InitDefaultData 初始化数据
func (slf *ProvinceSearch) InitDefaultData(errCh chan<- error, wg *sync.WaitGroup) {
    var (
        db          = slf.Orm.Table(slf.TableName())
        total int64 = 0
    )
    if err := db.Count(&total).Error; err != nil {
        errCh <- err
        return
    }
    if total != 0 {
        return
    }
    sql := "INSERT INTO `province` VALUES (29,'北京市',3),(30,'天津市',3),(31,'河北省',3),(32,'山西省',3),(33,'内蒙古自治区',3),(34,'辽宁省',3),(35,'吉林省',3),(36,'黑龙江省',3),(37,'上海市',3),(38,'江苏省',3),(39,'浙江省',3),(40,'安徽省',3),(41,'福建省',3),(42,'江西省',3),(43,'山东省',3),(44,'河南省',3),(45,'湖北省',3),(46,'湖南省',3),(47,'广东省',3),(48,'广西壮族自治区',3),(49,'海南省',3),(50,'重庆市',3),(51,'四川省',3),(52,'贵州省',3),(53,'云南省',3),(54,'西藏自治区',3),(55,'陕西省',3),(56,'甘肃省',3),(57,'青海省',3),(58,'宁夏回族自治区',3),(59,'新疆维吾尔自治区',3),(60,'台湾省',3),(61,'香港',3),(62,'澳门',3)"
 
    err := slf.Orm.Exec(sql).Error
    if err != nil {
        errCh <- err
        return
    }
}