arno/clair
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

cmd/clair: fix pprof

Browse Source
This commit is contained in:
Quentin Machu 2016-03-07 16:22:38 -05:00
parent ee28073d61
commit 09dda9bfd7
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
cmd/clair/main.go
View File

@ -67,25 +67,30 @@ func main() {
// Enable CPU Profiling if specified // Enable CPU Profiling if specified
if *flagCPUProfilePath != "" { if *flagCPUProfilePath != "" {
startCPUProfiling(*flagCPUProfilePath) defer stopCPUProfiling(startCPUProfiling(*flagCPUProfilePath))
defer stopCPUProfiling()
} }
clair.Boot(config) clair.Boot(config)
} }
func startCPUProfiling(path string) { func startCPUProfiling(path string) *os.File {
f, err := os.Create(path) f, err := os.Create(path)
if err != nil { if err != nil {
log.Fatalf("failed to create profile file: %s", err) log.Fatalf("failed to create profile file: %s", err)
} }
defer f.Close()
pprof.StartCPUProfile(f) err = pprof.StartCPUProfile(f)
log.Info("started CPU profiling") if err != nil {
log.Fatalf("failed to start CPU profiling: %s", err)
} }
func stopCPUProfiling() { log.Info("started CPU profiling")
return f
}
func stopCPUProfiling(f *os.File) {
pprof.StopCPUProfile() pprof.StopCPUProfile()
f.Close()
log.Info("stopped CPU profiling") log.Info("stopped CPU profiling")
} }