From 238dd3930e9c58397a86f649c77912ee32e4d7f0 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Wed, 4 Mar 2020 16:32:27 +0000 Subject: can tail probe with a given timeout and then write a mapreduce result --- cmd/dserver/main.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'cmd/dserver') diff --git a/cmd/dserver/main.go b/cmd/dserver/main.go index 94f1a87..f9207d5 100644 --- a/cmd/dserver/main.go +++ b/cmd/dserver/main.go @@ -3,6 +3,10 @@ package main import ( "context" "flag" + "fmt" + "net/http" + _ "net/http" + _ "net/http/pprof" "os" "os/signal" "syscall" @@ -22,6 +26,7 @@ func main() { var debugEnable bool var displayVersion bool var noColor bool + var pprof int var shutdownAfter int var sshPort int @@ -32,6 +37,7 @@ func main() { flag.BoolVar(&noColor, "noColor", false, "Disable ANSII terminal colors") flag.IntVar(&shutdownAfter, "shutdownAfter", 0, "Automatically shutdown after so many seconds") flag.IntVar(&sshPort, "port", 2222, "SSH server port") + flag.IntVar(&pprof, "pprof", -1, "Start PProf server this port") flag.StringVar(&cfgFile, "cfg", "", "Config file path") flag.Parse() @@ -65,6 +71,13 @@ func main() { nothingEnable := false logger.Start(ctx, serverEnable, debugEnable, silentEnable, nothingEnable) + if pprof > -1 { + // For debugging purposes only + pprofArgs := fmt.Sprintf("0.0.0.0:%d", pprof) + logger.Info("Starting PProf", pprofArgs) + go http.ListenAndServe(pprofArgs, nil) + } + serv := server.New() status := serv.Start(ctx) logger.Flush() -- cgit v1.2.3