go/chroma
1
0
Fork 0
mirror of https://github.com/alecthomas/chroma.git synced 2025-03-29 21:56:56 +02:00
Code Issues Releases Activity

Ensure ^C captures profile information.

Browse Source
This commit is contained in:
Alec Thomas 2017-06-07 20:39:58 +10:00
parent a0db747409
commit a3862242d5
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/chroma/main.go
View File

@ -6,6 +6,7 @@ import (
"io"
"io/ioutil"
"os"
"os/signal"
"runtime/pprof"
"strings"
@ -38,6 +39,13 @@ func main() {
f, err := os.Create(*profileFlag)
kingpin.FatalIfError(err, "")
pprof.StartCPUProfile(f)
signals := make(chan os.Signal, 1)
signal.Notify(signals, os.Interrupt)
go func() {
<-signals
pprof.StopCPUProfile()
os.Exit(128 + 3)
}()
defer pprof.StopCPUProfile()
}
w := bufio.NewWriterSize(os.Stdout, 16384)