fix: adapted disableServer config in server tracer

kitex-contrib · Sep 29, 2023 · c6f3c75 · c6f3c75
1 parent a784486
commit c6f3c75
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/tracer.go b/tracer.go
@@ -186,15 +186,17 @@ func NewServerTracer(addr, path string, options ...Option) stats.Tracer {
 		opt.apply(cfg)
 	}
 
-	cfg.serveMux.Handle(path, promhttp.HandlerFor(cfg.registry, promhttp.HandlerOpts{
-		ErrorHandling: promhttp.ContinueOnError,
-		Registry:      cfg.registry,
-	}))
-	go func() {
-		if err := http.ListenAndServe(addr, cfg.serveMux); err != nil {
-			log.Fatal("Unable to start a promhttp server, err: " + err.Error())
-		}
-	}()
+	if !cfg.disableServer {
+		cfg.serveMux.Handle(path, promhttp.HandlerFor(cfg.registry, promhttp.HandlerOpts{
+			ErrorHandling: promhttp.ContinueOnError,
+			Registry:      cfg.registry,
+		}))
+		go func() {
+			if err := http.ListenAndServe(addr, cfg.serveMux); err != nil {
+				log.Fatal("Unable to start a promhttp server, err: " + err.Error())
+			}
+		}()
+	}
 
 	serverHandledCounter := prom.NewCounterVec(
 		prom.CounterOpts{