package chihaya import ( "flag" "net/http" "os" "runtime/pprof" _ "net/http/pprof" "github.com/golang/glog" ) var ( profile string debugAddr string profileFile *os.File ) func init() { flag.StringVar(&profile, "profile", "", "if non-empty, path to write CPU profiling data") flag.StringVar(&debugAddr, "debug", "", "if non-empty, address to serve debug data") } func debugBoot() { var err error if debugAddr != "" { go func() { glog.Info("Starting debug HTTP on ", debugAddr) glog.Fatal(http.ListenAndServe(debugAddr, nil)) }() } if profile != "" { profileFile, err = os.Create(profile) if err != nil { glog.Fatalf("Failed to create profile file: %s\n", err) } pprof.StartCPUProfile(profileFile) glog.Info("Started profiling") } } func debugShutdown() { if profileFile != nil { profileFile.Close() pprof.StopCPUProfile() glog.Info("Stopped profiling") } }