Add go profiling CLI flag

1 files changed, 14 insertions, 0 deletions

diff --git a/main.go b/main.go
index ebeaf87c1..8e01567cc 100644
--- a/main.go
+++ b/main.go
@@ -5,14 +5,17 @@ import (
 	"fmt"
 	"github.com/golang/protobuf/proto"
 	"github.com/ry/v8worker2"
+	"log"
 	"net/url"
 	"os"
 	"path"
+	"runtime/pprof"
 )
 
 var flagReload = flag.Bool("reload", false, "Reload cached remote source code.")
 var flagV8Options = flag.Bool("v8-options", false, "Print V8 command line options.")
 var flagDebug = flag.Bool("debug", false, "Enable debug output.")
+var flagGoProf = flag.String("goprof", "", "Write golang cpu profile to file.")
 
 var DenoDir string
 var CompileDir string
@@ -53,6 +56,17 @@ func main() {
 	}
 	args = v8worker2.SetFlags(args)
 
+	// Maybe start Golang CPU profiler.
+	// Use --prof for profiling JS.
+	if *flagGoProf != "" {
+		f, err := os.Create(*flagGoProf)
+		if err != nil {
+			log.Fatal(err)
+		}
+		pprof.StartCPUProfile(f)
+		defer pprof.StopCPUProfile()
+	}
+
 	createDirs()
 	createWorker()