mirror of
https://github.com/grafana/grafana.git
synced 2024-11-27 11:20:27 -06:00
134 lines
3.7 KiB
Go
134 lines
3.7 KiB
Go
package notifiers
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/alerting"
|
|
)
|
|
|
|
func init() {
|
|
alerting.RegisterNotifier(&alerting.NotifierPlugin{
|
|
Type: "kafka",
|
|
Name: "Kafka REST Proxy",
|
|
Description: "Sends notifications to Kafka Rest Proxy",
|
|
Heading: "Kafka settings",
|
|
Factory: NewKafkaNotifier,
|
|
Options: []alerting.NotifierOption{
|
|
{
|
|
Label: "Kafka REST Proxy",
|
|
Element: alerting.ElementTypeInput,
|
|
InputType: alerting.InputTypeText,
|
|
Placeholder: "http://localhost:8082",
|
|
PropertyName: "kafkaRestProxy",
|
|
Required: true,
|
|
},
|
|
{
|
|
Label: "Topic",
|
|
Element: alerting.ElementTypeInput,
|
|
InputType: alerting.InputTypeText,
|
|
Placeholder: "topic1",
|
|
PropertyName: "kafkaTopic",
|
|
Required: true,
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
// NewKafkaNotifier is the constructor function for the Kafka notifier.
|
|
func NewKafkaNotifier(model *models.AlertNotification) (alerting.Notifier, error) {
|
|
endpoint := model.Settings.Get("kafkaRestProxy").MustString()
|
|
if endpoint == "" {
|
|
return nil, alerting.ValidationError{Reason: "Could not find kafka rest proxy endpoint property in settings"}
|
|
}
|
|
topic := model.Settings.Get("kafkaTopic").MustString()
|
|
if topic == "" {
|
|
return nil, alerting.ValidationError{Reason: "Could not find kafka topic property in settings"}
|
|
}
|
|
|
|
return &KafkaNotifier{
|
|
NotifierBase: NewNotifierBase(model),
|
|
Endpoint: endpoint,
|
|
Topic: topic,
|
|
log: log.New("alerting.notifier.kafka"),
|
|
}, nil
|
|
}
|
|
|
|
// KafkaNotifier is responsible for sending
|
|
// alert notifications to Kafka.
|
|
type KafkaNotifier struct {
|
|
NotifierBase
|
|
Endpoint string
|
|
Topic string
|
|
log log.Logger
|
|
}
|
|
|
|
// Notify sends the alert notification.
|
|
func (kn *KafkaNotifier) Notify(evalContext *alerting.EvalContext) error {
|
|
state := evalContext.Rule.State
|
|
|
|
customData := triggMetrString
|
|
for _, evt := range evalContext.EvalMatches {
|
|
customData += fmt.Sprintf("%s: %v\n", evt.Metric, evt.Value)
|
|
}
|
|
|
|
kn.log.Info("Notifying Kafka", "alert_state", state)
|
|
|
|
recordJSON := simplejson.New()
|
|
records := make([]interface{}, 1)
|
|
|
|
bodyJSON := simplejson.New()
|
|
// get alert state in the kafka output issue #11401
|
|
bodyJSON.Set("alert_state", state)
|
|
bodyJSON.Set("description", evalContext.Rule.Name+" - "+evalContext.Rule.Message)
|
|
bodyJSON.Set("client", "Grafana")
|
|
bodyJSON.Set("details", customData)
|
|
bodyJSON.Set("incident_key", "alertId-"+strconv.FormatInt(evalContext.Rule.ID, 10))
|
|
|
|
ruleURL, err := evalContext.GetRuleURL()
|
|
if err != nil {
|
|
kn.log.Error("Failed get rule link", "error", err)
|
|
return err
|
|
}
|
|
bodyJSON.Set("client_url", ruleURL)
|
|
|
|
if kn.NeedsImage() && evalContext.ImagePublicURL != "" {
|
|
contexts := make([]interface{}, 1)
|
|
imageJSON := simplejson.New()
|
|
imageJSON.Set("type", "image")
|
|
imageJSON.Set("src", evalContext.ImagePublicURL)
|
|
contexts[0] = imageJSON
|
|
bodyJSON.Set("contexts", contexts)
|
|
}
|
|
|
|
valueJSON := simplejson.New()
|
|
valueJSON.Set("value", bodyJSON)
|
|
records[0] = valueJSON
|
|
recordJSON.Set("records", records)
|
|
body, _ := recordJSON.MarshalJSON()
|
|
|
|
topicURL := kn.Endpoint + "/topics/" + kn.Topic
|
|
|
|
cmd := &models.SendWebhookSync{
|
|
Url: topicURL,
|
|
Body: string(body),
|
|
HttpMethod: "POST",
|
|
HttpHeader: map[string]string{
|
|
"Content-Type": "application/vnd.kafka.json.v2+json",
|
|
"Accept": "application/vnd.kafka.v2+json",
|
|
},
|
|
}
|
|
|
|
if err := bus.DispatchCtx(evalContext.Ctx, cmd); err != nil {
|
|
kn.log.Error("Failed to send notification to Kafka", "error", err, "body", string(body))
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|