feat(alerting): add datasource ref to alert rule

2016-05-25 12:57:49 +02:00
parent 0f58f8a679
commit b2a4d8083e
8 changed files with 122 additions and 61 deletions
@@ -0,0 +1,80 @@
+package alerting
+
+import (
+	"github.com/grafana/grafana/pkg/bus"
+	"github.com/grafana/grafana/pkg/components/simplejson"
+	"github.com/grafana/grafana/pkg/log"
+	m "github.com/grafana/grafana/pkg/models"
+)
+
+func ParseAlertsFromDashboard(cmd *m.SaveDashboardCommand) []m.AlertRule {
+	alerts := make([]m.AlertRule, 0)
+
+	for _, rowObj := range cmd.Dashboard.Get("rows").MustArray() {
+		row := simplejson.NewFromAny(rowObj)
+
+		for _, panelObj := range row.Get("panels").MustArray() {
+			panel := simplejson.NewFromAny(panelObj)
+
+			alerting := panel.Get("alerting")
+			alert := m.AlertRule{
+				DashboardId:  cmd.Result.Id,
+				OrgId:        cmd.Result.OrgId,
+				PanelId:      panel.Get("id").MustInt64(),
+				Id:           alerting.Get("id").MustInt64(),
+				QueryRefId:   alerting.Get("queryRef").MustString(),
+				WarnLevel:    alerting.Get("warnLevel").MustInt64(),
+				CritLevel:    alerting.Get("critLevel").MustInt64(),
+				WarnOperator: alerting.Get("warnOperator").MustString(),
+				CritOperator: alerting.Get("critOperator").MustString(),
+				Interval:     alerting.Get("interval").MustString(),
+				Title:        alerting.Get("title").MustString(),
+				Description:  alerting.Get("description").MustString(),
+				QueryRange:   alerting.Get("queryRange").MustString(),
+				Aggregator:   alerting.Get("aggregator").MustString(),
+			}
+
+			for _, targetsObj := range panel.Get("targets").MustArray() {
+				target := simplejson.NewFromAny(targetsObj)
+
+				if target.Get("refId").MustString() == alert.QueryRefId {
+					targetJson, err := target.MarshalJSON()
+					if err == nil {
+						alert.Query = string(targetJson)
+					}
+					continue
+				}
+			}
+
+			log.Info("datasource is %s", panel.Get("datasource").MustString())
+			log.Info("is datasource null? %v", panel.Get("datasource").MustString() == "")
+			if panel.Get("datasource").MustString() == "" {
+
+				query := &m.GetDataSourcesQuery{OrgId: cmd.OrgId}
+				if err := bus.Dispatch(query); err == nil {
+
+					for _, ds := range query.Result {
+						log.Info("found datasource %s", ds.Name)
+						if ds.IsDefault {
+							alert.DatasourceId = ds.Id
+							log.Info("setting default datasource! %d", ds.Id)
+						}
+					}
+				}
+			} else {
+				query := &m.GetDataSourceByNameQuery{
+					Name:  panel.Get("datasource").MustString(),
+					OrgId: cmd.OrgId,
+				}
+				bus.Dispatch(query)
+				alert.DatasourceId = query.Result.Id
+			}
+
+			if alert.Query != "" {
+				alerts = append(alerts, alert)
+			}
+		}
+	}
+
+	return alerts
+}