feat: introducing chi middleware for warehouse

rudderlabs · Oct 25, 2023 · 4bde439 · 4bde439
1 parent 28497cf
commit 4bde439
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/warehouse/api/http.go b/warehouse/api/http.go
@@ -25,6 +25,7 @@ import (
  ierrors "github.com/rudderlabs/rudder-server/warehouse/internal/errors"
  lf "github.com/rudderlabs/rudder-server/warehouse/logfield"
 
+ "github.com/rudderlabs/rudder-go-kit/chiware"
  "github.com/rudderlabs/rudder-go-kit/config"
  kithttputil "github.com/rudderlabs/rudder-go-kit/httputil"
  "github.com/rudderlabs/rudder-go-kit/logger"
@@ -127,6 +128,9 @@ func NewApi(
 
 func (a *Api) Start(ctx context.Context) error {
  srvMux := chi.NewRouter()
+ srvMux.Use(
+ chiware.StatMiddleware(ctx, srvMux, a.statsFactory, "warehouse"),
+ )
 
  if mode.IsStandAlone(a.mode) {
  srvMux.Get("/health", a.healthHandler)