From a7e276f688f38afdfc6c4a94cf4fa1ca438c3a65 Mon Sep 17 00:00:00 2001 From: Pedram Sadeghian Date: Mon, 8 Apr 2024 06:34:35 +0330 Subject: [PATCH] fixed webservice metric label --- pkg/auth/server.go | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pkg/auth/server.go b/pkg/auth/server.go index e5e6f40..573a350 100644 --- 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