Add the ability to generate a CPU profile

mpchadwick · Nov 15, 2020 · 9298d40 · 9298d40
1 parent e7c8a72
commit 9298d40
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,5 +1,6 @@
 dbanon
 dbanon.log
+dbanon.prof
 tmp/*
 /bindata
 .DS_Store

diff --git a/main.go b/main.go
@@ -10,6 +10,7 @@ import (
 	"github.com/rhysd/go-github-selfupdate/selfupdate"
 	"io/ioutil"
 	"log"
+	"runtime/pprof"
 	"os"
 )
 
@@ -40,9 +41,16 @@ func main() {
 	silent := flag.Bool("silent", false, "Disable all logging")
 	logFile := flag.String("log-file", "", "File to write logs to")
 	logLevel := flag.String("log-level", "", "Specify desired log level")
+	profile := flag.Bool("profile", false, "Generate a profile")
 
 	flag.Parse()
 
+	if *profile {
+		profF, _ := os.Create("dbanon.prof")
+		pprof.StartCPUProfile(profF)
+		defer pprof.StopCPUProfile()
+	}
+
 	if *ver {
 		fmt.Println(version)
 		os.Exit(0)