fixed webservice metric label

snapp-incubator · Apr 8, 2024 · a7e276f · a7e276f
1 parent 2d7be3a
commit a7e276f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pkg/auth/server.go b/pkg/auth/server.go
@@ -49,10 +49,9 @@ func (a *authV2) Check(ctx context.Context, check *CheckRequestV2) (*CheckRespon
 	labels = AddUpstreamAuthLabel(labels, request.Context[HasUpstreamAuth])
 	labels[CheckRequestVersionLabel] = MetricsCheckRequestVersion2
 	reqLatency.With(labels).Observe(time.Since(reqStartTime).Seconds())
-	labels = AddWebserviceLabel(labels, string(CerberusHeaderWebservice))
+	labels = AddWebserviceLabel(labels, response.Response.Header.Get(string(CerberusHeaderWebservice)))
 	reqCount.With(labels).Inc()
 
-
 	return final_response, nil
 }
 
@@ -77,7 +76,7 @@ func (a *authV3) Check(ctx context.Context, check *CheckRequestV3) (*CheckRespon
 	labels = AddUpstreamAuthLabel(labels, request.Context[HasUpstreamAuth])
 	labels[CheckRequestVersionLabel] = MetricsCheckRequestVersion3
 	reqLatency.With(labels).Observe(time.Since(reqStartTime).Seconds())
-	labels = AddWebserviceLabel(labels, string(CerberusHeaderWebservice))
+	labels = AddWebserviceLabel(labels, response.Response.Header.Get(string(CerberusHeaderWebservice)))
 	reqCount.With(labels).Inc()
 
 	return final_response, nil