LBRYCommunity/tracker
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

cmd/chihaya: add tracing flag

Browse source
This commit is contained in:
Jimmy Zelinskie 2018-07-03 23:14:33 -04:00
parent ff15955dcc
commit 17f22e77a3
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
cmd/chihaya/main.go
View file

@ -6,6 +6,7 @@ import (
"os/signal" "os/signal"
"runtime" "runtime"
"runtime/pprof" "runtime/pprof"
"runtime/trace"
"strings" "strings"
"syscall" "syscall"
@ -200,6 +201,19 @@ func PreRunCmdFunc(cmd *cobra.Command, args []string) error {
log.Info("enabled debug logging") log.Info("enabled debug logging")
} }
tracePath, err := cmd.Flags().GetString("trace")
if err != nil {
return err
}
if tracePath != "" {
f, err := os.Create(tracePath)
if err != nil {
return err
}
trace.Start(f)
log.Info("enabled tracing", log.Fields{"path": tracePath})
}
cpuProfilePath, err := cmd.Flags().GetString("cpuprofile") cpuProfilePath, err := cmd.Flags().GetString("cpuprofile")
if err != nil { if err != nil {
return err return err
@ -219,8 +233,9 @@ func PreRunCmdFunc(cmd *cobra.Command, args []string) error {
// PostRunCmdFunc handles clean up of any state initialized by command line // PostRunCmdFunc handles clean up of any state initialized by command line
// flags. // flags.
func PostRunCmdFunc(cmd *cobra.Command, args []string) error { func PostRunCmdFunc(cmd *cobra.Command, args []string) error {
// This can be called regardless because it noops when not profiling. // These can be called regardless because it noops when not profiling.
pprof.StopCPUProfile() pprof.StopCPUProfile()
trace.Stop()
return nil return nil
} }
@ -237,6 +252,7 @@ func main() {
rootCmd.Flags().String("config", "/etc/chihaya.yaml", "location of configuration file") rootCmd.Flags().String("config", "/etc/chihaya.yaml", "location of configuration file")
rootCmd.Flags().String("cpuprofile", "", "location to save a CPU profile") rootCmd.Flags().String("cpuprofile", "", "location to save a CPU profile")
rootCmd.Flags().String("trace", "", "location to save a trace")
rootCmd.Flags().Bool("debug", false, "enable debug logging") rootCmd.Flags().Bool("debug", false, "enable debug logging")
rootCmd.Flags().Bool("json", false, "enable json logging") rootCmd.Flags().Bool("json", false, "enable json logging")
if runtime.GOOS == "windows" { if runtime.GOOS == "windows" {