mirror of
https://github.com/grafana/grafana.git
synced 2025-02-15 10:03:33 -06:00
76ea0b15ae
* Update GetAlertRulesForScheduling to query for folders (if needed) * Update scheduler's alertRulesRegistry to cache folder titles along with rules * Update rule eval loop to take folder title from the * Extract interface RuleStore * Pre-fetch the rule keys with the version to detect changes, and query the full table only if there are changes.
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package schedule
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"hash/fnv"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
)
|
|
|
|
// hashUIDs returns a fnv64 hash of the UIDs for all alert rules.
|
|
// The order of the alert rules does not matter as hashUIDs sorts
|
|
// the UIDs in increasing order.
|
|
func hashUIDs(alertRules []*models.AlertRule) uint64 {
|
|
h := fnv.New64()
|
|
for _, uid := range sortedUIDs(alertRules) {
|
|
// We can ignore err as fnv64 does not return an error
|
|
// nolint:errcheck,gosec
|
|
h.Write([]byte(uid))
|
|
}
|
|
return h.Sum64()
|
|
}
|
|
|
|
// sortedUIDs returns a slice of sorted UIDs.
|
|
func sortedUIDs(alertRules []*models.AlertRule) []string {
|
|
uids := make([]string, 0, len(alertRules))
|
|
for _, alertRule := range alertRules {
|
|
uids = append(uids, alertRule.UID)
|
|
}
|
|
sort.Strings(uids)
|
|
return uids
|
|
}
|
|
|
|
// updateSchedulableAlertRules updates the alert rules for the scheduler.
|
|
// It returns an error if the database is unavailable or the query returned
|
|
// an error.
|
|
func (sch *schedule) updateSchedulableAlertRules(ctx context.Context) error {
|
|
start := time.Now()
|
|
defer func() {
|
|
sch.metrics.UpdateSchedulableAlertRulesDuration.Observe(
|
|
time.Since(start).Seconds())
|
|
}()
|
|
|
|
if !sch.schedulableAlertRules.isEmpty() {
|
|
keys, err := sch.ruleStore.GetAlertRulesKeysForScheduling(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !sch.schedulableAlertRules.needsUpdate(keys) {
|
|
sch.log.Info("no changes detected. Skip updating")
|
|
return nil
|
|
}
|
|
}
|
|
|
|
q := models.GetAlertRulesForSchedulingQuery{
|
|
PopulateFolders: !sch.disableGrafanaFolder,
|
|
}
|
|
if err := sch.ruleStore.GetAlertRulesForScheduling(ctx, &q); err != nil {
|
|
return fmt.Errorf("failed to get alert rules: %w", err)
|
|
}
|
|
sch.log.Debug("alert rules fetched", "rules_count", len(q.ResultRules), "folders_count", len(q.ResultFoldersTitles))
|
|
sch.schedulableAlertRules.set(q.ResultRules, q.ResultFoldersTitles)
|
|
return nil
|
|
}
|