From 4eca2bd520025b3e66e3758c9b391e7039a0ce4a Mon Sep 17 00:00:00 2001
From: Vamshi Aruru <vamshi@tectonic.so>
Date: Fri, 25 Oct 2024 16:21:52 +0530
Subject: [PATCH] adding should send metrics key

---
 middleware/middleware.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/middleware/middleware.go b/middleware/middleware.go
index 05a0a80..bc3909f 100644
--- a/middleware/middleware.go
+++ b/middleware/middleware.go
@@ -34,8 +34,9 @@ type Config struct {
 	// by default measuring inflights is enabled (`DisableMeasureInflight` is false).
 	DisableMeasureInflight bool
 	// UseChi will use the chi router to get the handler ID, by default it will be false.
-	UseChi              bool
-	OrgSlugPathParamKey string
+	UseChi                 bool
+	OrgSlugPathParamKey    string
+	ShouldReportMetricsKey *string
 }
 
 func (c *Config) defaults() {
@@ -96,6 +97,13 @@ func (m Middleware) Measure(handlerID string, reporter Reporter, next func()) {
 	start := time.Now()
 	defer func() {
 		duration := time.Since(start)
+		if m.cfg.ShouldReportMetricsKey != nil {
+			shouldShow, ok := reporter.Context().Value(*m.cfg.ShouldReportMetricsKey).(bool)
+			if ok && !shouldShow {
+				// no need to send metrics
+				return
+			}
+		}
 
 		// If we need to group the status code, it uses the
 		// first number of the status code because is the least