28 lines
443 B
Go
28 lines
443 B
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
"os/signal"
|
||
|
"runtime/pprof"
|
||
|
|
||
|
"github.com/coreos/etcd/log"
|
||
|
)
|
||
|
|
||
|
// profile starts CPU profiling.
|
||
|
func profile(path string) {
|
||
|
f, err := os.Create(path)
|
||
|
if err != nil {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
pprof.StartCPUProfile(f)
|
||
|
|
||
|
c := make(chan os.Signal, 1)
|
||
|
signal.Notify(c, os.Interrupt)
|
||
|
go func() {
|
||
|
sig := <-c
|
||
|
log.Infof("captured %v, stopping profiler and exiting..", sig)
|
||
|
pprof.StopCPUProfile()
|
||
|
os.Exit(1)
|
||
|
}()
|
||
|
}
|