zhangzengfei
2023-09-04 e8e536d1cb52d2126c8c7ce2ba1c7a76f7208678
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
package models
 
type Modules struct {
    ProcName string `gorm:"column:proc_name;not null;default:''" json:"proc_name"` //模块名称
    NameZh   string `gorm:"column:name_zh;not null;default:''" json:"name_zh"`     //中文名称
}
 
type AllModules struct {
    modules []Modules
}
 
func (Modules) TableName() string {
    return "t_modules"
}
 
func (a *Modules) FindAll() (rows []Modules, err error) {
    if err := GetLogSetsDb().Table(a.TableName()).Find(&rows).Error; err != nil {
        return nil, err
    }
 
    return rows, nil
}
 
func (a *Modules) FindAllMap() map[string]string {
    m := make(map[string]string)
 
    list, err := a.FindAll()
    if nil == err {
        for _, item := range list {
            m[item.ProcName] = item.NameZh
        }
    }
 
    return m
}
 
func (a *Modules) Insert() bool {
    var count int64 = 0
    logSetsDb = GetLogSetsDb()
    result := logSetsDb.Table(a.TableName()).Where("proc_name=?", a.ProcName).Count(&count)
    if count > 0 {
        logSetsDb.Table(a.TableName()).Where("proc_name", a.ProcName).Update("name_zh", a.NameZh)
    }
    result = logSetsDb.Table(a.TableName()).Create(&a)
    if result.Error != nil {
        return false
    }
    return result.RowsAffected > 0
}