energylink/pkg/services/sqlstore/alert_state.go

package sqlstore

import (
	"fmt"
	"time"

	"github.com/go-xorm/xorm"
	"github.com/grafana/grafana/pkg/bus"
	m "github.com/grafana/grafana/pkg/models"
)

func init() {
	bus.AddHandler("sql", SetNewAlertState)
	bus.AddHandler("sql", GetAlertStateLogByAlertId)
}

func SetNewAlertState(cmd *m.UpdateAlertStateCommand) error {
	return inTransaction(func(sess *xorm.Session) error {
		if !cmd.IsValidState() {
			return fmt.Errorf("new state is invalid")
		}

		alert := m.Alert{}
		has, err := sess.Id(cmd.AlertId).Get(&alert)
		if err != nil {
			return err
		}

		if !has {
			return fmt.Errorf("Could not find alert")
		}

		if alert.State == cmd.NewState {
			cmd.Result = &m.Alert{}
			return nil
		}

		alert.State = cmd.NewState
		sess.Id(alert.Id).Update(&alert)

		alertState := m.AlertState{
			AlertId:  cmd.AlertId,
			OrgId:    cmd.AlertId,
			NewState: cmd.NewState,
			Info:     cmd.Info,
			Created:  time.Now(),
		}

		sess.Insert(&alertState)

		cmd.Result = &alert
		return nil
	})
}

func GetAlertStateLogByAlertId(cmd *m.GetAlertsStateQuery) error {
	alertLogs := make([]m.AlertState, 0)

	if err := x.Where("alert_id = ?", cmd.AlertId).Desc("created").Find(&alertLogs); err != nil {
		return err
	}

	cmd.Result = &alertLogs
	return nil
}