|
@@ -5,10 +5,12 @@ import (
|
|
|
"github.com/go-xorm/xorm"
|
|
"github.com/go-xorm/xorm"
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
|
|
|
+ "time"
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
|
func init() {
|
|
|
bus.AddHandler("sql", SetNewAlertState)
|
|
bus.AddHandler("sql", SetNewAlertState)
|
|
|
|
|
+ bus.AddHandler("sql", GetAlertStateLogByAlertId)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func SetNewAlertState(cmd *m.UpdateAlertStateCommand) error {
|
|
func SetNewAlertState(cmd *m.UpdateAlertStateCommand) error {
|
|
@@ -29,11 +31,29 @@ func SetNewAlertState(cmd *m.UpdateAlertStateCommand) error {
|
|
|
|
|
|
|
|
alert.State = cmd.NewState
|
|
alert.State = cmd.NewState
|
|
|
sess.Id(alert.Id).Update(&alert)
|
|
sess.Id(alert.Id).Update(&alert)
|
|
|
- //update alert
|
|
|
|
|
|
|
|
|
|
- //insert alert state log
|
|
|
|
|
|
|
+ log := m.AlertStateLog{
|
|
|
|
|
+ AlertId: cmd.AlertId,
|
|
|
|
|
+ OrgId: cmd.AlertId,
|
|
|
|
|
+ NewState: cmd.NewState,
|
|
|
|
|
+ Info: cmd.Info,
|
|
|
|
|
+ Created: time.Now(),
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ sess.Insert(&log)
|
|
|
|
|
|
|
|
cmd.Result = &alert
|
|
cmd.Result = &alert
|
|
|
return nil
|
|
return nil
|
|
|
})
|
|
})
|
|
|
}
|
|
}
|
|
|
|
|
+
|
|
|
|
|
+func GetAlertStateLogByAlertId(cmd *m.GetAlertsStateLogCommand) error {
|
|
|
|
|
+ alertLogs := make([]m.AlertStateLog, 0)
|
|
|
|
|
+
|
|
|
|
|
+ if err := x.Where("alert_id = ?", cmd.AlertId).Find(&alertLogs); err != nil {
|
|
|
|
|
+ return err
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ cmd.Result = &alertLogs
|
|
|
|
|
+ return nil
|
|
|
|
|
+}
|