zhangqian
2023-08-03 a89c3af49d0be3e635ef13b3bd3dd9410e4d6d46
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
package model
 
import (
    "aps_crm/constvar"
    "aps_crm/pkg/mysqlx"
    "errors"
    "fmt"
    "gorm.io/gorm"
)
 
type (
    // Severity 严重程度
    Severity struct {
        Id   int    `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
        Name   string    `json:"name" gorm:"column:name"`
    }
 
    // SeveritySearch 严重程度搜索条件
    SeveritySearch struct {
        Severity
        Orm *gorm.DB
        QueryClass  constvar.SeverityQueryClass
        KeywordType constvar.SeverityKeywordType
        Keyword     string
        PageNum  int
        PageSize int
    }
)
 
func (Severity) TableName() string {
    return "severity"
}
 
func NewSeveritySearch() *SeveritySearch {
    return &SeveritySearch{
        Orm: mysqlx.GetDB(),
    }
}
 
func (slf *SeveritySearch) build() *gorm.DB {
    var db = slf.Orm.Model(&Severity{})
    if slf.Id != 0 {
        db = db.Where("id = ?", slf.Id)
    }
 
    return db
}
 
func (slf *SeveritySearch) Create(record *Severity) error {
    var db = slf.build()
    return db.Create(record).Error
}
 
func (slf *SeveritySearch) Delete() error {
    var db = slf.build()
    return db.Delete(&Severity{}).Error
}
 
func (slf *SeveritySearch) Update(record *Severity) error {
    var db = slf.build()
    return db.Updates(record).Error
}
 
func (slf *SeveritySearch) FindAll() ([]*Severity, error) {
    var db = slf.build()
    var record = make([]*Severity, 0)
    err := db.Find(&record).Error
    return record, err
}
 
func (slf *SeveritySearch) SetId(id int) *SeveritySearch {
    slf.Id = id
    return slf
}
 
func (slf *SeveritySearch) First() (*Severity, error) {
    var db = slf.build()
    var record = new(Severity)
    err := db.First(record).Error
    return record, err
}
 
func (slf *SeveritySearch) Updates(values interface{}) error {
    var db = slf.build()
    return db.Updates(values).Error
}
 
func (slf *SeveritySearch) Save(record *Severity) error {
    if record.Id == 0 {
        return errors.New("id为空")
    }
    var db = slf.build()
 
    if err := db.Save(record).Error; err != nil {
        return fmt.Errorf("save err: %v, record: %+v", err, record)
    }
 
    return nil
}
 
func (slf *SeveritySearch) Find() ([]*Severity, int64, error) {
    var db = slf.build()
    var records = make([]*Severity, 0)
    var total int64
    if err := db.Count(&total).Error; err != nil {
        return records, total, err
    }
    if slf.PageNum > 0 && slf.PageSize > 0 {
        db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize)
    }
 
    err := db.Find(&records).Error
    return records, total, err
}