energylink/pkg/models/alerts.go

package models

import (
	"time"
)

type AlertRule struct {
	Id           int64   `json:"id"`
	OrgId        int64   `json:"-"`
	DatasourceId int64   `json:"datasourceId"`
	DashboardId  int64   `json:"dashboardId"`
	PanelId      int64   `json:"panelId"`
	Query        string  `json:"query"`
	QueryRefId   string  `json:"queryRefId"`
	WarnLevel    float64 `json:"warnLevel"`
	CritLevel    float64 `json:"critLevel"`
	WarnOperator string  `json:"warnOperator"`
	CritOperator string  `json:"critOperator"`
	Frequency    int64   `json:"frequency"`
	Name         string  `json:"name"`
	Description  string  `json:"description"`
	QueryRange   int     `json:"queryRange"`
	Aggregator   string  `json:"aggregator"`
	State        string  `json:"state"`

	Created time.Time `json:"created"`
	Updated time.Time `json:"updated"`
}

func (alertRule *AlertRule) ValidToSave() bool {
	return alertRule.Query != "" && alertRule.Frequency != 0 && alertRule.QueryRange != 0 && alertRule.Name != ""
}

func (this *AlertRule) Equals(other *AlertRule) bool {
	result := false

	result = result || this.Aggregator != other.Aggregator
	result = result || this.CritLevel != other.CritLevel
	result = result || this.WarnLevel != other.WarnLevel
	result = result || this.WarnOperator != other.WarnOperator
	result = result || this.CritOperator != other.CritOperator
	result = result || this.Query != other.Query
	result = result || this.QueryRefId != other.QueryRefId
	result = result || this.Frequency != other.Frequency
	result = result || this.Name != other.Name
	result = result || this.Description != other.Description
	result = result || this.QueryRange != other.QueryRange
	//don't compare .State! That would be insane.

	return result
}

type AlertingClusterInfo struct {
	ServerId       string
	ClusterSize    int
	UptimePosition int
}

type HeartBeat struct {
	Id       int64
	ServerId string
	Updated  time.Time
	Created  time.Time
}

type HeartBeatCommand struct {
	ServerId string

	Result AlertingClusterInfo
}

type AlertRuleChange struct {
	Id      int64     `json:"id"`
	OrgId   int64     `json:"-"`
	AlertId int64     `json:"alertId"`
	Type    string    `json:"type"`
	Created time.Time `json:"created"`
}

// Commands
type SaveAlertsCommand struct {
	DashboardId int64
	UserId      int64
	OrgId       int64

	Alerts []*AlertRule
}

type DeleteAlertCommand struct {
	AlertId int64
}

//Queries
type GetAlertsQuery struct {
	OrgId       int64
	State       []string
	DashboardId int64
	PanelId     int64

	Result []*AlertRule
}

type GetAllAlertsQuery struct {
	Result []*AlertRule
}

type GetAlertsForExecutionQuery struct {
	Timestamp int64

	Result []*AlertRule
}

type GetAlertByIdQuery struct {
	Id int64

	Result *AlertRule
}

type GetAlertChangesQuery struct {
	OrgId   int64
	Limit   int64
	SinceId int64

	Result []*AlertRuleChange
}