Add correlationid to dimensions

Azure · Jun 14, 2024 · c100a23 · c100a23
1 parent 5f4953c
commit c100a23
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/cluster/install_metrics.go b/pkg/cluster/install_metrics.go
@@ -9,6 +9,7 @@ import (
 	"strconv"
 
 	"github.com/Azure/ARO-RP/pkg/api"
+	"github.com/Azure/ARO-RP/pkg/frontend/middleware"
 )
 
 const (
@@ -23,6 +24,8 @@ const (
 func (m *manager) gatherEmitInstallMetrics(ctx context.Context) error {
 	dimensions := map[string]string{}
 
+	m.gatherCorrelationID(ctx, dimensions)
+
 	m.gatherMiscMetrics(dimensions)
 
 	err := m.gatherAuthMetrics(dimensions)
@@ -45,6 +48,13 @@ func (m *manager) gatherEmitInstallMetrics(ctx context.Context) error {
 	return nil
 }
 
+func (m *manager) gatherCorrelationID(ctx context.Context, dimensions map[string]string) {
+	correlationData := ctx.Value(middleware.ContextKeyCorrelationData).(*api.CorrelationData)
+
+	dimensions["correlationid"] = correlationData.CorrelationID
+	dimensions["clientrequestid"] = correlationData.ClientRequestID
+}
+
 func (m *manager) gatherMiscMetrics(dimensions map[string]string) {
 	dimensions["resourceid"] = m.doc.ResourceID
 	dimensions["subscriptionid"] = m.env.SubscriptionID()