diff --git a/api/main.go b/api/main.go index 40c930b73..c09af2ecc 100644 --- a/api/main.go +++ b/api/main.go @@ -10,6 +10,7 @@ import ( "github.com/romana/rlog" "github.com/up9inc/mizu/shared" "github.com/up9inc/mizu/tap" + "log" "mizuserver/pkg/api" "mizuserver/pkg/middleware" "mizuserver/pkg/models" @@ -18,6 +19,7 @@ import ( "mizuserver/pkg/utils" "os" "os/signal" + "runtime/pprof" "strings" ) @@ -65,6 +67,12 @@ func main() { go pipeChannelToSocket(socketConnection, harOutputChannel) go api.StartReadingOutbound(outboundLinkOutputChannel) } else if *aggregator { + f, err := os.Create("/app/profile.pprof") + if err != nil { + log.Fatal(err) + } + fmt.Println("Starting profiling4") + pprof.StartCPUProfile(f) socketHarOutChannel := make(chan *tap.OutputChannelItem, 1000) filteredHarChannel := make(chan *tap.OutputChannelItem)