Attach global labels to outgoing alerts.

This commit is contained in:
Fabian Reinartz 2015-09-01 22:17:02 +02:00
parent 9bbd9264e2
commit 5fed076a76
2 changed files with 24 additions and 2 deletions

View File

@ -93,7 +93,7 @@ func Main() int {
webHandler := web.New(memStorage, queryEngine, ruleManager, status, &cfg.web)
reloadables := []Reloadable{status, targetManager, ruleManager, webHandler}
reloadables := []Reloadable{status, targetManager, ruleManager, webHandler, notificationHandler}
if !reloadConfig(cfg.configFile, reloadables...) {
return 1

View File

@ -20,12 +20,14 @@ import (
"io/ioutil"
"net/http"
"strings"
"sync"
"time"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/common/model"
"github.com/prometheus/log"
"github.com/prometheus/prometheus/config"
"github.com/prometheus/prometheus/util/httputil"
)
@ -86,6 +88,8 @@ type NotificationHandler struct {
notificationsQueueLength prometheus.Gauge
notificationsQueueCapacity prometheus.Metric
globalLabels model.LabelSet
mtx sync.RWMutex
stopped chan struct{}
}
@ -141,10 +145,28 @@ func NewNotificationHandler(o *NotificationHandlerOptions) *NotificationHandler
}
}
// ApplyConfig updates the status state as the new config requires.
// Returns true on success.
func (n *NotificationHandler) ApplyConfig(conf *config.Config) bool {
n.mtx.Lock()
defer n.mtx.Unlock()
n.globalLabels = conf.GlobalConfig.Labels
return true
}
// Send a list of notifications to the configured alert manager.
func (n *NotificationHandler) sendNotifications(reqs NotificationReqs) error {
n.mtx.RLock()
defer n.mtx.RUnlock()
alerts := make([]map[string]interface{}, 0, len(reqs))
for _, req := range reqs {
for ln, lv := range n.globalLabels {
if _, ok := req.Labels[ln]; !ok {
req.Labels[ln] = lv
}
}
alerts = append(alerts, map[string]interface{}{
"summary": req.Summary,
"description": req.Description,