Add pprof listen addr flag

MinterTeam · Feb 12, 2020 · 685200b · 685200b
1 parent d0e4ea9
commit 685200b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/cmd/minter/cmd/node.go b/cmd/minter/cmd/node.go
@@ -47,11 +47,16 @@ func runNode(cmd *cobra.Command) error {
 	}
 
 	if pprofOn {
+		pprofAddr, err := cmd.Flags().GetString("pprof-addr")
+		if err != nil {
+			return err
+		}
+
 		pprofMux := http.DefaultServeMux
 		http.DefaultServeMux = http.NewServeMux()
 		go func() {
 			logger.Error((&http.Server{
-				Addr:    "localhost:6060",
+				Addr:    pprofAddr,
 				Handler: pprofMux,
 			}).ListenAndServe().Error())
 		}()

diff --git a/cmd/minter/main.go b/cmd/minter/main.go
@@ -19,6 +19,7 @@ func main() {
 	rootCmd.PersistentFlags().StringVar(&utils.MinterHome, "home-dir", "", "base dir (default is $HOME/.minter)")
 	rootCmd.PersistentFlags().StringVar(&utils.MinterConfig, "config", "", "path to config (default is $(home-dir)/config/config.toml)")
 	rootCmd.PersistentFlags().Bool("pprof", false, "enable pprof")
+	rootCmd.PersistentFlags().String("pprof-addr", "0.0.0.0:6060", "pprof listen addr")
 
 	if err := rootCmd.Execute(); err != nil {
 		panic(err)