add if for enable debug

Labbs · Mar 13, 2021 · 6899c92 · 6899c92
1 parent 05e74d0
commit 6899c92
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/pkg/server/server.go b/pkg/server/server.go
@@ -20,11 +20,14 @@ func RunServer(ctx *cli.Context) error {
 		ctx.WriteString("/metrics")
 	})
 	r.GET("/metrics", prometheusHandler())
-	r.GET("/debug/pprof/", pprofHandlerIndex)
-	r.GET("/debug/pprof/cmdline", pprofHandlerCmdline)
-	r.GET("/debug/pprof/profile", pprofHandlerIndex)
-	r.GET("/debug/pprof/trace", pprofHandlerTrace)
-	r.GET("/debug/pprof/{profile}", pprofHandlerIndex)
+
+	if config.Debug {
+		r.GET("/debug/pprof/", pprofHandlerIndex)
+		r.GET("/debug/pprof/cmdline", pprofHandlerCmdline)
+		r.GET("/debug/pprof/profile", pprofHandlerIndex)
+		r.GET("/debug/pprof/trace", pprofHandlerTrace)
+		r.GET("/debug/pprof/{profile}", pprofHandlerIndex)
+	}
 
 	log.Print("exporter listening on 0.0.0.0:" + strconv.Itoa(config.Port))
 	return fasthttp.ListenAndServe(":"+strconv.Itoa(config.Port), r.Handler)