feat: add alert firing since (#27)

freak12techno · Dec 11, 2023 · c75ff59 · c75ff59
1 parent 0b7871f
commit c75ff59
Show file tree

Hide file tree

Showing 5 changed files with 67 additions and 16 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,3 +3,4 @@ main
 .idea
 dist/
 grafana-interacter
+**/.DS_Store
diff --git a/pkg/templates/templates.go b/pkg/templates/templates.go
@@ -31,6 +31,8 @@ func (manager *TemplateManager) GetTemplate(name string) (*template.Template, er
 		"GetEmojiByStatus":        utils.GetEmojiByStatus,
 		"GetEmojiBySilenceStatus": utils.GetEmojiBySilenceStatus,
 		"StrToFloat64":            utils.StrToFloat64,
+		"FormatDuration":          utils.FormatDuration,
+		"FormatDate":              utils.FormatDate,
 	}).ParseFS(templatesList.Templates, filename)
 	if err != nil {
 		return nil, err

diff --git a/pkg/types/alert.go b/pkg/types/alert.go
@@ -3,6 +3,7 @@ package types
 import (
 	"main/pkg/utils/normalize"
 	"strings"
+	"time"
 )
 
 type GrafanaAlertRulesResponse struct {
@@ -26,9 +27,14 @@ type GrafanaAlertRule struct {
 }
 
 type GrafanaAlert struct {
-	Labels map[string]string `json:"labels"`
-	State  string            `json:"state"`
-	Value  string            `json:"value"`
+	Labels   map[string]string `json:"labels"`
+	State    string            `json:"state"`
+	Value    string            `json:"value"`
+	ActiveAt time.Time         `json:"activeAt"`
+}
+
+func (a GrafanaAlert) ActiveSince() time.Duration {
+	return time.Since(a.ActiveAt)
 }
 
 type GrafanaAlertGroups []GrafanaAlertGroup

diff --git a/pkg/utils/utils.go b/pkg/utils/utils.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"main/pkg/logger"
 	"main/pkg/types"
+	"math"
 	"strconv"
 	"strings"
 	"time"
@@ -192,3 +193,39 @@ func StrToFloat64(s string) float64 {
 
 	return f
 }
+
+func FormatDuration(duration time.Duration) string {
+	days := int64(duration.Hours() / 24)
+	hours := int64(math.Mod(duration.Hours(), 24))
+	minutes := int64(math.Mod(duration.Minutes(), 60))
+	seconds := int64(math.Mod(duration.Seconds(), 60))
+
+	chunks := []struct {
+		singularName string
+		amount       int64
+	}{
+		{"day", days},
+		{"hour", hours},
+		{"minute", minutes},
+		{"second", seconds},
+	}
+
+	parts := []string{}
+
+	for _, chunk := range chunks {
+		switch chunk.amount {
+		case 0:
+			continue
+		case 1:
+			parts = append(parts, fmt.Sprintf("%d %s", chunk.amount, chunk.singularName))
+		default:
+			parts = append(parts, fmt.Sprintf("%d %ss", chunk.amount, chunk.singularName))
+		}
+	}
+
+	return strings.Join(parts, " ")
+}
+
+func FormatDate(date time.Time) string {
+	return date.Format(time.RFC1123)
+}
diff --git a/templates/alerts_firing.html b/templates/alerts_firing.html
@@ -1,6 +1,9 @@
-<strong>Grafana alerts:</strong>
-{{- if not .Data.GrafanaGroups }}
-No Grafana alerts
+{{- if and (not .Data.GrafanaGroups) (not .Data.PrometheusGroups) }}
+No firing alerts.
+{{ end }}
+
+{{- if .Data.GrafanaGroups }}
+<strong>Grafana alerts ({{ len .Data.GrafanaGroups }}):</strong>
 {{- end }}
 {{- range $groupId, $group := .Data.GrafanaGroups }}
 {{- range $ruleId, $rule := $group.Rules }}
@@ -14,26 +17,28 @@
 {{- if $alert.Value }}
 value = {{ StrToFloat64 $alert.Value }}
 {{- end }}
-{{ end }}
+firing for {{ FormatDuration $alert.ActiveSince }} (since {{ FormatDate $alert.ActiveAt }})
 {{- end }}
 {{- end }}
+{{ end }}
 
 
-<strong>Prometheus alerts:</strong>
-{{- if not .Data.PrometheusGroups }}
-No Prometheus alerts
+{{- if .Data.PrometheusGroups }}
+<strong>Prometheus alerts ({{ len .Data.PrometheusGroups }}):</strong>
 {{- end }}
-{{- range $groupId, $group := .Data.PrometheusGroups }}
+{{ range $groupId, $group := .Data.PrometheusGroups }}
 {{- range $ruleId, $rule := $group.Rules }}
 - {{ GetEmojiByStatus $rule.State }} {{ $group.Name }} -> {{ $rule.Name }} ({{ len ($rule.Alerts) }}):
-{{- range $alertId, $alert := $rule.Alerts }}
+{{ range $alertId, $alert := $rule.Alerts }}
+<strong>Firing for:</strong> {{ FormatDuration $alert.ActiveSince }} (since {{ FormatDate $alert.ActiveAt }})
+{{- if $alert.Value }}
+<strong>Value: </strong>{{ StrToFloat64 $alert.Value }}
+{{- end }}
+<strong>Labels: </strong>
 {{- range $key, $label := $alert.Labels }}
 {{- if ne $key "alertname" }}
-{{ $key }} = {{ $label }}
-{{- end }}
+  {{ $key }} = {{ $label }}
 {{- end }}
-{{- if $alert.Value }}
-value = {{ StrToFloat64 $alert.Value }}
 {{- end }}
 {{ end }}
 {{- end }}