55 lines
961 B
Go
55 lines
961 B
Go
package alerting
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/log"
|
|
)
|
|
|
|
type SchedulerImpl struct {
|
|
jobs map[int64]*Job
|
|
log log.Logger
|
|
}
|
|
|
|
func NewScheduler() Scheduler {
|
|
return &SchedulerImpl{
|
|
jobs: make(map[int64]*Job, 0),
|
|
log: log.New("alerting.scheduler"),
|
|
}
|
|
}
|
|
|
|
func (s *SchedulerImpl) Update(rules []*Rule) {
|
|
s.log.Debug("Scheduling update", "ruleCount", len(rules))
|
|
|
|
jobs := make(map[int64]*Job, 0)
|
|
|
|
for i, rule := range rules {
|
|
var job *Job
|
|
if s.jobs[rule.Id] != nil {
|
|
job = s.jobs[rule.Id]
|
|
} else {
|
|
job = &Job{
|
|
Running: false,
|
|
}
|
|
}
|
|
|
|
job.Rule = rule
|
|
job.Offset = int64(i)
|
|
|
|
jobs[rule.Id] = job
|
|
}
|
|
|
|
s.jobs = jobs
|
|
}
|
|
|
|
func (s *SchedulerImpl) Tick(tickTime time.Time, execQueue chan *Job) {
|
|
now := tickTime.Unix()
|
|
|
|
for _, job := range s.jobs {
|
|
if now%job.Rule.Frequency == 0 && job.Running == false {
|
|
s.log.Debug("Scheduler: Putting job on to exec queue", "name", job.Rule.Name)
|
|
execQueue <- job
|
|
}
|
|
}
|
|
}
|