mirror of
https://github.com/grafana/grafana.git
synced 2025-09-21 08:32:51 +08:00
90646e7f41
* Alerting: Don't stop the migration when alert rule tags are invalid As we migrate we expect the `alertRuleTags` on a dashboard alert to be a JSON object. However, it seems this is not really validated by Grafana and an user can change the format to something else that the JSON parser is not able to marshal into a `map[string]string`. Let's do a bit better by "attempting" to parse the tags and if we can't we'll simple return an empty map. The data is still there so if the user wishes they can go back, fix the data and attemp the migration again.
130 lines
3.1 KiB
Go
130 lines
3.1 KiB
Go
package ualert
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
type dashAlert struct {
|
|
Id int64
|
|
OrgId int64
|
|
DashboardId int64
|
|
PanelId int64
|
|
Name string
|
|
Message string
|
|
Frequency int64
|
|
For time.Duration
|
|
State string
|
|
|
|
Settings json.RawMessage
|
|
ParsedSettings *dashAlertSettings
|
|
DashboardUID string // Set from separate call
|
|
}
|
|
|
|
var slurpDashSQL = `
|
|
SELECT id,
|
|
org_id,
|
|
dashboard_id,
|
|
panel_id,
|
|
org_id,
|
|
name,
|
|
message,
|
|
frequency,
|
|
%s,
|
|
state,
|
|
settings
|
|
FROM
|
|
alert
|
|
WHERE org_id IN (SELECT id from org)
|
|
AND dashboard_id IN (SELECT id from dashboard)
|
|
`
|
|
|
|
// slurpDashAlerts loads all alerts from the alert database table into
|
|
// the dashAlert type. If there are alerts that belong to either organization or dashboard that does not exist, those alerts will not be returned/
|
|
// Additionally it unmarshals the json settings for the alert into the
|
|
// ParsedSettings property of the dash alert.
|
|
func (m *migration) slurpDashAlerts() ([]dashAlert, error) {
|
|
dashAlerts := []dashAlert{}
|
|
err := m.sess.SQL(fmt.Sprintf(slurpDashSQL, m.mg.Dialect.Quote("for"))).Find(&dashAlerts)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for i := range dashAlerts {
|
|
err = json.Unmarshal(dashAlerts[i].Settings, &dashAlerts[i].ParsedSettings)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return dashAlerts, nil
|
|
}
|
|
|
|
// dashAlertSettings is a type for the JSON that is in the settings field of
|
|
// the alert table.
|
|
type dashAlertSettings struct {
|
|
NoDataState string `json:"noDataState"`
|
|
ExecutionErrorState string `json:"executionErrorState"`
|
|
Conditions []dashAlertCondition `json:"conditions"`
|
|
AlertRuleTags interface{} `json:"alertRuleTags"`
|
|
Notifications []dashAlertNot `json:"notifications"`
|
|
}
|
|
|
|
// dashAlertNot is the object that represents the Notifications array in
|
|
// dashAlertSettings
|
|
type dashAlertNot struct {
|
|
UID string `json:"uid,omitempty"`
|
|
ID int64 `json:"id,omitempty"`
|
|
}
|
|
|
|
// dashAlertingConditionJSON is like classic.ClassicConditionJSON except that it
|
|
// includes the model property with the query.
|
|
type dashAlertCondition struct {
|
|
Evaluator conditionEvalJSON `json:"evaluator"`
|
|
|
|
Operator struct {
|
|
Type string `json:"type"`
|
|
} `json:"operator"`
|
|
|
|
Query struct {
|
|
Params []string `json:"params"`
|
|
DatasourceID int64 `json:"datasourceId"`
|
|
Model json.RawMessage
|
|
} `json:"query"`
|
|
|
|
Reducer struct {
|
|
// Params []interface{} `json:"params"` (Unused)
|
|
Type string `json:"type"`
|
|
}
|
|
}
|
|
|
|
type conditionEvalJSON struct {
|
|
Params []float64 `json:"params"`
|
|
Type string `json:"type"` // e.g. "gt"
|
|
}
|
|
|
|
// slurpDashUIDs returns a map of [orgID, dashboardId] -> dashUID.
|
|
func (m *migration) slurpDashUIDs() (map[[2]int64]string, error) {
|
|
dashIDs := []struct {
|
|
OrgID int64 `xorm:"org_id"`
|
|
ID int64 `xorm:"id"`
|
|
UID string `xorm:"uid"`
|
|
}{}
|
|
|
|
err := m.sess.SQL(`SELECT org_id, id, uid FROM dashboard`).Find(&dashIDs)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
idToUID := make(map[[2]int64]string, len(dashIDs))
|
|
|
|
for _, ds := range dashIDs {
|
|
idToUID[[2]int64{ds.OrgID, ds.ID}] = ds.UID
|
|
}
|
|
|
|
return idToUID, nil
|
|
}
|