mirror of
https://github.com/grafana/grafana.git
synced 2025-02-14 17:43:35 -06:00
122 lines
3.2 KiB
Go
122 lines
3.2 KiB
Go
package notifiers
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/log"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/alerting"
|
|
)
|
|
|
|
const (
|
|
triggMetrString = "Triggered metrics:\n\n"
|
|
)
|
|
|
|
type NotifierBase struct {
|
|
Name string
|
|
Type string
|
|
Id int64
|
|
IsDeault bool
|
|
UploadImage bool
|
|
SendReminder bool
|
|
DisableResolveMessage bool
|
|
Frequency time.Duration
|
|
|
|
log log.Logger
|
|
}
|
|
|
|
func NewNotifierBase(model *models.AlertNotification) NotifierBase {
|
|
uploadImage := true
|
|
value, exist := model.Settings.CheckGet("uploadImage")
|
|
if exist {
|
|
uploadImage = value.MustBool()
|
|
}
|
|
|
|
return NotifierBase{
|
|
Id: model.Id,
|
|
Name: model.Name,
|
|
IsDeault: model.IsDefault,
|
|
Type: model.Type,
|
|
UploadImage: uploadImage,
|
|
SendReminder: model.SendReminder,
|
|
DisableResolveMessage: model.DisableResolveMessage,
|
|
Frequency: model.Frequency,
|
|
log: log.New("alerting.notifier." + model.Name),
|
|
}
|
|
}
|
|
|
|
// ShouldNotify checks this evaluation should send an alert notification
|
|
func (n *NotifierBase) ShouldNotify(ctx context.Context, context *alerting.EvalContext, notiferState *models.AlertNotificationState) bool {
|
|
// Only notify on state change.
|
|
if context.PrevAlertState == context.Rule.State && !n.SendReminder {
|
|
return false
|
|
}
|
|
|
|
if context.PrevAlertState == context.Rule.State && n.SendReminder {
|
|
// Do not notify if interval has not elapsed
|
|
lastNotify := time.Unix(notiferState.UpdatedAt, 0)
|
|
if notiferState.UpdatedAt != 0 && lastNotify.Add(n.Frequency).After(time.Now()) {
|
|
return false
|
|
}
|
|
|
|
// Do not notify if alert state is OK or pending even on repeated notify
|
|
if context.Rule.State == models.AlertStateOK || context.Rule.State == models.AlertStatePending {
|
|
return false
|
|
}
|
|
}
|
|
|
|
// Do not notify when we become OK for the first time.
|
|
if context.PrevAlertState == models.AlertStatePending && context.Rule.State == models.AlertStateOK {
|
|
return false
|
|
}
|
|
|
|
// Do not notify when we OK -> Pending
|
|
if context.PrevAlertState == models.AlertStateOK && context.Rule.State == models.AlertStatePending {
|
|
return false
|
|
}
|
|
|
|
// Do not notifu if state pending and it have been updated last minute
|
|
if notiferState.State == models.AlertNotificationStatePending {
|
|
lastUpdated := time.Unix(notiferState.UpdatedAt, 0)
|
|
if lastUpdated.Add(1 * time.Minute).After(time.Now()) {
|
|
return false
|
|
}
|
|
}
|
|
|
|
// Do not notify when state is OK if DisableResolveMessage is set to true
|
|
if context.Rule.State == models.AlertStateOK && n.DisableResolveMessage {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func (n *NotifierBase) GetType() string {
|
|
return n.Type
|
|
}
|
|
|
|
func (n *NotifierBase) NeedsImage() bool {
|
|
return n.UploadImage
|
|
}
|
|
|
|
func (n *NotifierBase) GetNotifierId() int64 {
|
|
return n.Id
|
|
}
|
|
|
|
func (n *NotifierBase) GetIsDefault() bool {
|
|
return n.IsDeault
|
|
}
|
|
|
|
func (n *NotifierBase) GetSendReminder() bool {
|
|
return n.SendReminder
|
|
}
|
|
|
|
func (n *NotifierBase) GetDisableResolveMessage() bool {
|
|
return n.DisableResolveMessage
|
|
}
|
|
|
|
func (n *NotifierBase) GetFrequency() time.Duration {
|
|
return n.Frequency
|
|
}
|