getting org slug from path param

middleware/middleware.go

@@ -10,6 +10,7 @@ import (
 	"strconv"
 	"time"
 
+	"github.com/go-chi/chi/v5"
 	"github.com/slok/go-http-metrics/metrics"
 )
 
@@ -33,8 +34,8 @@ 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
-	OrgSlugContextKey any
+	UseChi              bool
+	OrgSlugPathParamKey string
 }
 
 func (c *Config) defaults() {
@@ -107,11 +108,7 @@ func (m Middleware) Measure(handlerID string, reporter Reporter, next func()) {
 		}
 
 		hid = reporter.URLPath()
-		orgSlug, ok := reporter.Context().Value(m.cfg.OrgSlugContextKey).(string)
-		if !ok {
-			orgSlug = ""
-		}
-
+		orgSlug := chi.URLParamFromCtx(reporter.Context(), m.cfg.OrgSlugPathParamKey)
 		props := metrics.HTTPReqProperties{
 			Service: m.cfg.Service,
 			ID:      hid,