add pprof to load test cli

examples/loadtest/cli/main.go

@@ -2,13 +2,17 @@ package main
 
 import (
 	"log"
+	"os"
 	"time"
 
 	"github.com/rs/zerolog"
 	"github.com/spf13/cobra"
 
 	"github.com/hatchet-dev/hatchet/pkg/config/shared"
 	"github.com/hatchet-dev/hatchet/pkg/logger"
+
+	"net/http"
+	_ "net/http/pprof"
 )
 
 var l zerolog.Logger
@@ -36,6 +40,13 @@ func main() {
 				"loadtest",
 			)
 
+			// enable pprof if requested
+			if os.Getenv("PPROF_ENABLED") == "true" {
+				go func() {
+					log.Println(http.ListenAndServe("localhost:6060", nil))
+				}()
+			}
+
 			if err := do(duration, events, delay, wait, concurrency, workerDelay, slots, failureRate, payloadSize); err != nil {
 				log.Println(err)
 				panic("load test failed")