feat: add test config option to disable metrics

ipfs · Jun 25, 2024 · 16d71e2 · 16d71e2
1 parent 316f53c
commit 16d71e2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/handlers.go b/handlers.go
@@ -178,8 +178,15 @@ func setupGatewayHandler(cfg Config, nd *Node, tracingAuth string) (http.Handler
 		NoDNSLink:             noDNSLink,
 	}
 	gwHandler := gateway.NewHandler(gwConf, backend)
-	ipfsHandler := withHTTPMetrics(gwHandler, "ipfs")
-	ipnsHandler := withHTTPMetrics(gwHandler, "ipns")
+
+	var ipfsHandler, ipnsHandler http.Handler
+	if cfg.disableMetrics {
+		ipfsHandler = gwHandler
+		ipnsHandler = gwHandler
+	} else {
+		ipfsHandler = withHTTPMetrics(gwHandler, "ipfs")
+		ipnsHandler = withHTTPMetrics(gwHandler, "ipns")
+	}
 
 	topMux := http.NewServeMux()
 	topMux.Handle("/ipfs/", ipfsHandler)

diff --git a/setup.go b/setup.go
@@ -123,6 +123,8 @@ type Config struct {
 
 	GCInterval  time.Duration
 	GCThreshold float64
+
+	disableMetrics bool // only meant to be used during testing
 }
 
 func SetupNoLibp2p(ctx context.Context, cfg Config, dnsCache *cachedDNS) (*Node, error) {