forked from statping-ng/statping-ng
-
Notifications
You must be signed in to change notification settings - Fork 0
/
database.go
93 lines (84 loc) · 1.83 KB
/
database.go
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
package notifications
import (
"github.com/nordcloud/statping-ng/database"
)
var (
db database.Database
)
func SetDB(database database.Database) {
db = database.Model(&Notification{})
}
func (n *Notification) Values() Values {
return Values{
Host: n.Host.String,
Port: n.Port.Int64,
Username: n.Username.String,
Password: n.Password.String,
Var1: n.Var1.String,
Var2: n.Var2.String,
ApiKey: n.ApiKey.String,
ApiSecret: n.ApiSecret.String,
}
}
func All() []*Notification {
var n []*Notification
q := db.Find(&n)
if q.Error() != nil {
return nil
}
return n
}
func Find(method string) (*Notification, error) {
var n Notification
q := db.Where("method = ?", method).Find(&n)
if q.Error() != nil {
return nil, q.Error()
}
return &n, nil
}
func (n *Notification) Create() error {
var p Notification
q := db.Where("method = ?", n.Method).Find(&p)
if q.RecordNotFound() {
log.Infof("Notifier '%s' was not found, adding into database...\n", n.Method)
if err := db.Create(n).Error(); err != nil {
return err
}
return nil
}
if p.FailureData.String == "" {
p.FailureData = n.FailureData
}
if p.SuccessData.String == "" {
p.SuccessData = n.SuccessData
}
if err := p.Update(); err != nil {
return err
}
return nil
}
func (n *Notification) UpdateFields(notif *Notification) *Notification {
if notif == nil {
return n
}
n.Id = notif.Id
n.Limits = notif.Limits
n.Enabled = notif.Enabled
n.Host = notif.Host
n.Port = notif.Port
n.Username = notif.Username
n.Password = notif.Password
n.ApiKey = notif.ApiKey
n.ApiSecret = notif.ApiSecret
n.Var1 = notif.Var1
n.Var2 = notif.Var2
n.SuccessData = notif.SuccessData
n.FailureData = notif.FailureData
return n
}
func (n *Notification) Update() error {
if err := db.Update(n).Error(); err != nil {
return err
}
return nil
}