94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
package database
|
|
|
|
import (
|
|
"regexp"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const AlertMsg = "alert service from logmania, device did not send new message for a while"
|
|
|
|
type DB struct {
|
|
// depraced Format -> transformation to new format by db.update()
|
|
Hostname map[string]string `json:"hostname,omitempty"`
|
|
HostTo map[string]map[string]bool `json:"host_to,omitempty"`
|
|
MaxPrioIn map[string]log.Level `json:"maxLevel,omitempty"`
|
|
RegexIn map[string]map[string]*regexp.Regexp `json:"regexIn,omitempty"`
|
|
Lastseen map[string]time.Time `json:"lastseen,omitempty"`
|
|
LastseenNotify map[string]time.Time `json:"-"`
|
|
// new format
|
|
Hosts []*Host `json:"hosts"`
|
|
HostsByAddress map[string]*Host `json:"-"`
|
|
HostsByName map[string]*Host `json:"-"`
|
|
Notifies []*Notify `json:"notifies"`
|
|
NotifiesByAddress map[string]*Notify `json:"-"`
|
|
DefaultNotify []*Notify `json:"-"`
|
|
}
|
|
|
|
func (db *DB) SendTo(e *log.Entry) (*log.Entry, *Host, []*Notify) {
|
|
addr, ok := e.Data["hostname"].(string)
|
|
if !ok {
|
|
return e, nil, nil
|
|
}
|
|
var host *Host
|
|
if host, ok := db.HostsByAddress[addr]; ok {
|
|
if e.Message != AlertMsg {
|
|
host.Lastseen = time.Now()
|
|
}
|
|
var toList []*Notify
|
|
entry := e
|
|
if host.Name != "" {
|
|
entry = entry.WithField("hostname", host.Name)
|
|
entry.Level = e.Level
|
|
entry.Message = e.Message
|
|
}
|
|
// return default notify list
|
|
if host.Notifies == nil || len(host.Notifies) == 0 {
|
|
return entry, host, db.DefaultNotify
|
|
}
|
|
// return with host specify list
|
|
for _, notify := range host.NotifiesByAddress {
|
|
if lvl := notify.MaxPrioIn; e.Level >= lvl {
|
|
continue
|
|
}
|
|
stopForTo := false
|
|
for _, expr := range notify.RegexIn {
|
|
if expr.MatchString(e.Message) {
|
|
stopForTo = true
|
|
continue
|
|
}
|
|
}
|
|
if stopForTo {
|
|
continue
|
|
}
|
|
toList = append(toList, notify)
|
|
}
|
|
return entry, host, toList
|
|
}
|
|
host = db.NewHost(addr)
|
|
return e, host, db.DefaultNotify
|
|
}
|
|
|
|
func (db *DB) Alert(expired time.Duration, send func(e *log.Entry, n *Notify) bool) {
|
|
c := time.Tick(time.Minute)
|
|
|
|
for range c {
|
|
now := time.Now()
|
|
for _, h := range db.Hosts {
|
|
if !h.Lastseen.Before(now.Add(expired * -2)) {
|
|
continue
|
|
}
|
|
if h.LastseenNotify.Year() <= 1 && h.Lastseen.Before(h.LastseenNotify) {
|
|
continue
|
|
}
|
|
h.LastseenNotify = now
|
|
entry := log.NewEntry(log.New())
|
|
entry.Level = log.ErrorLevel
|
|
entry.Message = AlertMsg
|
|
entry.WithField("hostname", h.Address)
|
|
send(entry, nil)
|
|
}
|
|
}
|
|
}
|