feat(alerting): trying to get things to work with new model
This commit is contained in:
@@ -0,0 +1,77 @@
|
||||
package alerting
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/grafana/grafana/pkg/bus"
|
||||
"github.com/grafana/grafana/pkg/log"
|
||||
m "github.com/grafana/grafana/pkg/models"
|
||||
)
|
||||
|
||||
type RuleReader interface {
|
||||
Fetch() []*AlertRule
|
||||
}
|
||||
|
||||
type AlertRuleReader struct {
|
||||
sync.RWMutex
|
||||
serverID string
|
||||
serverPosition int
|
||||
clusterSize int
|
||||
}
|
||||
|
||||
func NewRuleReader() *AlertRuleReader {
|
||||
ruleReader := &AlertRuleReader{}
|
||||
|
||||
go ruleReader.initReader()
|
||||
return ruleReader
|
||||
}
|
||||
|
||||
func (arr *AlertRuleReader) initReader() {
|
||||
heartbeat := time.NewTicker(time.Second * 10)
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-heartbeat.C:
|
||||
arr.heartbeat()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (arr *AlertRuleReader) Fetch() []*AlertRule {
|
||||
cmd := &m.GetAllAlertsQuery{}
|
||||
err := bus.Dispatch(cmd)
|
||||
|
||||
if err != nil {
|
||||
log.Error(1, "Alerting: ruleReader.fetch(): Could not load alerts", err)
|
||||
return []*AlertRule{}
|
||||
}
|
||||
|
||||
res := make([]*AlertRule, len(cmd.Result))
|
||||
for i, ruleDef := range cmd.Result {
|
||||
model, _ := ConvetAlertModelToAlertRule(ruleDef)
|
||||
res[i] = model
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
func (arr *AlertRuleReader) heartbeat() {
|
||||
|
||||
//Lets cheat on this until we focus on clustering
|
||||
//log.Info("Heartbeat: Sending heartbeat from " + this.serverId)
|
||||
arr.clusterSize = 1
|
||||
arr.serverPosition = 1
|
||||
|
||||
/*
|
||||
cmd := &m.HeartBeatCommand{ServerId: this.serverId}
|
||||
err := bus.Dispatch(cmd)
|
||||
|
||||
if err != nil {
|
||||
log.Error(1, "Failed to send heartbeat.")
|
||||
} else {
|
||||
this.clusterSize = cmd.Result.ClusterSize
|
||||
this.serverPosition = cmd.Result.UptimePosition
|
||||
}
|
||||
*/
|
||||
}
|
||||
Reference in New Issue
Block a user