DEV-45327 - Add Logzio Account Id to alert annotations (#25)

*What is this feature?*
For Logzio alert notifications, we add as part of the PostableAlert annotations the LogzioAccountId so we can use it on the notification templates for generating urls to the Logzio app

*Why do we need this feature?*
So url generated in alert notifications can redirect to the relevant account

pkg/services/ngalert/models/alert_rule.go

@@ -138,6 +138,8 @@ const (
 	// AutogeneratedRouteSettingsHashLabel a label name that contains the hash of the notification settings that will be used to send notifications for the alert.
 	// This should uniquely identify the notification settings (group_by, group_wait, group_interval, repeat_interval, mute_time_intervals) for the alert.
 	AutogeneratedRouteSettingsHashLabel = "__grafana_route_settings_hash__"
+
+	LogzioAccountIdAnnotation = "__logzioAccountId__" // LOGZ.IO GRAFANA CHANGE :: DEV-45327: Add switch to account query param
 )
 
 const (

pkg/services/ngalert/schedule/schedule.go

@@ -505,7 +505,7 @@ func (sch *schedule) ruleRoutine(grafanaCtx context.Context, key ngmodels.AlertR
 		processDuration.Observe(sch.clock.Now().Sub(start).Seconds())
 
 		start = sch.clock.Now()
-		alerts := state.FromStateTransitionToPostableAlerts(processedStates, sch.stateManager, sch.appURL)
+		alerts := state.FromStateTransitionToPostableAlerts(processedStates, sch.stateManager, sch.appURL, e.logzHeaders) // LOGZ.IO GRAFANA CHANGE :: DEV-45327: Add switch to account query param
 		span.AddEvent("results processed", trace.WithAttributes(
 			attribute.Int64("state_transitions", int64(len(processedStates))),
 			attribute.Int64("alerts_to_send", int64(len(alerts.PostableAlerts))),

pkg/services/ngalert/state/compat.go

@@ -3,6 +3,7 @@ package state
 import (
 	"encoding/json"
 	"fmt"
+	"net/http" // LOGZ.IO GRAFANA CHANGE :: DEV-45327: Add switch to account query param
 	"net/url"
 	"path"
 	"strconv"
@@ -139,7 +140,7 @@ func errorAlert(labels, annotations data.Labels, alertState *State, urlStr strin
 	}
 }
 
-func FromStateTransitionToPostableAlerts(firingStates []StateTransition, stateManager *Manager, appURL *url.URL) apimodels.PostableAlerts {
+func FromStateTransitionToPostableAlerts(firingStates []StateTransition, stateManager *Manager, appURL *url.URL, logzHeaders http.Header) apimodels.PostableAlerts { // LOGZ.IO GRAFANA CHANGE :: DEV-45327: Add switch to account query param
 	alerts := apimodels.PostableAlerts{PostableAlerts: make([]models.PostableAlert, 0, len(firingStates))}
 	var sentAlerts []*State
 	ts := time.Now()
@@ -149,6 +150,14 @@ func FromStateTransitionToPostableAlerts(firingStates []StateTransition, stateMa
 			continue
 		}
 		alert := StateToPostableAlert(alertState, appURL)
+		// LOGZ.IO GRAFANA CHANGE :: DEV-45327: Add switch to account query param
+		if logzHeaders != nil {
+			logzAccountId := logzHeaders.Get("Logzio-Account-Id")
+			if logzAccountId != "" {
+				alert.Annotations[ngModels.LogzioAccountIdAnnotation] = logzAccountId
+			}
+		}
+		// LOGZ.IO GRAFANA CHANGE :: End
 		alerts.PostableAlerts = append(alerts.PostableAlerts, *alert)
 		if alertState.StateReason == ngModels.StateReasonMissingSeries { // do not put stale state back to state manager
 			continue