add pprof http server support with a build tag switch

pkg/cmd/pprof.go

@@ -0,0 +1,6 @@
+//go:build pprof
+// +build pprof
+
+package cmd
+
+import _ "net/http/pprof"

pkg/cmd/root.go

@@ -7,6 +7,7 @@ import (
 	"runtime/pprof"
 	"strings"
 	"time"
+	_ "time/tzdata"
 
 	"github.com/heroku/rollrus"
 	"github.com/joho/godotenv"
@@ -20,8 +21,6 @@ import (
 
 	"github.com/c9s/bbgo/pkg/bbgo"
 	"github.com/c9s/bbgo/pkg/util"
-
-	_ "time/tzdata"
 )
 
 var cpuProfileFile *os.File
@@ -81,6 +80,24 @@ var RootCmd = &cobra.Command{
 			}()
 		}
 
+		enableProfileServer, err := cmd.Flags().GetBool("enable-profile-server")
+		if err != nil {
+			return err
+		}
+
+		if enableProfileServer {
+			profileServerBind, err := cmd.Flags().GetString("profile-server-bind")
+			if err != nil {
+				return err
+			}
+
+			go func() {
+				if err := http.ListenAndServe(profileServerBind, nil); err != nil {
+					log.WithError(err).Errorf("profile server error")
+				}
+			}()
+		}
+
 		cpuProfile, err := cmd.Flags().GetString("cpu-profile")
 		if err != nil {
 			return err
@@ -195,6 +212,8 @@ func init() {
 	RootCmd.PersistentFlags().String("max-api-secret", "", "max api secret")
 
 	RootCmd.PersistentFlags().String("cpu-profile", "", "cpu profile")
+	RootCmd.PersistentFlags().Bool("enable-profile-server", false, "enable profile server binding")
+	RootCmd.PersistentFlags().String("profile-server-bind", "localhost:6060", "profile server binding")
 
 	viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))