2016-04-13 16:33:45 +08:00
|
|
|
package models
|
|
|
|
|
|
|
|
import (
|
2016-12-19 20:24:45 +08:00
|
|
|
"fmt"
|
2019-04-23 16:24:47 +08:00
|
|
|
"time"
|
2016-12-19 20:24:45 +08:00
|
|
|
|
2016-06-10 04:21:28 +08:00
|
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
2016-04-13 16:33:45 +08:00
|
|
|
)
|
|
|
|
|
2016-07-22 19:14:09 +08:00
|
|
|
type AlertStateType string
|
|
|
|
type AlertSeverityType string
|
2016-10-22 16:56:04 +08:00
|
|
|
type NoDataOption string
|
2016-11-07 19:42:39 +08:00
|
|
|
type ExecutionErrorOption string
|
2016-07-22 19:14:09 +08:00
|
|
|
|
|
|
|
const (
|
2016-11-07 19:42:39 +08:00
|
|
|
AlertStateNoData AlertStateType = "no_data"
|
|
|
|
AlertStatePaused AlertStateType = "paused"
|
|
|
|
AlertStateAlerting AlertStateType = "alerting"
|
|
|
|
AlertStateOK AlertStateType = "ok"
|
|
|
|
AlertStatePending AlertStateType = "pending"
|
2018-11-05 17:23:43 +08:00
|
|
|
AlertStateUnknown AlertStateType = "unknown"
|
2016-07-22 19:14:09 +08:00
|
|
|
)
|
|
|
|
|
2016-10-22 16:56:04 +08:00
|
|
|
const (
|
2016-12-15 18:13:19 +08:00
|
|
|
NoDataSetOK NoDataOption = "ok"
|
2016-10-22 16:56:04 +08:00
|
|
|
NoDataSetNoData NoDataOption = "no_data"
|
|
|
|
NoDataKeepState NoDataOption = "keep_state"
|
2016-12-15 18:13:19 +08:00
|
|
|
NoDataSetAlerting NoDataOption = "alerting"
|
2016-10-22 16:56:04 +08:00
|
|
|
)
|
|
|
|
|
2016-11-07 19:42:39 +08:00
|
|
|
const (
|
|
|
|
ExecutionErrorSetAlerting ExecutionErrorOption = "alerting"
|
|
|
|
ExecutionErrorKeepState ExecutionErrorOption = "keep_state"
|
|
|
|
)
|
|
|
|
|
2016-12-19 20:24:45 +08:00
|
|
|
var (
|
2018-04-28 04:14:36 +08:00
|
|
|
ErrCannotChangeStateOnPausedAlert = fmt.Errorf("Cannot change state on pause alert")
|
2019-04-23 16:24:47 +08:00
|
|
|
ErrRequiresNewState = fmt.Errorf("update alert state requires a new state")
|
2016-12-19 20:24:45 +08:00
|
|
|
)
|
|
|
|
|
2016-07-22 19:14:09 +08:00
|
|
|
func (s AlertStateType) IsValid() bool {
|
2018-11-05 17:23:43 +08:00
|
|
|
return s == AlertStateOK ||
|
|
|
|
s == AlertStateNoData ||
|
|
|
|
s == AlertStatePaused ||
|
|
|
|
s == AlertStatePending ||
|
|
|
|
s == AlertStateAlerting ||
|
|
|
|
s == AlertStateUnknown
|
2016-07-22 19:14:09 +08:00
|
|
|
}
|
|
|
|
|
2016-10-22 16:56:04 +08:00
|
|
|
func (s NoDataOption) IsValid() bool {
|
2016-12-15 18:13:19 +08:00
|
|
|
return s == NoDataSetNoData || s == NoDataSetAlerting || s == NoDataKeepState || s == NoDataSetOK
|
2016-10-22 16:56:04 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s NoDataOption) ToAlertState() AlertStateType {
|
|
|
|
return AlertStateType(s)
|
|
|
|
}
|
|
|
|
|
2016-11-07 19:42:39 +08:00
|
|
|
func (s ExecutionErrorOption) IsValid() bool {
|
|
|
|
return s == ExecutionErrorSetAlerting || s == ExecutionErrorKeepState
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s ExecutionErrorOption) ToAlertState() AlertStateType {
|
|
|
|
return AlertStateType(s)
|
|
|
|
}
|
|
|
|
|
2016-06-11 16:26:48 +08:00
|
|
|
type Alert struct {
|
2016-08-15 21:12:43 +08:00
|
|
|
Id int64
|
2016-08-16 15:52:45 +08:00
|
|
|
Version int64
|
2016-08-15 21:12:43 +08:00
|
|
|
OrgId int64
|
|
|
|
DashboardId int64
|
|
|
|
PanelId int64
|
|
|
|
Name string
|
|
|
|
Message string
|
2018-11-05 18:05:30 +08:00
|
|
|
Severity string //Unused
|
2016-08-15 21:12:43 +08:00
|
|
|
State AlertStateType
|
2018-11-05 18:05:30 +08:00
|
|
|
Handler int64 //Unused
|
2016-08-15 21:12:43 +08:00
|
|
|
Silenced bool
|
|
|
|
ExecutionError string
|
|
|
|
Frequency int64
|
2018-11-05 18:05:30 +08:00
|
|
|
For time.Duration
|
2016-08-15 21:12:43 +08:00
|
|
|
|
2016-08-16 15:52:45 +08:00
|
|
|
EvalData *simplejson.Json
|
|
|
|
NewStateDate time.Time
|
2018-10-01 20:13:03 +08:00
|
|
|
StateChanges int64
|
2016-07-14 19:32:16 +08:00
|
|
|
|
2016-06-10 04:21:28 +08:00
|
|
|
Created time.Time
|
|
|
|
Updated time.Time
|
|
|
|
|
2016-06-13 21:18:19 +08:00
|
|
|
Settings *simplejson.Json
|
2016-04-13 16:33:45 +08:00
|
|
|
}
|
|
|
|
|
2016-06-11 16:26:48 +08:00
|
|
|
func (alert *Alert) ValidToSave() bool {
|
2016-06-13 21:01:07 +08:00
|
|
|
return alert.DashboardId != 0 && alert.OrgId != 0 && alert.PanelId != 0
|
2016-06-10 16:00:00 +08:00
|
|
|
}
|
|
|
|
|
2016-07-22 19:14:09 +08:00
|
|
|
func (alert *Alert) ShouldUpdateState(newState AlertStateType) bool {
|
2016-06-15 15:19:22 +08:00
|
|
|
return alert.State != newState
|
|
|
|
}
|
|
|
|
|
2016-06-11 16:26:48 +08:00
|
|
|
func (this *Alert) ContainsUpdates(other *Alert) bool {
|
2016-05-30 23:50:35 +08:00
|
|
|
result := false
|
2016-06-06 23:11:46 +08:00
|
|
|
result = result || this.Name != other.Name
|
2016-08-12 16:12:04 +08:00
|
|
|
result = result || this.Message != other.Message
|
2016-06-10 04:21:28 +08:00
|
|
|
|
2016-06-13 21:18:19 +08:00
|
|
|
if this.Settings != nil && other.Settings != nil {
|
|
|
|
json1, err1 := this.Settings.Encode()
|
|
|
|
json2, err2 := other.Settings.Encode()
|
2016-06-10 04:21:28 +08:00
|
|
|
|
2016-06-10 17:37:03 +08:00
|
|
|
if err1 != nil || err2 != nil {
|
|
|
|
return false
|
|
|
|
}
|
2016-06-10 04:21:28 +08:00
|
|
|
|
2016-06-10 17:37:03 +08:00
|
|
|
result = result || string(json1) != string(json2)
|
|
|
|
}
|
2016-06-10 04:21:28 +08:00
|
|
|
|
2016-05-30 23:50:35 +08:00
|
|
|
//don't compare .State! That would be insane.
|
|
|
|
return result
|
|
|
|
}
|
|
|
|
|
2019-06-06 19:29:30 +08:00
|
|
|
func (alert *Alert) GetTagsFromSettings() []*Tag {
|
|
|
|
tags := []*Tag{}
|
|
|
|
if alert.Settings != nil {
|
|
|
|
if data, ok := alert.Settings.CheckGet("alertRuleTags"); ok {
|
|
|
|
for tagNameString, tagValue := range data.MustMap() {
|
|
|
|
// MustMap() already guarantees the return of a `map[string]interface{}`.
|
|
|
|
// Therefore we only need to verify that tagValue is a String.
|
|
|
|
tagValueString := simplejson.NewFromAny(tagValue).MustString()
|
|
|
|
tags = append(tags, &Tag{Key: tagNameString, Value: tagValueString})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return tags
|
|
|
|
}
|
|
|
|
|
2016-05-23 20:51:41 +08:00
|
|
|
type AlertingClusterInfo struct {
|
|
|
|
ServerId string
|
|
|
|
ClusterSize int
|
|
|
|
UptimePosition int
|
|
|
|
}
|
|
|
|
|
2016-06-09 16:00:34 +08:00
|
|
|
type HeartBeat struct {
|
|
|
|
Id int64
|
|
|
|
ServerId string
|
|
|
|
Updated time.Time
|
|
|
|
Created time.Time
|
|
|
|
}
|
|
|
|
|
2016-05-23 20:51:41 +08:00
|
|
|
type HeartBeatCommand struct {
|
2016-05-23 16:02:17 +08:00
|
|
|
ServerId string
|
2016-06-11 16:26:48 +08:00
|
|
|
Result AlertingClusterInfo
|
2016-05-23 16:02:17 +08:00
|
|
|
}
|
|
|
|
|
2016-04-13 16:33:45 +08:00
|
|
|
type SaveAlertsCommand struct {
|
|
|
|
DashboardId int64
|
|
|
|
UserId int64
|
|
|
|
OrgId int64
|
|
|
|
|
2016-06-11 16:26:48 +08:00
|
|
|
Alerts []*Alert
|
2016-04-13 16:33:45 +08:00
|
|
|
}
|
2016-04-26 23:36:50 +08:00
|
|
|
|
2016-10-10 20:26:09 +08:00
|
|
|
type PauseAlertCommand struct {
|
2016-12-19 23:07:55 +08:00
|
|
|
OrgId int64
|
|
|
|
AlertIds []int64
|
|
|
|
ResultCount int64
|
|
|
|
Paused bool
|
2016-10-10 20:26:09 +08:00
|
|
|
}
|
|
|
|
|
2016-12-19 20:24:45 +08:00
|
|
|
type PauseAllAlertCommand struct {
|
|
|
|
ResultCount int64
|
|
|
|
Paused bool
|
|
|
|
}
|
|
|
|
|
2016-07-22 19:14:09 +08:00
|
|
|
type SetAlertStateCommand struct {
|
2016-09-14 20:35:05 +08:00
|
|
|
AlertId int64
|
|
|
|
OrgId int64
|
|
|
|
State AlertStateType
|
|
|
|
Error string
|
|
|
|
EvalData *simplejson.Json
|
|
|
|
|
2018-10-01 20:13:03 +08:00
|
|
|
Result Alert
|
2016-07-22 19:14:09 +08:00
|
|
|
}
|
|
|
|
|
2016-04-26 23:36:50 +08:00
|
|
|
//Queries
|
|
|
|
type GetAlertsQuery struct {
|
2018-06-01 20:36:40 +08:00
|
|
|
OrgId int64
|
|
|
|
State []string
|
|
|
|
DashboardIDs []int64
|
|
|
|
PanelId int64
|
|
|
|
Limit int64
|
|
|
|
Query string
|
|
|
|
User *SignedInUser
|
2016-04-26 23:36:50 +08:00
|
|
|
|
2018-02-16 20:56:04 +08:00
|
|
|
Result []*AlertListItemDTO
|
2016-04-26 23:36:50 +08:00
|
|
|
}
|
|
|
|
|
2016-06-03 21:24:53 +08:00
|
|
|
type GetAllAlertsQuery struct {
|
2016-06-11 16:26:48 +08:00
|
|
|
Result []*Alert
|
2016-05-16 21:39:09 +08:00
|
|
|
}
|
|
|
|
|
2016-04-28 14:23:50 +08:00
|
|
|
type GetAlertByIdQuery struct {
|
2016-04-26 23:36:50 +08:00
|
|
|
Id int64
|
|
|
|
|
2016-06-11 16:26:48 +08:00
|
|
|
Result *Alert
|
2016-04-26 23:36:50 +08:00
|
|
|
}
|
2016-09-30 23:37:47 +08:00
|
|
|
|
|
|
|
type GetAlertStatesForDashboardQuery struct {
|
|
|
|
OrgId int64
|
|
|
|
DashboardId int64
|
|
|
|
|
|
|
|
Result []*AlertStateInfoDTO
|
|
|
|
}
|
|
|
|
|
2018-02-16 20:56:04 +08:00
|
|
|
type AlertListItemDTO struct {
|
|
|
|
Id int64 `json:"id"`
|
|
|
|
DashboardId int64 `json:"dashboardId"`
|
|
|
|
DashboardUid string `json:"dashboardUid"`
|
|
|
|
DashboardSlug string `json:"dashboardSlug"`
|
|
|
|
PanelId int64 `json:"panelId"`
|
|
|
|
Name string `json:"name"`
|
|
|
|
State AlertStateType `json:"state"`
|
|
|
|
NewStateDate time.Time `json:"newStateDate"`
|
|
|
|
EvalDate time.Time `json:"evalDate"`
|
|
|
|
EvalData *simplejson.Json `json:"evalData"`
|
|
|
|
ExecutionError string `json:"executionError"`
|
|
|
|
Url string `json:"url"`
|
|
|
|
}
|
|
|
|
|
2016-09-30 23:37:47 +08:00
|
|
|
type AlertStateInfoDTO struct {
|
|
|
|
Id int64 `json:"id"`
|
|
|
|
DashboardId int64 `json:"dashboardId"`
|
|
|
|
PanelId int64 `json:"panelId"`
|
|
|
|
State AlertStateType `json:"state"`
|
|
|
|
NewStateDate time.Time `json:"newStateDate"`
|
|
|
|
}
|
2018-02-19 18:12:56 +08:00
|
|
|
|
|
|
|
// "Internal" commands
|
|
|
|
|
|
|
|
type UpdateDashboardAlertsCommand struct {
|
|
|
|
OrgId int64
|
|
|
|
Dashboard *Dashboard
|
2018-11-05 21:25:19 +08:00
|
|
|
User *SignedInUser
|
2018-02-19 18:12:56 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
type ValidateDashboardAlertsCommand struct {
|
|
|
|
UserId int64
|
|
|
|
OrgId int64
|
|
|
|
Dashboard *Dashboard
|
2018-11-05 21:25:19 +08:00
|
|
|
User *SignedInUser
|
2018-02-19 18:12:56 +08:00
|
|
|
}
|