From 445a20f1c1772129baab48156fac083f4fc8f1e4 Mon Sep 17 00:00:00 2001 From: Trong Huu Nguyen Date: Mon, 18 Jul 2022 09:40:18 +0200 Subject: [PATCH] refactor(middleware/prometheus): add host label --- pkg/middleware/prometheus.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pkg/middleware/prometheus.go b/pkg/middleware/prometheus.go index 5a18b13..376aee4 100644 --- a/pkg/middleware/prometheus.go +++ b/pkg/middleware/prometheus.go @@ -37,7 +37,7 @@ func NewPrometheusMiddleware(name, provider string, buckets ...float64) *Prometh Help: "How many HTTP requests processed, partitioned by status code, method and HTTP path.", ConstLabels: prometheus.Labels{"service": name, "provider": provider}, }, - []string{"code", "method", "path"}, + []string{"code", "method", "path", "host"}, ) if len(buckets) == 0 { @@ -49,7 +49,7 @@ func NewPrometheusMiddleware(name, provider string, buckets ...float64) *Prometh ConstLabels: prometheus.Labels{"service": name, "provider": provider}, Buckets: buckets, }, - []string{"code", "method", "path"}, + []string{"code", "method", "path", "host"}, ) prometheus.Register(m.reqs) @@ -73,8 +73,8 @@ func (m *PrometheusMiddleware) Handler(next http.Handler) http.Handler { next.ServeHTTP(ww, r) statusCode := strconv.Itoa(ww.Status()) duration := time.Since(start) - m.reqs.WithLabelValues(statusCode, r.Method, r.URL.Path).Inc() - m.latency.WithLabelValues(statusCode, r.Method, r.URL.Path).Observe(duration.Seconds()) + m.reqs.WithLabelValues(statusCode, r.Method, r.URL.Path, r.Host).Inc() + m.latency.WithLabelValues(statusCode, r.Method, r.URL.Path, r.Host).Observe(duration.Seconds()) } return http.HandlerFunc(fn) }